From c603c40a5322c8dbb775b552db811ab7063eb30c Mon Sep 17 00:00:00 2001 From: bcmmbaga Date: Mon, 13 Jan 2025 20:56:51 +0300 Subject: [PATCH] Add get peer groups tests Signed-off-by: bcmmbaga --- management/server/store/sql_store_test.go | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/management/server/store/sql_store_test.go b/management/server/store/sql_store_test.go index 34578eca8..cb51dab51 100644 --- a/management/server/store/sql_store_test.go +++ b/management/server/store/sql_store_test.go @@ -2635,6 +2635,27 @@ func TestSqlStore_AddPeerToAccount(t *testing.T) { assert.WithinDurationf(t, peer.Status.LastSeen, storedPeer.Status.LastSeen.UTC(), time.Millisecond, "LastSeen should be equal") } +func TestSqlStore_GetPeerGroups(t *testing.T) { + store, cleanup, err := NewTestStoreFromSQL(context.Background(), "../testdata/store_policy_migrate.sql", t.TempDir()) + t.Cleanup(cleanup) + require.NoError(t, err) + + accountID := "bf1c8084-ba50-4ce7-9439-34653001fc3b" + peerID := "cfefqs706sqkneg59g4g" + + groups, err := store.GetPeerGroups(context.Background(), LockingStrengthShare, accountID, peerID) + require.NoError(t, err) + assert.Len(t, groups, 1) + assert.Equal(t, groups[0].Name, "All") + + err = store.AddPeerToGroup(context.Background(), LockingStrengthUpdate, accountID, peerID, "cfefqs706sqkneg59g4h") + require.NoError(t, err) + + groups, err = store.GetPeerGroups(context.Background(), LockingStrengthShare, accountID, peerID) + require.NoError(t, err) + assert.Len(t, groups, 2) +} + func TestSqlStore_GetAccountPeers(t *testing.T) { store, cleanup, err := NewTestStoreFromSQL(context.Background(), "../testdata/store_with_expired_peers.sql", t.TempDir()) t.Cleanup(cleanup)