test: improve test reliability with explicit waits and timeouts

1 files changed, 12 insertions, 12 deletions

diff --git a/internal/handlers/handlers_test.go b/internal/handlers/handlers_test.go
index 1ddaad3..89f4d73 100644
--- a/internal/handlers/handlers_test.go
+++ b/internal/handlers/handlers_test.go
@@ -55,10 +55,10 @@ func TestQuestionFlow(t *testing.T) {
 	}
 	assert.Contains(questionText, "Zweifel", "Question text should contain 'Zweifel'")
 
-	// Set reasonable timeout for light site
-	page.SetDefaultTimeout(2000)
+	// Set longer timeout for CI environment
+	page.SetDefaultTimeout(5000)
 
-	// Verify options using test IDs
+	// Verify options using test IDs - first wait for them to be visible
 	for i := 1; i <= 4; i++ {
 		testID := fmt.Sprintf("option%d", i)
 		locator := page.Locator(fmt.Sprintf("[data-testid='%s']", testID))
@@ -72,18 +72,18 @@ func TestQuestionFlow(t *testing.T) {
 	// Test correct answer (option2 is correct per test data)
 	err = page.Locator("[data-testid='option2']").Click()
 	assert.NoError(err)
-
-	feedback := page.Locator("[data-testid='feedback']")
-	text, err := feedback.TextContent()
-	assert.NoError(err)
-	assert.Contains(text, "Correct")
-
+	
+	// Wait for feedback to update and verify
+	err = page.Locator("[data-testid='feedback']:has-text('Correct')").WaitFor()
+	assert.NoError(err, "Correct answer feedback not shown")
+	
 	// Test wrong answer
 	err = page.Locator("[data-testid='option1']").Click()
 	assert.NoError(err)
-	text, err = feedback.TextContent()
-	assert.NoError(err)
-	assert.Contains(text, "Wrong")
+	
+	// Wait for feedback to update and verify
+	err = page.Locator("[data-testid='feedback']:has-text('Wrong')").WaitFor()
+	assert.NoError(err, "Wrong answer feedback not shown")
 }
 
 type MockRepo struct{}