package auth
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/MichaelMure/git-bug/entity"
"github.com/MichaelMure/git-bug/identity"
"github.com/MichaelMure/git-bug/repository"
)
func TestCredential(t *testing.T) {
repo := repository.NewMockRepoForTest()
user1 := identity.NewIdentity("user1", "email")
err := user1.Commit(repo)
assert.NoError(t, err)
user2 := identity.NewIdentity("user2", "email")
err = user2.Commit(repo)
assert.NoError(t, err)
storeToken := func(user identity.Interface, val string, target string) *Token {
token := NewToken(user.Id(), val, target)
err = Store(repo, token)
require.NoError(t, err)
return token
}
token := storeToken(user1, "foobar", "github")
// Store + Load
err = Store(repo, token)
assert.NoError(t, err)
token2, err := LoadWithId(repo, token.ID())
assert.NoError(t, err)
assert.Equal(t, token.createTime.Unix(), token2.CreateTime().Unix())
token.createTime = token2.CreateTime()
assert.Equal(t, token, token2)
prefix := string(token.ID())[:10]
// LoadWithPrefix
token3, err := LoadWithPrefix(repo, prefix)
assert.NoError(t, err)
assert.Equal(t, token.createTime.Unix(), token3.CreateTime().Unix())
token.createTime = token3.CreateTime()
assert.Equal(t, token, token3)
token4 := storeToken(user1, "foo", "gitlab")
token5 := storeToken(user2, "bar", "github")
// List + options
creds, err := List(repo, WithTarget("github"))
assert.NoError(t, err)
sameIds(t, creds, []Credential{token, token5})
creds, err = List(repo, WithTarget("gitlab"))
assert.NoError(t, err)
sameIds(t, creds, []Credential{token4})
creds, err = List(repo, WithUser(user1))
assert.NoError(t, err)
sameIds(t, creds, []Credential{token, token4})
creds, err = List(repo, WithUserId(user1.Id()))
assert.NoError(t, err)
sameIds(t, creds, []Credential{token, token4})
creds, err = List(repo, WithKind(KindToken))
assert.NoError(t, err)
sameIds(t, creds, []Credential{token, token4, token5})
creds, err = List(repo, WithKind(KindLoginPassword))
assert.NoError(t, err)
sameIds(t, creds, []Credential{})
// Exist
exist := IdExist(repo, token.ID())
assert.True(t, exist)
exist = PrefixExist(repo, prefix)
assert.True(t, exist)
// Remove
err = Remove(repo, token.ID())
assert.NoError(t, err)
creds, err = List(repo)
assert.NoError(t, err)
sameIds(t, creds, []Credential{token4, token5})
}
func sameIds(t *testing.T, a []Credential, b []Credential) {
t.Helper()
ids := func(creds []Credential) []entity.Id {
result := make([]entity.Id, len(creds))
for i, cred := range creds {
result[i] = cred.ID()
}
return result
}
assert.ElementsMatch(t, ids(a), ids(b))
}
