mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
fb6a3edb89
- Physically purged all legacy internal/ packages, centralized pkg/model/ and pkg/repository/ - Migrated domain models and database repositories into self-contained owner plugins (user, auth, message_gateway, admin, upload, risk_control) - Decoupled cross-plugin interactions via pure core/contracts and typed EventBus - Ensured 100% test coverage pass, zero data races (-race clean), and 0 lint issues in make code-check
77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package task
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/pkg/config"
|
|
"github.com/redis/go-redis/v9"
|
|
"github.com/redis/go-redis/v9/maintnotifications"
|
|
)
|
|
|
|
func TestNewRedisConnOptConfiguresMaintenanceNotifications(t *testing.T) {
|
|
previous := config.Config.Redis
|
|
t.Cleanup(func() { config.Config.Redis = previous })
|
|
|
|
for _, test := range []struct {
|
|
name string
|
|
enabled bool
|
|
want maintnotifications.Mode
|
|
}{
|
|
{name: "disabled by default", enabled: false, want: maintnotifications.ModeDisabled},
|
|
{name: "auto when enabled", enabled: true, want: maintnotifications.ModeAuto},
|
|
} {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
config.Config.Redis.MaintNotifications = test.enabled
|
|
|
|
t.Run("standalone", func(t *testing.T) {
|
|
config.Config.Redis.ClusterMode = false
|
|
config.Config.Redis.MasterName = ""
|
|
config.Config.Redis.Addrs = []string{"127.0.0.1:6379"}
|
|
|
|
client, ok := NewRedisConnOpt().MakeRedisClient().(*redis.Client)
|
|
if !ok {
|
|
t.Fatal("standalone option did not create *redis.Client")
|
|
}
|
|
defer func() { _ = client.Close() }()
|
|
assertMaintenanceNotificationsMode(t, client.Options().MaintNotificationsConfig, test.want)
|
|
})
|
|
|
|
t.Run("cluster", func(t *testing.T) {
|
|
config.Config.Redis.ClusterMode = true
|
|
config.Config.Redis.MasterName = ""
|
|
config.Config.Redis.Addrs = []string{"127.0.0.1:6379"}
|
|
|
|
client, ok := NewRedisConnOpt().MakeRedisClient().(*redis.ClusterClient)
|
|
if !ok {
|
|
t.Fatal("cluster option did not create *redis.ClusterClient")
|
|
}
|
|
defer func() { _ = client.Close() }()
|
|
assertMaintenanceNotificationsMode(t, client.Options().MaintNotificationsConfig, test.want)
|
|
})
|
|
|
|
t.Run("sentinel", func(t *testing.T) {
|
|
config.Config.Redis.ClusterMode = false
|
|
config.Config.Redis.MasterName = "openflare"
|
|
config.Config.Redis.Addrs = []string{"127.0.0.1:26379"}
|
|
|
|
client, ok := NewRedisConnOpt().MakeRedisClient().(*redis.Client)
|
|
if !ok {
|
|
t.Fatal("sentinel option did not create *redis.Client")
|
|
}
|
|
defer func() { _ = client.Close() }()
|
|
assertMaintenanceNotificationsMode(t, client.Options().MaintNotificationsConfig, test.want)
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
func assertMaintenanceNotificationsMode(t *testing.T, cfg *maintnotifications.Config, want maintnotifications.Mode) {
|
|
t.Helper()
|
|
if cfg == nil || cfg.Mode != want {
|
|
t.Fatalf("maintenance notifications mode = %v, want %v", cfg, want)
|
|
}
|
|
}
|