test: improve question flow test robustness

1 files changed, 24 insertions, 7 deletions

diff --git a/internal/handlers/handlers_test.go b/internal/handlers/handlers_test.go
index 33c7297..ef973c9 100644
--- a/internal/handlers/handlers_test.go
+++ b/internal/handlers/handlers_test.go
@@ -28,18 +28,35 @@ func TestQuestionFlow(t *testing.T) {
 	assert.NoError(err)
 
 	// Test 1: Load page and verify question
-	_, err = page.Goto(serverURL)
-	assert.NoError(err)
+	_, err = page.Goto(serverURL, playwright.PageGotoOptions{
+		WaitUntil: playwright.WaitUntilStateNetworkidle,
+		Timeout:   playwright.Float(10000),
+	})
+	assert.NoError(err, "Failed to load page")
+
+	// Wait for question to appear
+	err = page.Locator(".question-text").WaitFor(playwright.LocatorWaitForOptions{
+		State:   playwright.WaitForSelectorStateVisible,
+		Timeout: playwright.Float(5000),
+	})
+	assert.NoError(err, "Question text did not appear")
 
 	questionText, err := page.Locator(".question-text").TextContent()
-	assert.NoError(err)
-	assert.Contains(questionText, "Zweifel")
+	assert.NoError(err, "Failed to get question text")
+	assert.Contains(questionText, "Zweifel", "Question text mismatch")
 
-	// Verify all options are present
+	// Verify all options are present and clickable
 	options := []string{"Haben", "Hast", "Hat", "Habt"}
 	for _, opt := range options {
-		_, err := page.Locator("button:has-text('" + opt + "')").TextContent()
-		assert.NoError(err)
+		locator := page.Locator("button:has-text('" + opt + "')")
+		err := locator.WaitFor(playwright.LocatorWaitForOptions{
+			State:   playwright.WaitForSelectorStateVisible,
+			Timeout: playwright.Float(3000),
+		})
+		assert.NoError(err, "Option button '%s' not found", opt)
+		visible, err := locator.IsVisible()
+		assert.NoError(err, "Failed to check visibility for '%s'", opt)
+		assert.True(visible, "Option button '%s' not visible", opt)
 	}
 
 	// Test 2: Click correct answer