mirror of
https://github.com/netbirdio/netbird.git
synced 2026-08-27 18:11:29 +02:00
62 lines
2.6 KiB
Go
62 lines
2.6 KiB
Go
//go:build integration
|
|
|
|
package networkmap_pgsql
|
|
|
|
import (
|
|
"context"
|
|
"net/netip"
|
|
"testing"
|
|
|
|
"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetNameServerGroups(t *testing.T) {
|
|
ctx := context.TODO()
|
|
|
|
execQuery(t, ctx,
|
|
`insert into name_server_groups (id, public_id, name, description, name_servers, groups, domains, enabled, search_domains_enabled, "primary", account_id)
|
|
VALUES('nsgroup-1','nsgroup-1-public','nsgroup-1','nsgroup-1','[{"IP":"192.168.31.2","NSType":1,"Port":53}]','["group-one-resource-id"]','["test-1.com"]',TRUE,FALSE,TRUE,'account-1')`)
|
|
execQuery(t, ctx,
|
|
`insert into name_server_groups (id, public_id, name, description, name_servers, groups, domains, enabled, search_domains_enabled,"primary",account_id)
|
|
VALUES('nsgroup-2','nsgroup-2-public','nsgroup-2','nsgroup-2','[{"IP":"192.168.32.3","NSType":1,"Port":53}]','["group-one-resource-id","group-no-resources-id"]','["test-1.com","test-2.com"]',TRUE,FALSE,TRUE,'account-1')`)
|
|
execQuery(t, ctx,
|
|
`insert into name_server_groups (id, public_id, name, description, name_servers, groups, domains, enabled, search_domains_enabled,"primary",account_id)
|
|
VALUES('nsgroup-3','nsgroup-3-public',null,null,null,null,null,TRUE,FALSE,FALSE,'account-1')`)
|
|
|
|
nsgroups, err := conn(t, ctx).GetNameServerGroups(ctx, "account-1")
|
|
assert.NoError(t, err)
|
|
|
|
assert.Contains(t, nsgroups, nmdata.NameServerGroup{
|
|
ID: "nsgroup-1",
|
|
PublicID: "nsgroup-1-public",
|
|
Name: "nsgroup-1",
|
|
Description: "nsgroup-1",
|
|
NameServers: []nmdata.NameServer{{IP: netip.MustParseAddr("192.168.31.2"), NSType: 1, Port: 53}},
|
|
Groups: []string{"group-one-resource-id"},
|
|
Domains: []string{"test-1.com"},
|
|
Primary: true,
|
|
SearchDomainsEnabled: false,
|
|
Enabled: true,
|
|
})
|
|
assert.Contains(t, nsgroups, nmdata.NameServerGroup{
|
|
ID: "nsgroup-2",
|
|
PublicID: "nsgroup-2-public",
|
|
Name: "nsgroup-2",
|
|
Description: "nsgroup-2",
|
|
NameServers: []nmdata.NameServer{{IP: netip.MustParseAddr("192.168.32.3"), NSType: 1, Port: 53}},
|
|
Groups: []string{"group-one-resource-id", "group-no-resources-id"},
|
|
Domains: []string{"test-1.com", "test-2.com"},
|
|
Primary: true,
|
|
SearchDomainsEnabled: false,
|
|
Enabled: true,
|
|
})
|
|
assert.Contains(t, nsgroups, nmdata.NameServerGroup{
|
|
ID: "nsgroup-3",
|
|
PublicID: "nsgroup-3-public",
|
|
Primary: false,
|
|
SearchDomainsEnabled: false,
|
|
Enabled: true,
|
|
})
|
|
}
|