mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
ryan
f2ab94501c
- Relocated go.mod and go.sum into backend/ root directory - Stripped redundant backend/ segments from all Go imports (github.com/Rain-kl/Wavelet/...) - Unified Makefile, swagger, and build-test to execute in backend/ module context - Ensured 100% build-test, code-check, format, and swagger pass
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package driver_asynq_worker
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/pkg/config"
|
|
"github.com/redis/go-redis/v9/maintnotifications"
|
|
)
|
|
|
|
func TestMaintNotificationsConfig(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
enabled bool
|
|
wantMode maintnotifications.Mode
|
|
}{
|
|
{
|
|
name: "disabled mode when flag is false",
|
|
enabled: false,
|
|
wantMode: maintnotifications.ModeDisabled,
|
|
},
|
|
{
|
|
name: "auto mode when flag is true",
|
|
enabled: true,
|
|
wantMode: maintnotifications.ModeAuto,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
cfg := maintNotificationsConfig(tt.enabled)
|
|
if cfg == nil {
|
|
t.Fatalf("maintNotificationsConfig(%v) = nil, want non-nil", tt.enabled)
|
|
}
|
|
if cfg.Mode != tt.wantMode {
|
|
t.Fatalf("maintNotificationsConfig(%v).Mode = %v, want %v", tt.enabled, cfg.Mode, tt.wantMode)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPrefixedQueue(t *testing.T) {
|
|
oldPrefix := config.Config.Redis.KeyPrefix
|
|
defer func() {
|
|
config.Config.Redis.KeyPrefix = oldPrefix
|
|
}()
|
|
|
|
config.Config.Redis.KeyPrefix = "test:"
|
|
if got := PrefixedQueue("default"); got != "test:default" {
|
|
t.Fatalf("PrefixedQueue() = %q, want %q", got, "test:default")
|
|
}
|
|
|
|
config.Config.Redis.KeyPrefix = ""
|
|
if got := PrefixedQueue("default"); got != "default" {
|
|
t.Fatalf("PrefixedQueue() = %q, want %q", got, "default")
|
|
}
|
|
}
|