Fixed quickstart test

beam-cloud · Oct 4, 2024 · d93e856 · d93e856
1 parent fe57407
commit d93e856
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/light_test.py b/tests/light_test.py
@@ -75,7 +75,7 @@ def test_quickstart():
         command = [
             "beam",
             "deploy",
-            "app.py:run",
+            "app.py:hello_world",
             "--name",
             deployment_name,
         ]
@@ -91,7 +91,7 @@ def test_quickstart():
             assert (
                 response.status_code == 200
             ), f"{test_name} request to endpoint failed with status code: {response.status_code}"
-            assert "success" in response.text, f"{test_name} unexpected response"
+            assert "task_id" in response.text, f"{test_name} unexpected response"
     finally:
         delete_deployments(deployment_name)
         os.chdir(current_dir)