mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-20 13:49:07 +02:00
fix linter comments
This commit is contained in:
@@ -12,15 +12,15 @@ import (
|
|||||||
func TestGetDomains(t *testing.T) {
|
func TestGetDomains(t *testing.T) {
|
||||||
ctx := context.TODO()
|
ctx := context.TODO()
|
||||||
|
|
||||||
_, err := pgstore.Pool.Query(ctx,
|
_, err := pgstore.Pool.Exec(ctx,
|
||||||
`insert into domains (id, account_id, domain, target_cluster)
|
`insert into domains (id, account_id, domain, target_cluster)
|
||||||
VALUES('domain-1','account-1','test-1.com','target-1.cluster.local')`)
|
VALUES('domain-1','account-1','test-1.com','target-1.cluster.local')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into domains (id, account_id, domain, target_cluster)
|
`insert into domains (id, account_id, domain, target_cluster)
|
||||||
VALUES('domain-2','account-1','test-2.com','target-2.cluster.local')`)
|
VALUES('domain-2','account-1','test-2.com','target-2.cluster.local')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into domains (id, account_id, domain, target_cluster)
|
`insert into domains (id, account_id, domain, target_cluster)
|
||||||
VALUES('domain-3','account-1',null,null)`)
|
VALUES('domain-3','account-1',null,null)`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|||||||
@@ -48,7 +48,7 @@ func TestGetGroupsWithoutExpectedFields(t *testing.T) {
|
|||||||
|
|
||||||
acctId := xid.New().String()
|
acctId := xid.New().String()
|
||||||
|
|
||||||
_, err = s.Pool.Query(ctx,
|
_, err = s.Pool.Exec(ctx,
|
||||||
"insert into accounts (id) VALUES($1)", acctId)
|
"insert into accounts (id) VALUES($1)", acctId)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
|||||||
@@ -13,15 +13,15 @@ import (
|
|||||||
func TestGetNameServerGroups(t *testing.T) {
|
func TestGetNameServerGroups(t *testing.T) {
|
||||||
ctx := context.TODO()
|
ctx := context.TODO()
|
||||||
|
|
||||||
_, err := pgstore.Pool.Query(ctx,
|
_, err := pgstore.Pool.Exec(ctx,
|
||||||
`insert into name_server_groups (id, public_id, name, description, name_servers, groups, domains, enabled, search_domains_enabled, "primary", account_id)
|
`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')`)
|
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')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into name_server_groups (id, public_id, name, description, name_servers, groups, domains, enabled, search_domains_enabled,"primary",account_id)
|
`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')`)
|
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')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into name_server_groups (id, public_id, name, description, name_servers, groups, domains, enabled, search_domains_enabled,"primary",account_id)
|
`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')`)
|
VALUES('nsgroup-3','nsgroup-3-public',null,null,null,null,null,TRUE,FALSE,FALSE,'account-1')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|||||||
@@ -16,15 +16,18 @@ func TestGetNetworkResources(t *testing.T) {
|
|||||||
s, err := networkmap_pgsql.NewPostgresqlStore(ctx, dsn)
|
s, err := networkmap_pgsql.NewPostgresqlStore(ctx, dsn)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
_, err = s.Pool.Query(ctx,
|
_, err = s.Pool.Exec(ctx,
|
||||||
`insert into network_resources (id, account_id, network_id, public_id, name, description, type, domain, prefix, enabled)
|
`insert into network_resources (id, account_id, network_id, public_id, name, description, type, domain, prefix, enabled)
|
||||||
VALUES('net-resource-1','account-1','network-1','net-resource-public-1','network-resource-1','network-resource-1','subnet','','"10.0.0.0/16"',TRUE)`)
|
VALUES('net-resource-1','account-1','network-1','net-resource-public-1','network-resource-1','network-resource-1','subnet','','"10.0.0.0/16"',TRUE)`)
|
||||||
_, err = s.Pool.Query(ctx,
|
assert.NoError(t, err)
|
||||||
|
_, err = s.Pool.Exec(ctx,
|
||||||
`insert into network_resources (id, account_id, network_id, public_id, name, description, type, domain, prefix, enabled)
|
`insert into network_resources (id, account_id, network_id, public_id, name, description, type, domain, prefix, enabled)
|
||||||
VALUES('net-resource-2','account-1','network-2','net-resource-public-2','network-resource-2','network-resource-2','domain','test.com','',TRUE)`)
|
VALUES('net-resource-2','account-1','network-2','net-resource-public-2','network-resource-2','network-resource-2','domain','test.com','',TRUE)`)
|
||||||
_, err = s.Pool.Query(ctx,
|
assert.NoError(t, err)
|
||||||
|
_, err = s.Pool.Exec(ctx,
|
||||||
`insert into network_resources (id, account_id, network_id, public_id, name, description, type, domain, prefix, enabled)
|
`insert into network_resources (id, account_id, network_id, public_id, name, description, type, domain, prefix, enabled)
|
||||||
VALUES('net-resource-3','account-1','network-3','net-resource-public-3','network-resource-3','network-resource-3','host','','"10.0.0.1/32"',TRUE)`)
|
VALUES('net-resource-3','account-1','network-3','net-resource-public-3','network-resource-3','network-resource-3','host','','"10.0.0.1/32"',TRUE)`)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
resources, err := s.GetNetworkResources(ctx, "account-1")
|
resources, err := s.GetNetworkResources(ctx, "account-1")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|||||||
@@ -15,12 +15,14 @@ func TestGetNetworkRouters(t *testing.T) {
|
|||||||
s, err := networkmap_pgsql.NewPostgresqlStore(ctx, dsn)
|
s, err := networkmap_pgsql.NewPostgresqlStore(ctx, dsn)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
_, err = s.Pool.Query(ctx,
|
_, err = s.Pool.Exec(ctx,
|
||||||
`insert into network_routers (id, account_id, public_id, peer, network_id, masquerade, metric, enabled, peer_groups)
|
`insert into network_routers (id, account_id, public_id, peer, network_id, masquerade, metric, enabled, peer_groups)
|
||||||
VALUES('test-nr-id-1','account-1','public-id-1','peer-id-1','network-id-1',TRUE,999,TRUE,'["group-one-resource-id"]')`)
|
VALUES('test-nr-id-1','account-1','public-id-1','peer-id-1','network-id-1',TRUE,999,TRUE,'["group-one-resource-id"]')`)
|
||||||
_, err = s.Pool.Query(ctx,
|
assert.NoError(t, err)
|
||||||
|
_, err = s.Pool.Exec(ctx,
|
||||||
`insert into network_routers (id, account_id, public_id, peer, network_id, masquerade, metric, enabled, peer_groups)
|
`insert into network_routers (id, account_id, public_id, peer, network_id, masquerade, metric, enabled, peer_groups)
|
||||||
VALUES('test-nr-id-2','account-1','public-id-2','','network-id-2',TRUE,333,TRUE,'["group-two-resources-id","group-no-resources-id"]')`)
|
VALUES('test-nr-id-2','account-1','public-id-2','','network-id-2',TRUE,333,TRUE,'["group-two-resources-id","group-no-resources-id"]')`)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
routers, err := s.GetNetworkRouters(ctx, "account-1")
|
routers, err := s.GetNetworkRouters(ctx, "account-1")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|||||||
@@ -11,10 +11,10 @@ import (
|
|||||||
func TestGetNetworks(t *testing.T) {
|
func TestGetNetworks(t *testing.T) {
|
||||||
ctx := context.TODO()
|
ctx := context.TODO()
|
||||||
|
|
||||||
_, err := pgstore.Pool.Query(ctx,
|
_, err := pgstore.Pool.Exec(ctx,
|
||||||
`insert into networks (id, account_id, public_id) VALUES('network-1','account-1','network-1-public')`)
|
`insert into networks (id, account_id, public_id) VALUES('network-1','account-1','network-1-public')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into networks (id, account_id, public_id) VALUES('network-2','account-1','network-2-public')`)
|
`insert into networks (id, account_id, public_id) VALUES('network-2','account-1','network-2-public')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ import (
|
|||||||
func TestGetPostureChecks(t *testing.T) {
|
func TestGetPostureChecks(t *testing.T) {
|
||||||
ctx := context.TODO()
|
ctx := context.TODO()
|
||||||
|
|
||||||
_, err := pgstore.Pool.Query(ctx,
|
_, err := pgstore.Pool.Exec(ctx,
|
||||||
`insert into posture_checks (id, account_id, public_id, checks)
|
`insert into posture_checks (id, account_id, public_id, checks)
|
||||||
VALUES('posturecheck-1','account-1','posturecheck-1-public',
|
VALUES('posturecheck-1','account-1','posturecheck-1-public',
|
||||||
'{"NBVersionCheck":{"MinVersion":"0.25.0"},
|
'{"NBVersionCheck":{"MinVersion":"0.25.0"},
|
||||||
@@ -22,7 +22,7 @@ func TestGetPostureChecks(t *testing.T) {
|
|||||||
"PeerNetworkRangeCheck":{"Action":"deny","Ranges":["192.168.0.1/24"]}}')`)
|
"PeerNetworkRangeCheck":{"Action":"deny","Ranges":["192.168.0.1/24"]}}')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into posture_checks (id, account_id, public_id, checks)
|
`insert into posture_checks (id, account_id, public_id, checks)
|
||||||
VALUES('posturecheck-2','account-1','posturecheck-2-public',
|
VALUES('posturecheck-2','account-1','posturecheck-2-public',
|
||||||
'{"NBVersionCheck":{"MinVersion":"0.25.0"},
|
'{"NBVersionCheck":{"MinVersion":"0.25.0"},
|
||||||
@@ -30,7 +30,7 @@ func TestGetPostureChecks(t *testing.T) {
|
|||||||
"GeoLocationCheck":{"Locations":[{"CountryCode":"US","CityName":"Harker Heights"}],"Action":"allow"},
|
"GeoLocationCheck":{"Locations":[{"CountryCode":"US","CityName":"Harker Heights"}],"Action":"allow"},
|
||||||
"PeerNetworkRangeCheck":{"Action":"allow","Ranges":["0.0.0.0/0"]}}')`)
|
"PeerNetworkRangeCheck":{"Action":"allow","Ranges":["0.0.0.0/0"]}}')`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into posture_checks (id, account_id, public_id, checks)
|
`insert into posture_checks (id, account_id, public_id, checks)
|
||||||
VALUES('posturecheck-3','account-1','posturecheck-3-public', null)`)
|
VALUES('posturecheck-3','account-1','posturecheck-3-public', null)`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ import (
|
|||||||
func TestGetRoutes(t *testing.T) {
|
func TestGetRoutes(t *testing.T) {
|
||||||
ctx := context.TODO()
|
ctx := context.TODO()
|
||||||
|
|
||||||
_, err := pgstore.Pool.Query(ctx,
|
_, err := pgstore.Pool.Exec(ctx,
|
||||||
`insert into routes (id, account_id, public_id, network, domains, keep_route, net_id, description,
|
`insert into routes (id, account_id, public_id, network, domains, keep_route, net_id, description,
|
||||||
peer, peer_groups, network_type, masquerade, metric, enabled,
|
peer, peer_groups, network_type, masquerade, metric, enabled,
|
||||||
groups, access_control_groups, skip_auto_apply)
|
groups, access_control_groups, skip_auto_apply)
|
||||||
@@ -22,7 +22,7 @@ func TestGetRoutes(t *testing.T) {
|
|||||||
'peer-id-1','["group-one-resource-id"]',1,true,9999,true,
|
'peer-id-1','["group-one-resource-id"]',1,true,9999,true,
|
||||||
'["group-one-resource-id"]','["group-one-resource-id"]',false)`)
|
'["group-one-resource-id"]','["group-one-resource-id"]',false)`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into routes (id, account_id, public_id, network, domains, keep_route, net_id, description,
|
`insert into routes (id, account_id, public_id, network, domains, keep_route, net_id, description,
|
||||||
peer, peer_groups, network_type, masquerade, metric, enabled,
|
peer, peer_groups, network_type, masquerade, metric, enabled,
|
||||||
groups, access_control_groups, skip_auto_apply)
|
groups, access_control_groups, skip_auto_apply)
|
||||||
@@ -30,7 +30,7 @@ func TestGetRoutes(t *testing.T) {
|
|||||||
'peer-id-2','["group-two-resources-id"]',1,true,9999,true,
|
'peer-id-2','["group-two-resources-id"]',1,true,9999,true,
|
||||||
'["group-two-resources-id"]','["group-two-resources-id"]',false)`)
|
'["group-two-resources-id"]','["group-two-resources-id"]',false)`)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_, err = pgstore.Pool.Query(ctx,
|
_, err = pgstore.Pool.Exec(ctx,
|
||||||
`insert into routes (id, account_id, public_id, network, domains, keep_route, net_id, description,
|
`insert into routes (id, account_id, public_id, network, domains, keep_route, net_id, description,
|
||||||
peer, peer_groups, network_type, masquerade, metric, enabled,
|
peer, peer_groups, network_type, masquerade, metric, enabled,
|
||||||
groups, access_control_groups, skip_auto_apply)
|
groups, access_control_groups, skip_auto_apply)
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ const (
|
|||||||
NMAP_MAP_TO = "map_to"
|
NMAP_MAP_TO = "map_to"
|
||||||
)
|
)
|
||||||
|
|
||||||
type NetworkMapDBStore interface {
|
type NetworkMapDBStore interface { //nolint:revive // established name across the codebase
|
||||||
GetGroups(ctx context.Context, accountId string) ([]nmdata.Group, map[string]map[string]any, error)
|
GetGroups(ctx context.Context, accountId string) ([]nmdata.Group, map[string]map[string]any, error)
|
||||||
GetPeers(ctx context.Context, accountId string) ([]nmdata.Peer, map[string][]*nmdata.Peer, error)
|
GetPeers(ctx context.Context, accountId string) ([]nmdata.Peer, map[string][]*nmdata.Peer, error)
|
||||||
GetPolicies(ctx context.Context, accountId string) ([]nmdata.Policy, map[string]map[string]any, map[string]map[string]any, error)
|
GetPolicies(ctx context.Context, accountId string) ([]nmdata.Policy, map[string]map[string]any, map[string]map[string]any, error)
|
||||||
@@ -40,7 +40,7 @@ type NetworkMapDBStore interface {
|
|||||||
GetDnsSettings(ctx context.Context, accountId string) (nmdata.DNSSettings, error)
|
GetDnsSettings(ctx context.Context, accountId string) (nmdata.DNSSettings, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
type NetworkMapDBStoreImpl struct {
|
type NetworkMapDBStoreImpl struct { //nolint:revive // established name across the codebase
|
||||||
store NetworkMapDBStore
|
store NetworkMapDBStore
|
||||||
integratedPeerValidator integrated_validator.IntegratedValidator
|
integratedPeerValidator integrated_validator.IntegratedValidator
|
||||||
extraSettingsManager settings.Manager
|
extraSettingsManager settings.Manager
|
||||||
@@ -138,7 +138,9 @@ func FromSqlTypesToSharedTypes(src reflect.Value, dst reflect.Value) error {
|
|||||||
}
|
}
|
||||||
case "json.RawMessage":
|
case "json.RawMessage":
|
||||||
s := srcField.Interface().(json.RawMessage)
|
s := srcField.Interface().(json.RawMessage)
|
||||||
json.Unmarshal(s, dstField.Addr().Interface())
|
if err := json.Unmarshal(s, dstField.Addr().Interface()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
case "[]string":
|
case "[]string":
|
||||||
if srcField.IsNil() {
|
if srcField.IsNil() {
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ import (
|
|||||||
"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
|
"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
|
||||||
)
|
)
|
||||||
|
|
||||||
var DnsUnsupportedRecordTypeError = errors.New("unsupported record type")
|
var ErrDnsUnsupportedRecordType = errors.New("unsupported record type")
|
||||||
|
|
||||||
const (
|
const (
|
||||||
GetAccountZonesQuery = `
|
GetAccountZonesQuery = `
|
||||||
@@ -73,7 +73,7 @@ func GetAppliedZoneCandidatesViaPgxConnection(ctx context.Context, conn *pgx.Con
|
|||||||
|
|
||||||
rtype, rdata, err := recordTypeAndRdata(z.RecordType.String, z.RecordRData.String)
|
rtype, rdata, err := recordTypeAndRdata(z.RecordType.String, z.RecordRData.String)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, DnsUnsupportedRecordTypeError) {
|
if errors.Is(err, ErrDnsUnsupportedRecordType) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
@@ -112,7 +112,7 @@ func recordTypeAndRdata(t, rdata string) (int, string, error) {
|
|||||||
case "CNAME":
|
case "CNAME":
|
||||||
return int(dns.TypeCNAME), dns.Fqdn(rdata), nil
|
return int(dns.TypeCNAME), dns.Fqdn(rdata), nil
|
||||||
default:
|
default:
|
||||||
return 0, "", fmt.Errorf("record type: %s %w", t, DnsUnsupportedRecordTypeError)
|
return 0, "", fmt.Errorf("record type: %s %w", t, ErrDnsUnsupportedRecordType)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -18,7 +18,7 @@ func TestRecordTypeAndRdata(t *testing.T) {
|
|||||||
{recordType: "AAAA", expectedRecordType: 28, rdata: "test.com", expectedRdata: "test.com", expectedErr: nil},
|
{recordType: "AAAA", expectedRecordType: 28, rdata: "test.com", expectedRdata: "test.com", expectedErr: nil},
|
||||||
{recordType: "CNAME", expectedRecordType: 5, rdata: "test.com", expectedRdata: "test.com.", expectedErr: nil},
|
{recordType: "CNAME", expectedRecordType: 5, rdata: "test.com", expectedRdata: "test.com.", expectedErr: nil},
|
||||||
{recordType: "CNAME", expectedRecordType: 5, rdata: "test.com.", expectedRdata: "test.com.", expectedErr: nil},
|
{recordType: "CNAME", expectedRecordType: 5, rdata: "test.com.", expectedRdata: "test.com.", expectedErr: nil},
|
||||||
{recordType: "TypeMX", expectedErr: DnsUnsupportedRecordTypeError},
|
{recordType: "TypeMX", expectedErr: ErrDnsUnsupportedRecordType},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@@ -26,7 +26,7 @@ func TestRecordTypeAndRdata(t *testing.T) {
|
|||||||
recordType, rdata, err := recordTypeAndRdata(tt.recordType, tt.rdata)
|
recordType, rdata, err := recordTypeAndRdata(tt.recordType, tt.rdata)
|
||||||
|
|
||||||
if tt.expectedErr != nil {
|
if tt.expectedErr != nil {
|
||||||
assert.ErrorIs(t, err, DnsUnsupportedRecordTypeError)
|
assert.ErrorIs(t, err, ErrDnsUnsupportedRecordType)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ package networkmap_pgsql
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -20,7 +19,7 @@ func TestGetGroups(t *testing.T) {
|
|||||||
// err = loadSQL(ctx, s.pool, initDb)
|
// err = loadSQL(ctx, s.pool, initDb)
|
||||||
//assert.NoError(t, err)
|
//assert.NoError(t, err)
|
||||||
|
|
||||||
_, err = s.Pool.Query(ctx, "insert into groups (id, account_id, name, resources, public_id) VALUES('test-group-id-1','ck7bnf2t2r9s739pkug0','test-group-1', '[{\"ID\":\"cui7q2jl0ubs73d8qpi0\",\"Type\":\"host\"}]','public-id-1')")
|
_, err = s.Pool.Exec(ctx, "insert into groups (id, account_id, name, resources, public_id) VALUES('test-group-id-1','ck7bnf2t2r9s739pkug0','test-group-1', '[{\"ID\":\"cui7q2jl0ubs73d8qpi0\",\"Type\":\"host\"}]','public-id-1')")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
groups, _, err := s.GetGroups(ctx, "ck7bnf2t2r9s739pkug0") //"ckd7ee2fic3c73dtendg")
|
groups, _, err := s.GetGroups(ctx, "ck7bnf2t2r9s739pkug0") //"ckd7ee2fic3c73dtendg")
|
||||||
@@ -47,7 +46,7 @@ func TestGetPeers(t *testing.T) {
|
|||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.NotEmpty(t, clusterToPeerIdx)
|
assert.NotEmpty(t, clusterToPeerIdx)
|
||||||
|
|
||||||
fmt.Print(peers)
|
t.Log(peers)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -71,7 +70,7 @@ func TestGetPolicies(t *testing.T) {
|
|||||||
assert.NotEmpty(t, idx1)
|
assert.NotEmpty(t, idx1)
|
||||||
assert.NotEmpty(t, idx2)
|
assert.NotEmpty(t, idx2)
|
||||||
|
|
||||||
fmt.Print(peers)
|
t.Log(peers)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -93,7 +92,7 @@ func TestGetRoutes(t *testing.T) {
|
|||||||
peers, err := s.GetRoutes(ctx, "csg5iabl0ubs7398nf1g") //"ckd7ee2fic3c73dtendg")
|
peers, err := s.GetRoutes(ctx, "csg5iabl0ubs7398nf1g") //"ckd7ee2fic3c73dtendg")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
fmt.Print(peers)
|
t.Log(peers)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -115,7 +114,7 @@ func TestGetNSGroups(t *testing.T) {
|
|||||||
groups, err := s.GetNameServerGroups(ctx, "cl3h77qfic3c738mkja0") //"ckd7ee2fic3c73dtendg")
|
groups, err := s.GetNameServerGroups(ctx, "cl3h77qfic3c738mkja0") //"ckd7ee2fic3c73dtendg")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
fmt.Print(groups)
|
t.Log(groups)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -137,7 +136,7 @@ func TestGetNetworkResources(t *testing.T) {
|
|||||||
res, err := s.GetNetworkResources(ctx, "cag86v2t2r9s73d0416g") //"ckd7ee2fic3c73dtendg")
|
res, err := s.GetNetworkResources(ctx, "cag86v2t2r9s73d0416g") //"ckd7ee2fic3c73dtendg")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
fmt.Print(res)
|
t.Log(res)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -160,7 +159,7 @@ func TestGetNetworkRouters(t *testing.T) {
|
|||||||
res, err := GetNetworkRoutersViaPgxConnection(ctx, c.Conn(), "d29f99jl0ubs73cm8ce0") //"ckd7ee2fic3c73dtendg")
|
res, err := GetNetworkRoutersViaPgxConnection(ctx, c.Conn(), "d29f99jl0ubs73cm8ce0") //"ckd7ee2fic3c73dtendg")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
fmt.Print(res)
|
t.Log(res)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -182,7 +181,7 @@ func TestGetNetwork(t *testing.T) {
|
|||||||
n, err := s.GetNetwork(ctx, "d29f99jl0ubs73cm8ce0") //"ckd7ee2fic3c73dtendg")
|
n, err := s.GetNetwork(ctx, "d29f99jl0ubs73cm8ce0") //"ckd7ee2fic3c73dtendg")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
fmt.Print(n)
|
t.Log(n)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -204,7 +203,7 @@ func TestGetAccountZones(t *testing.T) {
|
|||||||
zones, err := s.GetAppliedZoneCandidates(ctx, "d4g66rjl0ubs73b2q3b0") //"ckd7ee2fic3c73dtendg")
|
zones, err := s.GetAppliedZoneCandidates(ctx, "d4g66rjl0ubs73b2q3b0") //"ckd7ee2fic3c73dtendg")
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
fmt.Print(zones)
|
t.Log(zones)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
@@ -245,7 +244,7 @@ func TestGetPostureChecks(t *testing.T) {
|
|||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.NotEmpty(t, idx)
|
assert.NotEmpty(t, idx)
|
||||||
|
|
||||||
fmt.Print(checks)
|
t.Log(checks)
|
||||||
// assert.Contains(t,
|
// assert.Contains(t,
|
||||||
// groups,
|
// groups,
|
||||||
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
// nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}},
|
||||||
|
|||||||
@@ -7,6 +7,8 @@ import (
|
|||||||
|
|
||||||
"github.com/jackc/pgx/v5"
|
"github.com/jackc/pgx/v5"
|
||||||
"github.com/miekg/dns"
|
"github.com/miekg/dns"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
|
||||||
"github.com/netbirdio/netbird/shared/management/networkmap"
|
"github.com/netbirdio/netbird/shared/management/networkmap"
|
||||||
"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
|
"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
|
||||||
)
|
)
|
||||||
@@ -111,9 +113,8 @@ func (pg *PgStore) GetNetworkMapData(ctx context.Context, accountId string) (*ne
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
err = tx.Commit(ctx)
|
if err = tx.Commit(ctx); err != nil {
|
||||||
if err != nil {
|
log.WithContext(ctx).Warnf("failed to commit network map read transaction: %v", err)
|
||||||
// TODO log and ignore?
|
|
||||||
}
|
}
|
||||||
|
|
||||||
toret := networkmap.NetworkMapData{
|
toret := networkmap.NetworkMapData{
|
||||||
@@ -143,7 +144,7 @@ func (pg *PgStore) GetNetworkMapData(ctx context.Context, accountId string) (*ne
|
|||||||
|
|
||||||
func rollbackAndReturnError(ctx context.Context, tx pgx.Tx, err error) (*networkmap.NetworkMapData, error) {
|
func rollbackAndReturnError(ctx context.Context, tx pgx.Tx, err error) (*networkmap.NetworkMapData, error) {
|
||||||
if errr := tx.Rollback(ctx); errr != nil {
|
if errr := tx.Rollback(ctx); errr != nil {
|
||||||
// TODO log and ignore?
|
log.WithContext(ctx).Warnf("failed to rollback network map read transaction: %v", errr)
|
||||||
}
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -31,7 +31,6 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
defaultTTL = 300
|
|
||||||
// privateServiceDNSRecordTTL is short so proxy-peer changes propagate quickly to clients.
|
// privateServiceDNSRecordTTL is short so proxy-peer changes propagate quickly to clients.
|
||||||
privateServiceDNSRecordTTL = 5
|
privateServiceDNSRecordTTL = 5
|
||||||
DefaultPeerLoginExpiration = 24 * time.Hour
|
DefaultPeerLoginExpiration = 24 * time.Hour
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ import (
|
|||||||
// fresh nmdata twin types. A builder converts an Account into a NetworkMapData
|
// fresh nmdata twin types. A builder converts an Account into a NetworkMapData
|
||||||
// once per account; the per-peer components calculation then runs on this twin
|
// once per account; the per-peer components calculation then runs on this twin
|
||||||
// with no reference back to the Account.
|
// with no reference back to the Account.
|
||||||
type NetworkMapData struct {
|
type NetworkMapData struct { //nolint:revive // established name across the codebase
|
||||||
Peers map[string]*nmdata.Peer
|
Peers map[string]*nmdata.Peer
|
||||||
Groups map[string]*nmdata.Group
|
Groups map[string]*nmdata.Group
|
||||||
Policies []*nmdata.Policy
|
Policies []*nmdata.Policy
|
||||||
|
|||||||
Reference in New Issue
Block a user