extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-01-22 05:49:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add tests

Browse Source 
Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
This commit is contained in:
bcmmbaga 2024-11-22 17:53:34 +03:00
parent 7a6ca3ee37
commit 2115e2c3f0
No known key found for this signature in database
GPG Key ID: 511EED5C928AD547
1 changed files with 73 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
management/server/sql_store_test.go
View File

@ -2589,3 +2589,76 @@ func TestSqlStore_DeleteRoute(t *testing.T) {
require.Error(t, err)
require.Nil(t, route)
}
func TestSqlStore_GetAccountSettings(t *testing.T) {
store, cleanup, err := NewTestStoreFromSQL(context.Background(), "testdata/store.sql", t.TempDir())
t.Cleanup(cleanup)
require.NoError(t, err)
tests := []struct {
name string
accountID string
expectError bool
}{
{
name: "retrieve existing account settings",
accountID: "bf1c8084-ba50-4ce7-9439-34653001fc3b",
expectError: false,
},
{
name: "retrieve non-existing account settings",
accountID: "non-existing",
expectError: true,
},
{
name: "retrieve account settings with empty account ID",
accountID: "",
expectError: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
settings, err := store.GetAccountSettings(context.Background(), LockingStrengthShare, tt.accountID)
if tt.expectError {
require.Error(t, err)
sErr, ok := status.FromError(err)
require.True(t, ok)
require.Equal(t, sErr.Type(), status.NotFound)
require.Nil(t, settings)
} else {
require.NoError(t, err)
require.False(t, settings.RegularUsersViewBlocked)
require.False(t, settings.JWTGroupsEnabled)
require.False(t, settings.GroupsPropagationEnabled)
require.False(t, settings.PeerInactivityExpirationEnabled)
require.False(t, settings.PeerLoginExpirationEnabled)
require.False(t, settings.Extra.PeerApprovalEnabled)
require.Equal(t, time.Duration(86400000000000), settings.PeerLoginExpiration)
require.Equal(t, time.Duration(0), settings.PeerInactivityExpiration)
require.Len(t, settings.Extra.IntegratedValidatorGroups, 0)
}
})
}
}
func TestSqlStore_SaveAccountSettings(t *testing.T) {
store, cleanup, err := NewTestStoreFromSQL(context.Background(), "testdata/store.sql", t.TempDir())
t.Cleanup(cleanup)
require.NoError(t, err)
accountID := "bf1c8084-ba50-4ce7-9439-34653001fc3b"
settings, err := store.GetAccountSettings(context.Background(), LockingStrengthShare, accountID)
require.NoError(t, err)
settings.Extra.IntegratedValidatorGroups = []string{"groupA"}
settings.RegularUsersViewBlocked = true
settings.PeerInactivityExpiration = 30 * time.Minute
err = store.SaveAccountSettings(context.Background(), LockingStrengthUpdate, accountID, settings)
require.NoError(t, err)
saveSettings, err := store.GetAccountSettings(context.Background(), LockingStrengthShare, accountID)
require.NoError(t, err)
require.Equal(t, settings, saveSettings)
}