snapshot of https://github.com/centrifugal/centrifugo.git (master, 2026-09-16; blobs > 131072B omitted)

internal/config/shared_poll_validate_test.go

@@ -0,0 +1,240 @@
+package config
+
+import (
+	"testing"
+	"time"
+
+	"github.com/centrifugal/centrifugo/v6/internal/configtypes"
+
+	"github.com/stretchr/testify/require"
+)
+
+func sharedPollDefaultProxy() configtypes.Proxy {
+	return configtypes.Proxy{
+		Endpoint: "http://localhost:3001/refresh",
+		Timeout:  configtypes.Duration(5 * time.Second),
+	}
+}
+
+func TestSharedPollConfig_Validation_MissingSecret(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+	cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+		{
+			Name: "poll",
+			ChannelOptions: configtypes.ChannelOptions{
+				SubscriptionType: "shared_poll",
+			},
+		},
+	}
+	err := cfg.Validate()
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "hmac_secret_key")
+}
+
+func TestSharedPollConfig_Validation_WithSecret(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.SharedPoll.HMACSecretKey = "my-secret"
+	cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+	cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+		{
+			Name: "poll",
+			ChannelOptions: configtypes.ChannelOptions{
+				SubscriptionType: "shared_poll",
+			},
+		},
+	}
+	err := cfg.Validate()
+	require.NoError(t, err)
+}
+
+func TestSharedPollConfig_Validation_InvalidProxyName(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.SharedPoll.HMACSecretKey = "my-secret"
+	cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+		{
+			Name: "poll",
+			ChannelOptions: configtypes.ChannelOptions{
+				SubscriptionType: "shared_poll",
+				SharedPoll: configtypes.SharedPollConfig{
+					ProxyName: "nonexistent_proxy",
+				},
+			},
+		},
+	}
+	err := cfg.Validate()
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "shared poll proxy")
+	require.Contains(t, err.Error(), "nonexistent_proxy")
+}
+
+func TestSharedPollConfig_Validation_ValidProxyName(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.SharedPoll.HMACSecretKey = "my-secret"
+	cfg.Proxies = []configtypes.NamedProxy{
+		{
+			Name: "poll_backend",
+			Proxy: configtypes.Proxy{
+				Endpoint: "http://localhost:3001/refresh",
+				Timeout:  configtypes.Duration(5 * time.Second),
+			},
+		},
+	}
+	cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+		{
+			Name: "poll",
+			ChannelOptions: configtypes.ChannelOptions{
+				SubscriptionType: "shared_poll",
+				SharedPoll: configtypes.SharedPollConfig{
+					ProxyName: "poll_backend",
+				},
+			},
+		},
+	}
+	err := cfg.Validate()
+	require.NoError(t, err)
+}
+
+func TestSharedPollConfig_SubscriptionTypeValid(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.SharedPoll.HMACSecretKey = "secret"
+	cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+	cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+		{
+			Name: "poll",
+			ChannelOptions: configtypes.ChannelOptions{
+				SubscriptionType: "shared_poll",
+			},
+		},
+	}
+	err := cfg.Validate()
+	require.NoError(t, err)
+}
+
+func TestSharedPollConfig_UnknownSubscriptionType(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.Channel.WithoutNamespace.SubscriptionType = "unknown_type"
+	err := cfg.Validate()
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "unknown subscription_type")
+}
+
+func TestSharedPollConfig_WithoutNamespaceValidation(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.SharedPoll.HMACSecretKey = "secret"
+	cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+	cfg.Channel.WithoutNamespace.SubscriptionType = "shared_poll"
+	err := cfg.Validate()
+	require.NoError(t, err)
+}
+
+func TestSharedPollConfig_WithoutNamespaceMissingSecret(t *testing.T) {
+	cfg := DefaultConfig()
+	cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+	cfg.Channel.WithoutNamespace.SubscriptionType = "shared_poll"
+	err := cfg.Validate()
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "hmac_secret_key")
+}
+
+func TestSharedPollConfig_VersionlessPublishEnabled(t *testing.T) {
+	t.Run("empty_mode_publish_enabled_rejected", func(t *testing.T) {
+		cfg := DefaultConfig()
+		cfg.SharedPoll.HMACSecretKey = "secret"
+		cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+		cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+			{
+				Name: "poll",
+				ChannelOptions: configtypes.ChannelOptions{
+					SubscriptionType: "shared_poll",
+					SharedPoll: configtypes.SharedPollConfig{
+						Mode:    "",
+						PublishEnabled: true,
+					},
+				},
+			},
+		}
+		err := cfg.Validate()
+		require.Error(t, err)
+		require.Contains(t, err.Error(), "publish_enabled is incompatible with versionless")
+	})
+
+	t.Run("versionless_mode_publish_enabled_rejected", func(t *testing.T) {
+		cfg := DefaultConfig()
+		cfg.SharedPoll.HMACSecretKey = "secret"
+		cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+		cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+			{
+				Name: "poll",
+				ChannelOptions: configtypes.ChannelOptions{
+					SubscriptionType: "shared_poll",
+					SharedPoll: configtypes.SharedPollConfig{
+						Mode:    "versionless",
+						PublishEnabled: true,
+					},
+				},
+			},
+		}
+		err := cfg.Validate()
+		require.Error(t, err)
+		require.Contains(t, err.Error(), "publish_enabled is incompatible with versionless")
+	})
+
+	t.Run("full_mode_publish_enabled_accepted", func(t *testing.T) {
+		cfg := DefaultConfig()
+		cfg.SharedPoll.HMACSecretKey = "secret"
+		cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+		cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+			{
+				Name: "poll",
+				ChannelOptions: configtypes.ChannelOptions{
+					SubscriptionType: "shared_poll",
+					SharedPoll: configtypes.SharedPollConfig{
+						Mode:    "versioned",
+						PublishEnabled: true,
+					},
+				},
+			},
+		}
+		err := cfg.Validate()
+		require.NoError(t, err)
+	})
+
+	t.Run("versionless_no_publish_accepted", func(t *testing.T) {
+		cfg := DefaultConfig()
+		cfg.SharedPoll.HMACSecretKey = "secret"
+		cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+		cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+			{
+				Name: "poll",
+				ChannelOptions: configtypes.ChannelOptions{
+					SubscriptionType: "shared_poll",
+					SharedPoll: configtypes.SharedPollConfig{
+						Mode: "versionless",
+					},
+				},
+			},
+		}
+		err := cfg.Validate()
+		require.NoError(t, err)
+	})
+
+	t.Run("versionless_string_accepted", func(t *testing.T) {
+		cfg := DefaultConfig()
+		cfg.SharedPoll.HMACSecretKey = "secret"
+		cfg.Channel.Proxy.SharedPollRefresh = sharedPollDefaultProxy()
+		cfg.Channel.Namespaces = []configtypes.ChannelNamespace{
+			{
+				Name: "poll",
+				ChannelOptions: configtypes.ChannelOptions{
+					SubscriptionType: "shared_poll",
+					SharedPoll: configtypes.SharedPollConfig{
+						Mode: "versionless",
+					},
+				},
+			},
+		}
+		err := cfg.Validate()
+		require.NoError(t, err)
+	})
+}