extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2024-12-12 09:50:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add missing store tests

Browse Source 
Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
This commit is contained in:
bcmmbaga 2024-11-27 15:40:46 +03:00
parent 875b8d662c
commit 1e24916dac
No known key found for this signature in database
GPG Key ID: 511EED5C928AD547
1 changed files with 85 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
management/server/sql_store_test.go
View File

@ -15,6 +15,7 @@ import (
"github.com/google/uuid" "github.com/google/uuid"
nbdns "github.com/netbirdio/netbird/dns" nbdns "github.com/netbirdio/netbird/dns"
"github.com/netbirdio/netbird/management/server/account"
nbgroup "github.com/netbirdio/netbird/management/server/group" nbgroup "github.com/netbirdio/netbird/management/server/group"
"github.com/netbirdio/netbird/management/server/posture" "github.com/netbirdio/netbird/management/server/posture"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
@ -1269,6 +1270,19 @@ func TestSqlStore_UpdateAccountDomainAttributes(t *testing.T) {
require.Error(t, err) require.Error(t, err)
}) })
t.Run("Should update domain and category but skip primary account when isPrimary is nil", func(t *testing.T) {
domain := "test.com"
category := "private"
err = store.UpdateAccountDomainAttributes(context.Background(), LockingStrengthUpdate, accountID, domain, category, nil)
require.NoError(t, err)
account, err := store.GetAccount(context.Background(), accountID)
require.NoError(t, err)
require.Equal(t, domain, account.Domain)
require.Equal(t, category, account.DomainCategory)
require.True(t, account.IsDomainPrimaryAccount)
})
} }
func TestSqlite_GetGroupByName(t *testing.T) { func TestSqlite_GetGroupByName(t *testing.T) {
@ -2737,3 +2751,74 @@ func TestSqlStore_GetTotalAccounts(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, int64(1), totalAccounts) require.Equal(t, int64(1), totalAccounts)
} }
func TestSqlStore_AccountExists(t *testing.T) {
store, cleanup, err := NewTestStoreFromSQL(context.Background(), "testdata/store.sql", t.TempDir())
t.Cleanup(cleanup)
require.NoError(t, err)
t.Run("existing account", func(t *testing.T) {
exists, err := store.AccountExists(context.Background(), LockingStrengthShare, "bf1c8084-ba50-4ce7-9439-34653001fc3b")
require.NoError(t, err)
require.True(t, exists)
})
t.Run("non-existing account", func(t *testing.T) {
exists, err := store.AccountExists(context.Background(), LockingStrengthShare, "non-existing")
require.NoError(t, err)
require.False(t, exists)
})
}
func TestSqlStore_CreateAccount(t *testing.T) {
store, cleanup, err := NewTestStoreFromSQL(context.Background(), "testdata/store.sql", t.TempDir())
t.Cleanup(cleanup)
require.NoError(t, err)
acc := &Account{
Id: "test-account",
CreatedAt: time.Now().UTC(),
Network: NewNetwork(),
CreatedBy: userID,
Domain: "test-domain",
DNSSettings: DNSSettings{
DisabledManagementGroups: make([]string, 0),
},
Settings: &Settings{
PeerLoginExpirationEnabled: true,
PeerLoginExpiration: DefaultPeerLoginExpiration,
GroupsPropagationEnabled: true,
RegularUsersViewBlocked: true,
PeerInactivityExpirationEnabled: false,
PeerInactivityExpiration: DefaultPeerInactivityExpiration,
Extra: &account.ExtraSettings{
PeerApprovalEnabled: false,
IntegratedValidatorGroups: make([]string, 0),
},
},
}
err = store.CreateAccount(context.Background(), LockingStrengthUpdate, acc)
require.NoError(t, err)
storeAccount, err := store.GetAccount(context.Background(), acc.Id)
require.NoError(t, err)
require.NotNil(t, storeAccount)
require.Equal(t, acc.Id, storeAccount.Id)
require.Equal(t, acc.CreatedBy, storeAccount.CreatedBy)
require.Equal(t, acc.Domain, storeAccount.Domain)
require.WithinDuration(t, acc.CreatedAt, storeAccount.CreatedAt, time.Second)
require.Equal(t, acc.DNSSettings.DisabledManagementGroups, storeAccount.DNSSettings.DisabledManagementGroups)
require.NotNil(t, storeAccount.Settings)
require.Equal(t, acc.Settings.PeerLoginExpirationEnabled, storeAccount.Settings.PeerLoginExpirationEnabled)
require.Equal(t, acc.Settings.PeerLoginExpiration, storeAccount.Settings.PeerLoginExpiration)
require.Equal(t, acc.Settings.GroupsPropagationEnabled, storeAccount.Settings.GroupsPropagationEnabled)
require.Equal(t, acc.Settings.RegularUsersViewBlocked, storeAccount.Settings.RegularUsersViewBlocked)
require.Equal(t, acc.Settings.PeerInactivityExpirationEnabled, storeAccount.Settings.PeerInactivityExpirationEnabled)
require.Equal(t, acc.Settings.PeerInactivityExpiration, storeAccount.Settings.PeerInactivityExpiration)
require.NotNil(t, storeAccount.Settings.Extra)
require.Equal(t, acc.Settings.Extra.PeerApprovalEnabled, storeAccount.Settings.Extra.PeerApprovalEnabled)
require.Equal(t, acc.Settings.Extra.IntegratedValidatorGroups, storeAccount.Settings.Extra.IntegratedValidatorGroups)
}