mtfyun/ragflow
1
0 Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-11-10 21:32:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: seed user rows in tenant service tests (#22060)

Browse Source
This commit is contained in:
buua436 authored and GitHub committed 2026-11-05 20:51:14 +08:00
1 parent 8e3be03a2b
commit cf34ed9b52
2 files changed
+7 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

internal/service/api_token_test.go
+3
View File
@@ -30,6 +30,9 @@ func TestCreateAPIKeyStoresUserAndTenantContext(t *testing.T) {
if err := db.AutoMigrate(&entity.APIToken{}, &entity.UserTenant{}); err != nil {
t.Fatalf("failed to migrate api_token and user_tenant: %v", err)
}
if err := db.Create(&entity.User{ID: "user-1", Email: "user-1@example.com", Nickname: "user-1"}).Error; err != nil {
t.Fatalf("failed to create API key owner: %v", err)
}
status := entity.TenantMembershipStatusJoined
if err := db.Create(&entity.UserTenant{ID: "membership-1", UserID: "user-1", TenantID: "tenant-1", Role: "owner", Status: &status}).Error; err != nil {
t.Fatalf("failed to create Tenant membership: %v", err)
internal/service/langfuse_service_test.go
+4 -1
View File
@@ -39,9 +39,12 @@ func setupLangfuseServiceTestDB(t *testing.T) *gorm.DB {
if err != nil {
t.Fatalf("failed to open sqlite: %v", err)
}
if err := db.AutoMigrate(&entity.TenantLangfuse{}, &entity.UserTenant{}); err != nil {
if err := db.AutoMigrate(&entity.TenantLangfuse{}, &entity.User{}, &entity.UserTenant{}); err != nil {
t.Fatalf("failed to migrate test schema: %v", err)
}
if err := db.Create(&entity.User{ID: "tenant-1", Email: "tenant-1@example.com", Nickname: "tenant-1"}).Error; err != nil {
t.Fatalf("create Tenant owner: %v", err)
}
active := string(entity.StatusValid)
if err := db.Create(&entity.UserTenant{
ID: "langfuse-membership", UserID: "tenant-1", TenantID: "tenant-1", Role: "owner", Status: &active,