feat: add server test script for HF Space - Add test_server.sh to run tests against deployed Space - Test logged-in state and UI elements on actual Space - Use saved auth state for authentication

test_server.sh

@@ -0,0 +1,92 @@
+#!/bin/bash
+
+# Exit on error
+set -e
+
+# 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."
+    exit 1
+fi
+
+# Activate virtual environment if it exists, create if it doesn't
+if [ ! -d "venv" ]; then
+    echo "Creating virtual environment..."
+    python -m venv venv
+fi
+
+echo "Activating virtual environment..."
+source venv/bin/activate
+
+# Install dependencies if needed
+echo "Installing dependencies..."
+if ! command -v uv &> /dev/null; then
+    echo "Installing uv package installer..."
+    curl -LsSf https://astral.sh/uv/install.sh | sh
+fi
+uv pip install -r requirements.txt
+
+# Install Playwright browsers if needed
+echo "Installing Playwright browsers..."
+playwright install chromium
+
+# Create a new test file specifically for server tests
+cat > test_server.py << 'EOL'
+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():
+    """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 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()
+        
+        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()
+            
+            # Check for logged-in state
+            await expect(page.locator("text=Logged in as:")).to_be_visible()
+            
+            # 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()
+            
+        finally:
+            await context.close()
+            await browser.close()
+EOL
+
+# Run the server tests
+echo "Running server tests..."
+python -m pytest test_server.py -v
+
+# Cleanup
+echo "Cleaning up..."
+rm test_server.py
+
+# Deactivate virtual environment
+deactivate