zero/bootstrap: reset back to inmem databroker if connection string is empty (#4955)

internal/zero/bootstrap/source_test.go

@@ -0,0 +1,68 @@
+package bootstrap_test
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/pomerium/pomerium/config"
+	"github.com/pomerium/pomerium/internal/zero/bootstrap"
+	cluster_api "github.com/pomerium/pomerium/pkg/zero/cluster"
+)
+
+func TestConfigChanges(t *testing.T) {
+	t.Parallel()
+
+	secret := []byte("secret")
+
+	src, err := bootstrap.New(secret)
+	require.NoError(t, err)
+
+	ptr := func(s string) *string { return &s }
+
+	var listenerCalled bool
+	src.OnConfigChange(nil, func(_ context.Context, _ *config.Config) {
+		listenerCalled = true
+	})
+
+	for i, tc := range []struct {
+		bootstrap                        cluster_api.BootstrapConfig
+		expectChanged                    bool
+		expectDatabrokerType             string
+		expectDatabrokerConnectionString string
+	}{
+		{
+			cluster_api.BootstrapConfig{},
+			false,
+			config.StorageInMemoryName,
+			"",
+		},
+		{
+			cluster_api.BootstrapConfig{
+				DatabrokerStorageConnection: ptr("postgres://"),
+			},
+			true,
+			config.StoragePostgresName,
+			"postgres://",
+		},
+		{
+			cluster_api.BootstrapConfig{},
+			true,
+			config.StorageInMemoryName,
+			"",
+		},
+	} {
+		t.Run(fmt.Sprintf("test-%d", i), func(t *testing.T) {
+			listenerCalled = false
+			changed := src.UpdateBootstrap(context.Background(), tc.bootstrap)
+			cfg := src.GetConfig()
+			assert.Equal(t, tc.expectChanged, changed, "changed")
+			assert.Equal(t, tc.expectChanged, listenerCalled, "listenerCalled")
+			assert.Equal(t, tc.expectDatabrokerType, cfg.Options.DataBrokerStorageType, "databroker type")
+			assert.Equal(t, tc.expectDatabrokerConnectionString, cfg.Options.DataBrokerStorageConnectionString, "databroker connection string")
+		})
+	}
+}