sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-21 09:46:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source 
Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
This commit is contained in:
bcmmbaga
2024-11-07 11:39:24 +03:00
parent df6c9a528a
commit 197d844a16
23 changed files with 314 additions and 262 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
management/server/setupkey_test.go
View File

@@ -30,14 +30,16 @@ func TestDefaultAccountManager_SaveSetupKey(t *testing.T) {
err = manager.SaveGroups(context.Background(), accountID, userID, []*nbgroup.Group{
{
ID: "group_1",
Name: "group_name_1",
Peers: []string{},
ID: "group_1",
AccountID: accountID,
Name: "group_name_1",
Peers: []string{},
},
{
ID: "group_2",
Name: "group_name_2",
Peers: []string{},
ID: "group_2",
AccountID: accountID,
Name: "group_name_2",
Peers: []string{},
},
})
if err != nil {
@@ -80,7 +82,7 @@ func TestDefaultAccountManager_SaveSetupKey(t *testing.T) {
assert.Equal(t, userID, ev.InitiatorID)
assert.Equal(t, key.Id, ev.TargetID)
groupAll, err := manager.GetGroupByName(context.Background(), accountID, "All")
groupAll, err := manager.GetGroupByName(context.Background(), "All", accountID)
require.NoError(t, err)
// saving setup key with All group assigned to auto groups should return error
@@ -105,24 +107,26 @@ func TestDefaultAccountManager_CreateSetupKey(t *testing.T) {
require.NoError(t, err, "failed to get or create account ID")
err = manager.SaveGroup(context.Background(), accountID, userID, &nbgroup.Group{
ID: "group_1",
Name: "group_name_1",
Peers: []string{},
ID: "group_1",
AccountID: accountID,
Name: "group_name_1",
Peers: []string{},
})
if err != nil {
t.Fatal(err)
}
err = manager.SaveGroup(context.Background(), accountID, userID, &nbgroup.Group{
ID: "group_2",
Name: "group_name_2",
Peers: []string{},
ID: "group_2",
AccountID: accountID,
Name: "group_name_2",
Peers: []string{},
})
if err != nil {
t.Fatal(err)
}
groupAll, err := manager.GetGroupByName(context.Background(), accountID, "All")
groupAll, err := manager.GetGroupByName(context.Background(), "All", accountID)
require.NoError(t, err)
type testCase struct {
@@ -378,20 +382,24 @@ func TestSetupKeyAccountPeersUpdate(t *testing.T) {
manager, account, peer1, peer2, peer3 := setupNetworkMapTest(t)
err := manager.SaveGroup(context.Background(), account.Id, userID, &nbgroup.Group{
ID: "groupA",
Name: "GroupA",
Peers: []string{peer1.ID, peer2.ID, peer3.ID},
ID: "groupA",
AccountID: account.Id,
Name: "GroupA",
Peers: []string{peer1.ID, peer2.ID, peer3.ID},
})
assert.NoError(t, err)
policy := Policy{
ID: "policy",
Enabled: true,
ID: "policy",
AccountID: account.Id,
Enabled: true,
Rules: []*PolicyRule{
{
ID: "Rule",
PolicyID: "policy",
Enabled: true,
Sources: []string{"groupA"},
Destinations: []string{"group"},
Destinations: []string{"groupA"},
Bidirectional: true,
Action: PolicyTrafficActionAccept,
},