mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-01 14:46:36 +08:00
ryan
299ac30ee4
- Purify core micro-kernel by removing context hardcoded helpers and reverse dependencies - Eliminate init() side effects in infra plugins with reversible lifecycle disposal - Completely isolate plugins by removing cross-plugin imports and using core/contracts - Introduce TaskService and RiskControlService contracts for unified cross-plugin APIs - Regenerate Swagger documentation and update developer guide matrix - Achieve 0 violations in check_cordis_architecture.sh and 100% test pass
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package message_gateway_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"Wavelet/pkg/testhelper"
|
|
"Wavelet/plugins/domain/message_gateway"
|
|
)
|
|
|
|
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)
|
|
}
|
|
}
|