// Copyright 2025 eat-pray-ai & OpenWaygate
// SPDX-License-Identifier: Apache-2.0
package auth
import (
"context"
"io/fs"
"os"
"path/filepath"
"reflect"
"testing"
"testing/fstest"
)
const (
credFile = "client_secret.json"
tokenFile = "youtube.token.json"
credential = `{"client_id":"test"}`
cacheToken = `{"access_token":"test"}`
credB64 = "eyJjbGllbnRfaWQiOiJ0ZXN0In0="
tokenB64 = "eyJhY2Nlc3NfdG9rZW4iOiJ0ZXN0In0="
)
func TestNewY2BService(t *testing.T) {
rootDir, _ := os.Getwd()
absCred := filepath.Join(rootDir, credFile)
absToken := filepath.Join(rootDir, tokenFile)
mockFS := fstest.MapFS{
rootDir: &fstest.MapFile{Mode: fs.ModeDir},
credFile: &fstest.MapFile{Data: []byte(credential)},
tokenFile: &fstest.MapFile{Data: []byte(cacheToken)},
}
type args struct {
opts []Option
}
tests := []struct {
name string
args args
want Svc
}{
{
name: "with all options - json",
args: args{
opts: []Option{
WithCredential(credential, mockFS),
WithCacheToken(cacheToken, mockFS),
},
},
want: &svc{
Credential: credential,
CacheToken: cacheToken,
credFile: credFile,
ctx: context.Background(),
},
},
{
name: "with all options - base64",
args: args{
opts: []Option{
WithCredential(credB64, mockFS),
WithCacheToken(tokenB64, mockFS),
},
},
want: &svc{
Credential: credential,
CacheToken: cacheToken,
credFile: credFile,
ctx: context.Background(),
},
},
{
name: "with all options - file",
args: args{
opts: []Option{
WithCredential(absCred, mockFS),
WithCacheToken(absToken, mockFS),
},
},
want: &svc{
Credential: credential,
CacheToken: cacheToken,
credFile: absCred,
tokenFile: tokenFile,
ctx: context.Background(),
},
},
{
name: "with no options",
args: args{
opts: []Option{},
},
want: &svc{
credFile: credFile,
ctx: context.Background(),
},
},
}
for _, tt := range tests {
t.Run(
tt.name, func(t *testing.T) {
if got := NewY2BService(tt.args.opts...); !reflect.DeepEqual(
got, tt.want,
) {
t.Errorf("NewY2BService() = %v, want %v", got, tt.want)
}
},
)
}
}
