From 452501463231d8495d7e1c894613b6f8cbbfa298 Mon Sep 17 00:00:00 2001 From: Dmitri Dolguikh Date: Wed, 29 Jul 2026 17:59:44 +0200 Subject: [PATCH] support for nameservergroups Signed-off-by: Dmitri Dolguikh --- .../internals/network_map_db/db_store.go | 1 + .../network_map_db/pgsql/group_test.go | 22 +++++++++++++++++++ 2 files changed, 23 insertions(+) diff --git a/management/internals/network_map_db/db_store.go b/management/internals/network_map_db/db_store.go index 7f1a26b8c..695db30f0 100644 --- a/management/internals/network_map_db/db_store.go +++ b/management/internals/network_map_db/db_store.go @@ -21,6 +21,7 @@ type NetworkMapDBStore interface { GetGroups(ctx context.Context, accountId string) ([]nmdata.Group, error) // TODO: join/populate peers GetPeers(ctx context.Context, accountId string) ([]nmdata.Peer, error) GetPolicies(ctx context.Context, accountId string) ([]nmdata.Policy, error) + GetRoutes(ctx context.Context, accountId string) ([]nmdata.Route, error) } type NetworkMapDBStoreImpl struct { diff --git a/management/internals/network_map_db/pgsql/group_test.go b/management/internals/network_map_db/pgsql/group_test.go index 58a73600c..517bafe66 100644 --- a/management/internals/network_map_db/pgsql/group_test.go +++ b/management/internals/network_map_db/pgsql/group_test.go @@ -105,6 +105,28 @@ func TestGetRoutes(t *testing.T) { // ) } +func TestGetNSGroups(t *testing.T) { + ctx := context.TODO() + + s, err := NewPostgresqlStore(ctx, "postgresql://root:netbird@localhost:5432/netbird") + assert.NoError(t, err) + // err = loadSQL(ctx, s.pool, initDb) + //assert.NoError(t, err) + + groups, err := s.GetNameServerGroups(ctx, "cl3h77qfic3c738mkja0") //"ckd7ee2fic3c73dtendg") + assert.NoError(t, err) + + fmt.Print(groups) + // assert.Contains(t, + // groups, + // nmdata.Group{Name: "test-group-1", PublicID: "public-id-1", Resources: []nmdata.Resource{{ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}}, + // ) + // assert.Contains(t, + // groups, + // nmdata.Group{Name: "All", PublicID: "d9aejspvcsu517nkh4a0", Resources: []nmdata.Resource{{ID: "cui7olrl0ubs73d8qpe0", Type: "subnet"}, {ID: "cui7q2jl0ubs73d8qpi0", Type: "host"}}}, + // ) +} + func loadSQL(ctx context.Context, pool *pgxpool.Pool, initdb string) error { queries := strings.Split(string(initdb), ";")