from src.worker_payload import CREATE_SPACE_WORKER_SCRIPT


def test_create_worker_discovers_gradio_api_names():
    assert "def extract_api_names" in CREATE_SPACE_WORKER_SCRIPT
    assert "def predict_with_available_endpoint" in CREATE_SPACE_WORKER_SCRIPT
    assert '"/greet"' in CREATE_SPACE_WORKER_SCRIPT
    assert '"/predict"' in CREATE_SPACE_WORKER_SCRIPT
    assert 'client.predict(value)' in CREATE_SPACE_WORKER_SCRIPT