Fix tests

Signed-off-by: bcmmbaga <[email protected]>
netbirdio · Nov 27, 2024 · 91b97d6 · 91b97d6
1 parent 3186876
commit 91b97d6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/management/server/group_test.go b/management/server/group_test.go
@@ -139,6 +139,8 @@ func TestDefaultAccountManager_DeleteGroup(t *testing.T) {
 }
 
 func TestDefaultAccountManager_DeleteGroups(t *testing.T) {
+	t.Setenv("NETBIRD_STORE_ENGINE", string(PostgresStoreEngine))
+
 	am, err := createManager(t)
 	assert.NoError(t, err, "Failed to create account manager")
 
@@ -389,7 +391,7 @@ func initTestGroupAccount(am *DefaultAccountManager) (*DefaultAccountManager, *A
 		return nil, nil, err
 	}
 
-	err = am.Store.SavePolicy(context.Background(), LockingStrengthUpdate, policy)
+	err = am.Store.CreatePolicy(context.Background(), LockingStrengthUpdate, policy)
 	if err != nil {
 		return nil, nil, err
 	}
@@ -406,7 +408,7 @@ func initTestGroupAccount(am *DefaultAccountManager) (*DefaultAccountManager, *A
 
 	err = am.SaveGroups(context.Background(), accountID, groupAdminUserID, []*nbgroup.Group{
 		groupForRoute, groupForRoute2, groupForNameServerGroups, groupForPolicies,
-		groupForSetupKeys, groupForUsers, groupForUsers, groupForIntegration,
+		groupForSetupKeys, groupForUsers, groupForIntegration,
 	})
 	if err != nil {
 		return nil, nil, err