WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

internal/application/repository/tenant_test.go

@@ -0,0 +1,57 @@
+package repository
+
+import (
+	"context"
+	"testing"
+
+	"github.com/Tencent/WeKnora/internal/types"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+)
+
+// setupTestDB creates an in-memory SQLite database with tenant table.
+func setupTestDB(t *testing.T) *gorm.DB {
+	t.Helper()
+	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
+	require.NoError(t, err)
+	require.NoError(t, db.AutoMigrate(&types.Tenant{}, &types.TenantMember{}))
+	return db
+}
+
+func TestDeleteTenant_SoftDeletesMemberships(t *testing.T) {
+	db := setupTestDB(t)
+	ctx := context.Background()
+	repo := NewTenantRepository(db)
+
+	tenant := &types.Tenant{Name: "gone", Status: "active"}
+	require.NoError(t, db.Create(tenant).Error)
+
+	member := &types.TenantMember{
+		UserID:   "user-1",
+		TenantID: tenant.ID,
+		Role:     types.TenantRoleOwner,
+		Status:   types.TenantMemberStatusActive,
+	}
+	require.NoError(t, db.Create(member).Error)
+
+	require.NoError(t, repo.DeleteTenant(ctx, tenant.ID))
+
+	var tenantCount int64
+	require.NoError(t, db.Model(&types.Tenant{}).Count(&tenantCount).Error)
+	assert.Equal(t, int64(0), tenantCount)
+
+	var memberCount int64
+	require.NoError(t, db.Model(&types.TenantMember{}).Count(&memberCount).Error)
+	assert.Equal(t, int64(0), memberCount)
+
+	// Unscoped: rows still exist but are soft-deleted.
+	var rawTenantCount int64
+	require.NoError(t, db.Unscoped().Model(&types.Tenant{}).Count(&rawTenantCount).Error)
+	assert.Equal(t, int64(1), rawTenantCount)
+
+	var rawMemberCount int64
+	require.NoError(t, db.Unscoped().Model(&types.TenantMember{}).Count(&rawMemberCount).Error)
+	assert.Equal(t, int64(1), rawMemberCount)
+}