From b0ad61d5bb73ab39a3fb773b6b1d206d79496da9 Mon Sep 17 00:00:00 2001
From: Five Grant <5@fivegrant.com>
Date: Wed, 16 Aug 2023 10:48:11 -0500
Subject: [PATCH] Run through simulate e2e fully [caveats]

---
 tests/integration/test_simulate.py | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/tests/integration/test_simulate.py b/tests/integration/test_simulate.py
index 7c2f93c..ed04a23 100644
--- a/tests/integration/test_simulate.py
+++ b/tests/integration/test_simulate.py
@@ -1,4 +1,5 @@
-# import json
+import json
+import re
 
 import pytest
 
@@ -9,10 +10,9 @@
 
 @pytest.mark.example_dir("simulate")
 def test_simulate_example(example_context, client, worker, requests_mock):
-    # job_id = example_context["tds_simulation"]["id"]
     request = example_context["request"]
-    # config_id = request["model_config_id"]
-    # model = json.loads(example_context["fetch"](config_id + ".json"))
+    config_id = request["model_config_id"]
+    model = json.loads(example_context["fetch"](config_id + ".json"))
 
     requests_mock.post(f"{TDS_URL}/simulations/", json={"id": None})
 
@@ -28,4 +28,24 @@ def test_simulate_example(example_context, client, worker, requests_mock):
     status = response.json()["status"]
     assert status == "queued"
 
+    tds_sim = example_context["tds_simulation"]
+    tds_sim["id"] = simulation_id
+
+    requests_mock.get(f"{TDS_URL}/simulations/{simulation_id}", json=tds_sim)
+    requests_mock.put(
+        f"{TDS_URL}/simulations/{simulation_id}", json={"status": "success"}
+    )
+    requests_mock.get(f"{TDS_URL}/model_configurations/{config_id}", json=model)
+    # TODO: Save files to locations where we can check against them
+    upload_url = re.compile("upload-url")
+    requests_mock.get(upload_url, json={"url": "http://WONTWORK"})
+    requests_mock.put("http://WONTWORK", json={"url": "WONTWORK"})
+
+    # TODO: Mock PyCIEMSS lib
     worker.work(burst=True)
+
+    response = client.get(
+        f"/status/{simulation_id}",
+    )
+    status = response.json()["status"]
+    assert status == "complete"