fix tests

Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
This commit is contained in:
bcmmbaga
2024-09-25 15:04:57 +03:00
parent b815393180
commit c384874d7d
2 changed files with 14 additions and 3 deletions

View File

@@ -635,7 +635,16 @@ func (s *FileStore) GetUserByUserID(_ context.Context, _ LockingStrength, userID
return nil, err return nil, err
} }
return account.Users[userID].Copy(), nil user := account.Users[userID].Copy()
pat := make([]PersonalAccessToken, 0, len(user.PATs))
for _, token := range user.PATs {
if token != nil {
pat = append(pat, *token)
}
}
user.PATsG = pat
return user, nil
} }
func (s *FileStore) GetAccountGroups(ctx context.Context, accountID string) ([]*nbgroup.Group, error) { func (s *FileStore) GetAccountGroups(ctx context.Context, accountID string) ([]*nbgroup.Group, error) {

View File

@@ -200,6 +200,7 @@ func TestUser_GetPAT(t *testing.T) {
account := newAccountWithId(context.Background(), mockAccountID, mockUserID, "") account := newAccountWithId(context.Background(), mockAccountID, mockUserID, "")
account.Users[mockUserID] = &User{ account.Users[mockUserID] = &User{
Id: mockUserID, Id: mockUserID,
AccountID: mockAccountID,
PATs: map[string]*PersonalAccessToken{ PATs: map[string]*PersonalAccessToken{
mockTokenID1: { mockTokenID1: {
ID: mockTokenID1, ID: mockTokenID1,
@@ -232,6 +233,7 @@ func TestUser_GetAllPATs(t *testing.T) {
account := newAccountWithId(context.Background(), mockAccountID, mockUserID, "") account := newAccountWithId(context.Background(), mockAccountID, mockUserID, "")
account.Users[mockUserID] = &User{ account.Users[mockUserID] = &User{
Id: mockUserID, Id: mockUserID,
AccountID: mockAccountID,
PATs: map[string]*PersonalAccessToken{ PATs: map[string]*PersonalAccessToken{
mockTokenID1: { mockTokenID1: {
ID: mockTokenID1, ID: mockTokenID1,