diff --git a/server/internal/db/sessions_test.go b/server/internal/db/sessions_test.go index 208d1ae..6fd52c0 100644 --- a/server/internal/db/sessions_test.go +++ b/server/internal/db/sessions_test.go @@ -29,6 +29,7 @@ func TestSessionOperations(t *testing.T) { DisplayName: "Test User", PasswordHash: "hash", Role: "editor", + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) diff --git a/server/internal/db/struct_query_test.go b/server/internal/db/struct_query_test.go index 71e2c64..d4d4be8 100644 --- a/server/internal/db/struct_query_test.go +++ b/server/internal/db/struct_query_test.go @@ -156,6 +156,7 @@ func TestStructQueries(t *testing.T) { DisplayName: "Struct Query Test", PasswordHash: "hashed_password", Role: models.RoleEditor, + Theme: "dark", } t.Run("InsertStructQuery", func(t *testing.T) { @@ -243,6 +244,7 @@ func TestStructQueries(t *testing.T) { DisplayName: "Struct Query Test 2", PasswordHash: "hashed_password2", Role: models.RoleViewer, + Theme: "light", } createdUser2, err := database.CreateUser(secondUser) @@ -437,6 +439,7 @@ func TestEncryptedFields(t *testing.T) { DisplayName: "Encryption Test", PasswordHash: "hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("Failed to create test user: %v", err) diff --git a/server/internal/db/system_test.go b/server/internal/db/system_test.go index 38055f7..71a2fe2 100644 --- a/server/internal/db/system_test.go +++ b/server/internal/db/system_test.go @@ -31,12 +31,14 @@ func TestSystemOperations(t *testing.T) { DisplayName: "User 1", PasswordHash: "hash1", Role: "editor", + Theme: "dark", }, { Email: "user2@test.com", DisplayName: "User 2", PasswordHash: "hash2", Role: "viewer", + Theme: "light", }, } diff --git a/server/internal/db/users_test.go b/server/internal/db/users_test.go index 8dfe4f0..cd11fb7 100644 --- a/server/internal/db/users_test.go +++ b/server/internal/db/users_test.go @@ -34,6 +34,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Test User", PasswordHash: "hashed_password", Role: models.RoleEditor, + Theme: "dark", }, wantErr: false, }, @@ -44,6 +45,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Another User", PasswordHash: "different_hash", Role: models.RoleViewer, + Theme: "light", }, wantErr: true, errContains: "UNIQUE constraint failed", @@ -108,6 +110,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Get By ID User", PasswordHash: "hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) @@ -159,6 +162,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Get By Email User", PasswordHash: "hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) @@ -210,6 +214,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Original Name", PasswordHash: "original_hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) @@ -249,12 +254,14 @@ func TestUserOperations(t *testing.T) { DisplayName: "User One", PasswordHash: "hash1", Role: models.RoleEditor, + Theme: "dark", }, { Email: "user2@example.com", DisplayName: "User Two", PasswordHash: "hash2", Role: models.RoleViewer, + Theme: "light", }, } @@ -305,6 +312,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Workspace User", PasswordHash: "hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) @@ -343,6 +351,7 @@ func TestUserOperations(t *testing.T) { DisplayName: "Delete User", PasswordHash: "hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) @@ -377,18 +386,21 @@ func TestUserOperations(t *testing.T) { DisplayName: "Admin One", PasswordHash: "hash1", Role: models.RoleAdmin, + Theme: "dark", }, { Email: "admin2@example.com", DisplayName: "Admin Two", PasswordHash: "hash2", Role: models.RoleAdmin, + Theme: "light", }, { Email: "editor@example.com", DisplayName: "Editor", PasswordHash: "hash3", Role: models.RoleEditor, + Theme: "dark", }, } diff --git a/server/internal/db/workspaces_test.go b/server/internal/db/workspaces_test.go index fda1c98..4b58aaf 100644 --- a/server/internal/db/workspaces_test.go +++ b/server/internal/db/workspaces_test.go @@ -26,6 +26,7 @@ func TestWorkspaceOperations(t *testing.T) { DisplayName: "Test User", PasswordHash: "hash", Role: models.RoleEditor, + Theme: "dark", }) if err != nil { t.Fatalf("failed to create test user: %v", err) diff --git a/server/internal/handlers/integration_test.go b/server/internal/handlers/integration_test.go index 64efca0..1ccf1a4 100644 --- a/server/internal/handlers/integration_test.go +++ b/server/internal/handlers/integration_test.go @@ -213,6 +213,7 @@ func (h *testHarness) createTestUser(t *testing.T, email, password string, role DisplayName: "Test User", PasswordHash: string(hashedPassword), Role: role, + Theme: "dark", } user, err = h.DB.CreateUser(user)