extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-02-12 08:19:30 +01: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-09-06 12:15:13 +03:00
parent 5dbdeff77a
commit 4e2cf9c63a
No known key found for this signature in database
GPG Key ID: 511EED5C928AD547
2 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
management/server/account_test.go
View File

@ -1185,7 +1185,17 @@ func TestAccountManager_NetworkUpdates_DeletePolicy(t *testing.T) {
} }
func TestAccountManager_NetworkUpdates_SavePolicy(t *testing.T) { func TestAccountManager_NetworkUpdates_SavePolicy(t *testing.T) {
manager, account, peer1, _, _ := setupNetworkMapTest(t) manager, account, peer1, peer2, _ := setupNetworkMapTest(t)
group := group.Group{
ID: "group-id",
Name: "GroupA",
Peers: []string{peer1.ID, peer2.ID},
}
if err := manager.SaveGroup(context.Background(), account.Id, userID, &group); err != nil {
t.Errorf("save group: %v", err)
return
}
updMsg := manager.peersUpdateManager.CreateChannel(context.Background(), peer1.ID) updMsg := manager.peersUpdateManager.CreateChannel(context.Background(), peer1.ID)
defer manager.peersUpdateManager.CloseChannel(context.Background(), peer1.ID) defer manager.peersUpdateManager.CloseChannel(context.Background(), peer1.ID)

13
management/server/dns_test.go
View File

@ -485,7 +485,7 @@ func TestDNSAccountPeerUpdate(t *testing.T) {
err := manager.SaveGroup(context.Background(), account.Id, userID, &group.Group{ err := manager.SaveGroup(context.Background(), account.Id, userID, &group.Group{
ID: "group-id", ID: "group-id",
Name: "GroupA", Name: "GroupA",
Peers: []string{peer1.ID, peer2.ID, peer3.ID}, Peers: []string{},
}) })
assert.NoError(t, err) assert.NoError(t, err)
@ -494,7 +494,7 @@ func TestDNSAccountPeerUpdate(t *testing.T) {
manager.peersUpdateManager.CloseChannel(context.Background(), peer1.ID) manager.peersUpdateManager.CloseChannel(context.Background(), peer1.ID)
}) })
// Saving DNS settings with unused groups should not update account peers and not send peer update // Saving DNS settings with groups that have no peers should not trigger updates to account peers or send peer updates
t.Run("saving dns setting with unused groups", func(t *testing.T) { t.Run("saving dns setting with unused groups", func(t *testing.T) {
done := make(chan struct{}) done := make(chan struct{})
go func() { go func() {
@ -514,6 +514,13 @@ func TestDNSAccountPeerUpdate(t *testing.T) {
} }
}) })
err = manager.SaveGroup(context.Background(), account.Id, userID, &group.Group{
ID: "group-id",
Name: "GroupA",
Peers: []string{peer1.ID, peer2.ID, peer3.ID},
})
assert.NoError(t, err)
_, err = manager.CreateNameServerGroup( _, err = manager.CreateNameServerGroup(
context.Background(), account.Id, "ns-group-1", "ns-group-1", []dns.NameServer{{ context.Background(), account.Id, "ns-group-1", "ns-group-1", []dns.NameServer{{
IP: netip.MustParseAddr(peer1.IP.String()), IP: netip.MustParseAddr(peer1.IP.String()),
@ -525,7 +532,7 @@ func TestDNSAccountPeerUpdate(t *testing.T) {
) )
assert.NoError(t, err) assert.NoError(t, err)
// Saving DNS settings with used groups should update account peers and send peer update // Saving DNS settings with groups that have peers should update account peers and send peer update
t.Run("saving dns setting with used groups", func(t *testing.T) { t.Run("saving dns setting with used groups", func(t *testing.T) {
done := make(chan struct{}) done := make(chan struct{})
go func() { go func() {