mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
528240026d
- make format 现在与 code-check 使用同一格式化器（golangci-lint fmt），消除 goimports -local 与 gofumpt 的格式拉锯 - .golangci.yml 关闭默认 50/3 截断，完整上报所有问题（只增强不弱化） - 全库 gofumpt 规范化（203 files, 纯格式无行为变更）
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package message_gateway_test
|
|
|
|
import (
|
|
"Wavelet/pkg/testhelper"
|
|
"Wavelet/plugins/domain/message_gateway"
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type mockDBService struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func (m *mockDBService) GORM() *gorm.DB {
|
|
return m.db
|
|
}
|
|
|
|
func (m *mockDBService) DB(ctx context.Context) *gorm.DB {
|
|
return m.db.WithContext(ctx)
|
|
}
|
|
|
|
func (m *mockDBService) Named(_ string) *gorm.DB {
|
|
return m.db
|
|
}
|
|
|
|
func TestUpsertPairingCode_ReusesUnexpired(t *testing.T) {
|
|
testDB, _, cleanup := testhelper.SetupTestEnvironment(t)
|
|
message_gateway.SetDBServiceForTest(&mockDBService{db: testDB})
|
|
defer func() {
|
|
message_gateway.SetDBServiceForTest(nil)
|
|
cleanup()
|
|
}()
|
|
ctx := context.Background()
|
|
first, err := message_gateway.UpsertPairingCode(ctx, 1, "tg-1", "ABCD1234", time.Now().Add(15*time.Minute))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
second, err := message_gateway.UpsertPairingCode(ctx, 1, "tg-1", "ZZZZ9999", time.Now().Add(15*time.Minute))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if first.Code != second.Code || first.Code != "ABCD1234" {
|
|
t.Fatalf("reuse failed: %+v %+v", first, second)
|
|
}
|
|
}
|