fix: tests

Author: mluukkai

e2e/prompts.spec.ts

@@ -65,11 +65,9 @@ test.describe('Prompts', () => {
     await page.getByTestId('system-message-input').fill('mocktest kurssitesti onnistui')
     await page.getByRole('button', { name: 'Save' }).click()
 
-    // Close the prompt modal to go back to the prompts list
-    await page.getByTestId('close-modal').click()
-
-    // Navigate directly to chat with the prompt using URL parameter
-    await page.goto(`/test-course-course-id?promptId=${newPromptName}`)
+    // Dialog closes automatically, now back at prompts list
+    // Select the prompt by clicking on its row
+    await page.getByTestId(`prompt-row-${newPromptName}`).click()
 
     // Now in chat view
     // The prompt is active.
@@ -118,11 +116,9 @@ test.describe('Prompts', () => {
     await page.getByTestId(`source-material-rag-${test.info().workerIndex}-teacher`).click()
     await page.getByRole('button', { name: 'Save' }).click()
 
-    // Close the prompt modal
-    await page.getByTestId('close-modal').click()
-
-    // Navigate directly to chat with the prompt using URL parameter
-    await page.goto(`/test-course-course-id?promptId=${newPromptName}`)
+    // Dialog closes automatically, now back at prompts list
+    // Select the prompt by clicking on its row
+    await page.getByTestId(`prompt-row-${newPromptName}`).click()
 
     // Now in chat view
     // The prompt is active.

e2e/student.spec.ts

@@ -45,10 +45,10 @@ test.describe('Student', () => {
     await page.goto('/test-course-course-id')
     await acceptDisclaimer(page)
 
-    // Students shouldn't see the edit prompt button in the sidebar
-    await expect(page.getByTestId('edit-prompt-button')).toBeHidden()
+    // Students shouldn't see the edit prompt button in the sidebar (when no prompt is selected)
+    await expect(page.getByTestId('edit-prompt-button')).not.toBeVisible()
 
-    // But they should see the prompt details button when a prompt is selected
-    await expect(page.getByTestId('prompt-details-button')).toBeVisible()
+    // They should see the button to choose a prompt
+    await expect(page.getByTestId('choose-prompt-button')).toBeVisible()
   })
 })