From 9ee08fc4413d5715dd350e6ec7be6969443e281c Mon Sep 17 00:00:00 2001 From: bcmmbaga Date: Wed, 9 Oct 2024 17:04:20 +0300 Subject: [PATCH] fix nameserver tests Signed-off-by: bcmmbaga --- management/server/nameserver_test.go | 38 +++++++++++++++++++--------- 1 file changed, 26 insertions(+), 12 deletions(-) diff --git a/management/server/nameserver_test.go b/management/server/nameserver_test.go index b810036a1..300f79c42 100644 --- a/management/server/nameserver_test.go +++ b/management/server/nameserver_test.go @@ -939,7 +939,7 @@ func TestValidateDomain(t *testing.T) { } func TestNameServerAccountPeersUpdate(t *testing.T) { - manager, account, peer1, peer2, _ := setupNetworkMapTest(t) + manager, account, peer1, peer2, peer3 := setupNetworkMapTest(t) var newNameServerGroupA *nbdns.NameServerGroup var newNameServerGroupB *nbdns.NameServerGroup @@ -953,7 +953,7 @@ func TestNameServerAccountPeersUpdate(t *testing.T) { { ID: "groupB", Name: "GroupB", - Peers: []string{peer1.ID, peer2.ID}, + Peers: []string{peer1.ID, peer2.ID, peer3.ID}, }, }) assert.NoError(t, err) @@ -1043,11 +1043,18 @@ func TestNameServerAccountPeersUpdate(t *testing.T) { close(done) }() - newNameServerGroupB.NameServers = []nbdns.NameServer{{ - IP: netip.MustParseAddr("1.1.1.2"), - NSType: nbdns.UDPNameServerType, - Port: nbdns.DefaultDNSPort, - }} + newNameServerGroupB.NameServers = []nbdns.NameServer{ + { + IP: netip.MustParseAddr("1.1.1.2"), + NSType: nbdns.UDPNameServerType, + Port: nbdns.DefaultDNSPort, + }, + { + IP: netip.MustParseAddr("8.8.8.8"), + NSType: nbdns.UDPNameServerType, + Port: nbdns.DefaultDNSPort, + }, + } err = manager.SaveNameServerGroup(context.Background(), account.Id, userID, newNameServerGroupB) assert.NoError(t, err) @@ -1066,11 +1073,18 @@ func TestNameServerAccountPeersUpdate(t *testing.T) { close(done) }() - newNameServerGroupB.NameServers = []nbdns.NameServer{{ - IP: netip.MustParseAddr("1.1.1.2"), - NSType: nbdns.UDPNameServerType, - Port: nbdns.DefaultDNSPort, - }} + newNameServerGroupB.NameServers = []nbdns.NameServer{ + { + IP: netip.MustParseAddr("1.1.1.2"), + NSType: nbdns.UDPNameServerType, + Port: nbdns.DefaultDNSPort, + }, + { + IP: netip.MustParseAddr("8.8.8.8"), + NSType: nbdns.UDPNameServerType, + Port: nbdns.DefaultDNSPort, + }, + } err = manager.SaveNameServerGroup(context.Background(), account.Id, userID, newNameServerGroupB) assert.NoError(t, err)