From 1bffee7190981affab9f1638b222b2f1b4548c79 Mon Sep 17 00:00:00 2001 From: Damian Debkowski Date: Thu, 5 Sep 2024 13:31:22 -0700 Subject: [PATCH] test(server): validate ipv6 address for upsert worker status --- internal/server/repository_worker_test.go | 66 +++++++++++++++++++++++ 1 file changed, 66 insertions(+) diff --git a/internal/server/repository_worker_test.go b/internal/server/repository_worker_test.go index 3ef5177524..1295535167 100644 --- a/internal/server/repository_worker_test.go +++ b/internal/server/repository_worker_test.go @@ -499,6 +499,72 @@ func TestUpsertWorkerStatus(t *testing.T) { require.NoError(t, err) assert.Len(t, workers, 2) }) + + t.Run("ipv4 address", func(t *testing.T) { + status := server.NewWorker(scope.Global.String(), + server.WithName("worker-with-ipv4-address"), + server.WithAddress("8.8.8.8"), + server.WithReleaseVersion("Boundary v0.11.0")) + _, err = repo.UpsertWorkerStatus(ctx, status) + require.NoError(t, err) + + workers, err := repo.ListWorkers(ctx, []string{scope.Global.String()}) + require.NoError(t, err) + require.NotEmpty(t, workers) + var actualWorker *server.Worker + for _, w := range workers { + if w.Name == "worker-with-ipv4-address" { + actualWorker = w + break + } + } + require.NotNil(t, actualWorker) + assert.Equal(t, actualWorker.Address, "8.8.8.8") + }) + + t.Run("ipv6 address", func(t *testing.T) { + status := server.NewWorker(scope.Global.String(), + server.WithName("worker-with-ipv6-address"), + server.WithAddress("2001:4860:4860:0:0:0:0:8888"), + server.WithReleaseVersion("Boundary v0.11.0")) + _, err = repo.UpsertWorkerStatus(ctx, status) + require.NoError(t, err) + + workers, err := repo.ListWorkers(ctx, []string{scope.Global.String()}) + require.NoError(t, err) + require.NotEmpty(t, workers) + var actualWorker *server.Worker + for _, w := range workers { + if w.Name == "worker-with-ipv6-address" { + actualWorker = w + break + } + } + require.NotNil(t, actualWorker) + assert.Equal(t, actualWorker.Address, "2001:4860:4860:0:0:0:0:8888") + }) + + t.Run("ipv6 abbreviated address", func(t *testing.T) { + status := server.NewWorker(scope.Global.String(), + server.WithName("worker-with-abbreviated-ipv6-address"), + server.WithAddress("2001:4860:4860::8888"), + server.WithReleaseVersion("Boundary v0.11.0")) + _, err = repo.UpsertWorkerStatus(ctx, status) + require.NoError(t, err) + + workers, err := repo.ListWorkers(ctx, []string{scope.Global.String()}) + require.NoError(t, err) + require.NotEmpty(t, workers) + var actualWorker *server.Worker + for _, w := range workers { + if w.Name == "worker-with-abbreviated-ipv6-address" { + actualWorker = w + break + } + } + require.NotNil(t, actualWorker) + assert.Equal(t, actualWorker.Address, "2001:4860:4860::8888") + }) } func TestTagUpdatingListing(t *testing.T) {