stuck on server test

test_server.py

@@ -0,0 +1,42 @@
+import os
+import pytest
+import pytest_asyncio
+from playwright.async_api import async_playwright, expect
+import asyncio
+
+@pytest_asyncio.fixture(scope="session")
+def event_loop():
+    policy = asyncio.get_event_loop_policy()
+    loop = policy.new_event_loop()
+    yield loop
+    loop.close()
+
+@pytest.mark.asyncio
+async def test_space_loads():
+    """Test that we can access the Space and see the heading"""
+    async with async_playwright() as playwright:
+        browser = await playwright.chromium.launch(headless=False)
+        page = await browser.new_page()
+        
+        try:
+            # Navigate to the Space
+            await page.goto("https://huggingface.co/spaces/rawwerks/handwriting-ocr")
+            
+            if os.getenv('PWDEBUG'):
+                await page.pause()
+            
+            # Now look for the login button from Gradio
+            print("Looking for login button...")
+            login_button = page.get_by_role("button").filter(has_text="Login")
+            await login_button.wait_for(state="visible", timeout=5000)
+            print("Found login button, clicking...")
+            
+            if os.getenv('PWDEBUG'):
+                await page.pause()
+            
+            await login_button.click()
+            print("Clicked login button")
+            
+        finally:
+            await browser.close()
+

test_server.sh

@@ -3,6 +3,16 @@
 # Exit on error
 set -e
 
+# Parse command line arguments
+DEBUG=0
+while [[ "$#" -gt 0 ]]; do
+    case $1 in
+        --debug) DEBUG=1 ;;
+        *) echo "Unknown parameter: $1"; exit 1 ;;
+    esac
+    shift
+done
+
 # Check if auth_state.json exists
 if [ ! -f "auth_state.json" ]; then
     echo "Error: auth_state.json not found. Please run save_hf_storage_state.py first to save your Hugging Face login state."
@@ -40,53 +50,55 @@ import asyncio
 
 @pytest_asyncio.fixture(scope="session")
 def event_loop():
-    """Create an instance of the default event loop for each test case."""
     policy = asyncio.get_event_loop_policy()
     loop = policy.new_event_loop()
     yield loop
     loop.close()
 
 @pytest.mark.asyncio
-async def test_space_loads_logged_in():
-    """Test that we can access the Space with our saved authentication state"""
-    auth_state_path = "auth_state.json"
-    assert os.path.exists(auth_state_path), f"Authentication state file not found at {auth_state_path}"
-    
+async def test_space_loads():
+    """Test that we can access the Space and see the heading"""
     async with async_playwright() as playwright:
-        # Launch browser with stored authentication state
         browser = await playwright.chromium.launch(headless=False)
-        context = await browser.new_context(storage_state=auth_state_path)
-        page = await context.new_page()
+        page = await browser.new_page()
         
         try:
             # Navigate to the Space
             await page.goto("https://huggingface.co/spaces/rawwerks/handwriting-ocr")
-            await page.wait_for_load_state("networkidle")
             
-            # Check if the main heading is visible
-            await expect(page.locator("h2", has_text="Crowdsourcing Handwriting OCR Dataset")).to_be_visible()
+            if os.getenv('PWDEBUG'):
+                await page.pause()
+            
+            # Now look for the login button from Gradio
+            print("Looking for login button...")
+            login_button = page.get_by_role("button").filter(has_text="Login")
+            await login_button.wait_for(state="visible", timeout=5000)
+            print("Found login button, clicking...")
             
-            # Check for logged-in state
-            await expect(page.locator("text=Logged in as:")).to_be_visible()
+            if os.getenv('PWDEBUG'):
+                await page.pause()
             
-            # Check that the interface elements are visible
-            await expect(page.get_by_label("Text to Handwrite")).to_be_visible()
-            await expect(page.get_by_label("Upload Handwritten Image")).to_be_visible()
-            await expect(page.get_by_role("button", name="Submit")).to_be_visible()
-            await expect(page.get_by_role("button", name="Skip")).to_be_visible()
+            await login_button.click()
+            print("Clicked login button")
             
         finally:
-            await context.close()
             await browser.close()
+
 EOL
 
-# Run the server tests
+# Run the server tests with or without Playwright Inspector based on debug flag
 echo "Running server tests..."
-python -m pytest test_server.py -v
+if [ "$DEBUG" -eq 1 ]; then
+    echo "Debug mode enabled - launching Playwright Inspector..."
+    PWDEBUG=1 python -m pytest test_server.py -v
+else
+    python -m pytest test_server.py -v
+fi
 
 # Cleanup
 echo "Cleaning up..."
 rm test_server.py
+rm -f error.png
 
 # Deactivate virtual environment
 deactivate