long-agent
582ad7a069
- Add new test files for auth, service, and handler modules - Improve test organization and coverage - Refactor code for better maintainability - Add captcha, settings, stats, and theme handler tests - Add auth module tests (CAS, OAuth, password, SSO, state) - Add service layer tests for auth, export, permissions, roles - All Go tests pass (exit code 0) - All frontend tests pass (325 tests in 59 files)
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"sync/atomic"
|
||
"testing"
|
||
"time"
|
||
|
||
gormsqlite "gorm.io/driver/sqlite"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/logger"
|
||
_ "modernc.org/sqlite"
|
||
|
||
"github.com/user-management-system/internal/domain"
|
||
"github.com/user-management-system/internal/pagination"
|
||
)
|
||
|
||
var operationLogTestCounter int64
|
||
|
||
func openOperationLogTestDB(t *testing.T) *gorm.DB {
|
||
t.Helper()
|
||
|
||
id := atomic.AddInt64(&operationLogTestCounter, 1)
|
||
dsn := fmt.Sprintf("file:operationlogtestdb%d?mode=memory&cache=private", id)
|
||
|
||
db, err := gorm.Open(gormsqlite.New(gormsqlite.Config{
|
||
DriverName: "sqlite",
|
||
DSN: dsn,
|
||
}), &gorm.Config{
|
||
Logger: logger.Default.LogMode(logger.Silent),
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("打开测试数据库失败: %v", err)
|
||
}
|
||
|
||
if err := db.AutoMigrate(&domain.OperationLog{}); err != nil {
|
||
t.Fatalf("数据库迁移失败: %v", err)
|
||
}
|
||
return db
|
||
}
|
||
|
||
func setupOperationLogTestDB(t *testing.T) *gorm.DB {
|
||
return openOperationLogTestDB(t)
|
||
}
|
||
|
||
func TestOperationLogRepository_ListCursor(t *testing.T) {
|
||
db := setupOperationLogTestDB(t)
|
||
repo := NewOperationLogRepository(db)
|
||
ctx := context.Background()
|
||
|
||
now := time.Now()
|
||
for i := 0; i < 5; i++ {
|
||
repo.Create(ctx, &domain.OperationLog{
|
||
UserID: nil,
|
||
OperationType: "test",
|
||
OperationName: "测试操作" + string(rune('0'+i)),
|
||
RequestMethod: "GET",
|
||
RequestPath: "/api/test",
|
||
ResponseStatus: 200,
|
||
IP: "192.168.1." + string(rune('0'+i)),
|
||
CreatedAt: now.Add(-time.Duration(i) * time.Minute),
|
||
})
|
||
}
|
||
|
||
// 第一次查询，获取前3个
|
||
logs, hasMore, err := repo.ListCursor(ctx, 3, nil)
|
||
if err != nil {
|
||
t.Fatalf("ListCursor() error = %v", err)
|
||
}
|
||
if len(logs) != 3 {
|
||
t.Errorf("len(logs) = %d, want 3", len(logs))
|
||
}
|
||
if !hasMore {
|
||
t.Error("hasMore should be true when more logs exist")
|
||
}
|
||
|
||
// 使用游标继续查询
|
||
lastLog := logs[len(logs)-1]
|
||
cursor := &pagination.Cursor{
|
||
LastID: lastLog.ID,
|
||
LastValue: lastLog.CreatedAt,
|
||
}
|
||
logs2, hasMore2, err := repo.ListCursor(ctx, 3, cursor)
|
||
if err != nil {
|
||
t.Fatalf("ListCursor() error = %v", err)
|
||
}
|
||
if len(logs2) != 2 {
|
||
t.Errorf("len(logs2) = %d, want 2", len(logs2))
|
||
}
|
||
if hasMore2 {
|
||
t.Error("hasMore2 should be false")
|
||
}
|
||
}
|