cornflow 2.0.0a11__py3-none-any.whl → 2.0.0a13__py3-none-any.whl

cornflow/tests/unit/test_executions.py

@@ -6,6 +6,8 @@ Unit test for the executions endpoints
 import json
 from unittest.mock import patch
 
+from cornflow.app import create_app
+
 # Import from internal modules
 from cornflow.models import ExecutionModel, InstanceModel
 from cornflow.tests.const import (
@@ -17,10 +19,10 @@ from cornflow.tests.const import (
     INSTANCE_URL,
     DAG_URL,
     BAD_EXECUTION_PATH,
-    EXECUTION_SOLUTION_PATH,
+    EXECUTION_SOLUTION_PATH, EDIT_EXECUTION_SOLUTION,
 )
 from cornflow.tests.custom_test_case import CustomTestCase, BaseTestCases
-from cornflow.tests.unit.tools import patch_af_client
+from cornflow.tests.unit.tools import patch_af_client, patch_db_client
 
 
 class TestExecutionsListEndpoint(BaseTestCases.ListFilters):
@@ -62,13 +64,13 @@ class TestExecutionsListEndpoint(BaseTestCases.ListFilters):
     def test_new_execution(self):
         self.create_new_row(self.url, self.model, payload=self.payload)
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
+    @patch("cornflow.endpoints.execution.Airflow")
     def test_new_execution_run(self, af_client_class):
         patch_af_client(af_client_class)
 
         self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
+    @patch("cornflow.endpoints.execution.Airflow")
     def test_new_execution_bad_config(self, af_client_class):
         patch_af_client(af_client_class)
         response = self.create_new_row(
@@ -81,7 +83,7 @@ class TestExecutionsListEndpoint(BaseTestCases.ListFilters):
         self.assertIn("error", response)
         self.assertIn("jsonschema_errors", response)
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
+    @patch("cornflow.endpoints.execution.Airflow")
     def test_new_execution_partial_config(self, af_client_class):
         patch_af_client(af_client_class)
         self.payload["config"].pop("solver")
@@ -91,7 +93,7 @@ class TestExecutionsListEndpoint(BaseTestCases.ListFilters):
         self.assertIn("solver", response["config"])
         self.assertEqual(response["config"]["solver"], "cbc")
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
+    @patch("cornflow.endpoints.execution.Airflow")
     def test_new_execution_with_solution(self, af_client_class):
         patch_af_client(af_client_class)
         self.payload["data"] = self.solution
@@ -102,7 +104,7 @@ class TestExecutionsListEndpoint(BaseTestCases.ListFilters):
             check_payload=False,
         )
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
+    @patch("cornflow.endpoints.execution.Airflow")
     def test_new_execution_with_solution_bad(self, af_client_class):
         patch_af_client(af_client_class)
         patch_af_client(af_client_class)
@@ -144,6 +146,131 @@ class TestExecutionsListEndpoint(BaseTestCases.ListFilters):
         self.assertEqual(len(rows.json), len(self.payloads))
 
 
+class TestExecutionsListEndpointDatabricks(BaseTestCases.ListFilters):
+    def setUp(self):
+        super().setUp()
+
+        with open(INSTANCE_PATH) as f:
+            payload = json.load(f)
+        fk_id = self.create_new_row(INSTANCE_URL, InstanceModel, payload)
+        self.url = EXECUTION_URL_NORUN
+        self.model = ExecutionModel
+
+        def load_file_fk(_file):
+            with open(_file) as f:
+                temp = json.load(f)
+            temp["instance_id"] = fk_id
+            return temp
+
+        self.payload = load_file_fk(EXECUTION_PATH)
+        self.bad_payload = load_file_fk(BAD_EXECUTION_PATH)
+        self.payloads = [load_file_fk(f) for f in EXECUTIONS_LIST]
+        self.solution = load_file_fk(EXECUTION_SOLUTION_PATH)
+        self.keys_to_check = [
+            "data_hash",
+            "created_at",
+            "config",
+            "state",
+            "message",
+            "schema",
+            "description",
+            "id",
+            "user_id",
+            "log",
+            "instance_id",
+            "name",
+            "indicators",
+        ]
+
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
+
+    def test_new_execution(self):
+        self.create_new_row(self.url, self.model, payload=self.payload)
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_new_execution_run_databricks(self, db_client_class):
+        patch_db_client(db_client_class)
+
+        self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_new_execution_bad_config(self, db_client_class):
+        patch_db_client(db_client_class)
+        response = self.create_new_row(
+            EXECUTION_URL,
+            self.model,
+            payload=self.bad_payload,
+            expected_status=400,
+            check_payload=False,
+        )
+        self.assertIn("error", response)
+        self.assertIn("jsonschema_errors", response)
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_new_execution_partial_config(self, db_client_class):
+        patch_db_client(db_client_class)
+        self.payload["config"].pop("solver")
+        response = self.create_new_row(
+            EXECUTION_URL, self.model, payload=self.payload, check_payload=False
+        )
+        self.assertIn("solver", response["config"])
+        self.assertEqual(response["config"]["solver"], "cbc")
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_new_execution_with_solution(self, db_client_class):
+        patch_db_client(db_client_class)
+        self.payload["data"] = self.solution
+        response = self.create_new_row(
+            EXECUTION_URL,
+            self.model,
+            payload=self.payload,
+            check_payload=False,
+        )
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_new_execution_with_solution_bad(self, db_client_class):
+        patch_db_client(db_client_class)
+        patch_db_client(db_client_class)
+        self.payload["data"] = {"message": "THIS IS NOT A VALID SOLUTION"}
+        response = self.create_new_row(
+            EXECUTION_URL,
+            self.model,
+            payload=self.payload,
+            check_payload=False,
+            expected_status=400,
+        )
+        self.assertIn("error", response)
+        self.assertIn("jsonschema_errors", response)
+
+    def test_new_execution_no_instance(self):
+        payload = dict(self.payload)
+        payload["instance_id"] = "bad_id"
+        response = self.client.post(
+            self.url,
+            data=json.dumps(payload),
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
+        )
+        self.assertEqual(404, response.status_code)
+        self.assertTrue("error" in response.json)
+
+    def test_get_executions(self):
+        self.get_rows(self.url, self.payloads, keys_to_check=self.keys_to_check)
+
+    def test_get_no_executions(self):
+        self.get_no_rows(self.url)
+
+    def test_get_executions_superadmin(self):
+        self.get_rows(self.url, self.payloads, keys_to_check=self.keys_to_check)
+        token = self.create_service_user()
+        rows = self.client.get(
+            self.url, follow_redirects=True, headers=self.get_header_with_auth(token)
+        )
+        self.assertEqual(len(rows.json), len(self.payloads))
+
+
 class TestExecutionRelaunchEndpoint(CustomTestCase):
     def setUp(self):
         super().setUp()
@@ -193,7 +320,7 @@ class TestExecutionRelaunchEndpoint(CustomTestCase):
         self.assertEqual(row["config"], self.payload["config"])
         self.assertIsNone(row["checks"])
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
+    @patch("cornflow.endpoints.execution.Airflow")
     def test_relaunch_execution_run(self, af_client_class):
         patch_af_client(af_client_class)
 
@@ -240,12 +367,113 @@ class TestExecutionRelaunchEndpoint(CustomTestCase):
         self.assertEqual(404, response.status_code)
 
 
+class TestExecutionRelaunchEndpointDatabricks(CustomTestCase):
+    def setUp(self):
+        super().setUp()
+
+        with open(INSTANCE_PATH) as f:
+            payload = json.load(f)
+        fk_id = self.create_new_row(INSTANCE_URL, InstanceModel, payload)
+        self.url = EXECUTION_URL_NORUN
+        self.model = ExecutionModel
+
+        def load_file_fk(_file):
+            with open(_file) as f:
+                temp = json.load(f)
+            temp["instance_id"] = fk_id
+            return temp
+
+        self.payload = load_file_fk(EXECUTION_PATH)
+
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
+
+    def test_relaunch_execution(self):
+        idx = self.create_new_row(self.url, self.model, payload=self.payload)
+
+        # Add solution checks to see if they are deleted correctly
+        token = self.create_service_user()
+        self.update_row(
+            url=DAG_URL + idx + "/",
+            payload_to_check=dict(),
+            change=dict(solution_schema="_data_checks", checks=dict(check_1=[])),
+            token=token,
+            check_payload=False,
+        )
+
+        url = EXECUTION_URL + idx + "/relaunch/?run=0"
+        self.payload["config"]["warmStart"] = False
+        response = self.client.post(
+            url,
+            data=json.dumps({"config": self.payload["config"]}),
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
+        )
+        self.assertEqual(201, response.status_code)
+
+        url = EXECUTION_URL + idx + "/data"
+        row = self.client.get(
+            url, follow_redirects=True, headers=self.get_header_with_auth(self.token)
+        ).json
+
+        self.assertEqual(row["config"], self.payload["config"])
+        self.assertIsNone(row["checks"])
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_relaunch_execution_run(self, db_client_class):
+        patch_db_client(db_client_class)
+
+        idx = self.create_new_row(self.url, self.model, payload=self.payload)
+
+        # Add solution checks to see if they are deleted correctly
+        token = self.create_service_user()
+        self.update_row(
+            url=DAG_URL + idx + "/",
+            payload_to_check=dict(),
+            change=dict(solution_schema="_data_checks", checks=dict(check_1=[])),
+            token=token,
+            check_payload=False,
+        )
+
+        url = EXECUTION_URL + idx + "/relaunch/"
+        self.payload["config"]["warmStart"] = False
+        response = self.client.post(
+            url,
+            data=json.dumps({"config": self.payload["config"]}),
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
+        )
+        self.assertEqual(201, response.status_code)
+
+        url = EXECUTION_URL + idx + "/data"
+        row = self.client.get(
+            url, follow_redirects=True, headers=self.get_header_with_auth(self.token)
+        ).json
+
+        self.assertEqual(row["config"], self.payload["config"])
+        self.assertIsNone(row["checks"])
+
+    def test_relaunch_invalid_execution(self):
+        idx = "thisIsAnInvalidExecutionId"
+        url = EXECUTION_URL + idx + "/relaunch/?run=0"
+        self.payload["config"]["warmStart"] = False
+        response = self.client.post(
+            url,
+            data=json.dumps({"config": self.payload["config"]}),
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
+        )
+        self.assertEqual(404, response.status_code)
+
+
 class TestExecutionsDetailEndpointMock(CustomTestCase):
     def setUp(self):
         super().setUp()
         with open(INSTANCE_PATH) as f:
             payload = json.load(f)
         fk_id = self.create_new_row(INSTANCE_URL, InstanceModel, payload)
+        self.instance_payload = payload
         self.model = ExecutionModel
         self.response_items = {
             "id",
@@ -269,7 +497,7 @@ class TestExecutionsDetailEndpointMock(CustomTestCase):
         self.payload["instance_id"] = fk_id
 
 
-class TestExecutionsDetailEndpoint(
+class TestExecutionsDetailEndpointAirflow(
     TestExecutionsDetailEndpointMock, BaseTestCases.DetailEndpoint
 ):
     def setUp(self):
@@ -277,115 +505,193 @@ class TestExecutionsDetailEndpoint(
         self.url = self.url
         self.query_arguments = {"run": 0}
 
-    # TODO: this test should be moved as it is not using the detail endpoint
-    def test_incomplete_payload2(self):
-        payload = {"description": "arg", "instance_id": self.payload["instance_id"]}
-        response = self.create_new_row(
-            self.url + "?run=0",
-            self.model,
-            payload,
-            expected_status=400,
-            check_payload=False,
+    def create_app(self):
+        app = create_app("testing")  # configuración para Airflow
+        return app
+
+    @patch("cornflow.endpoints.execution.Airflow")
+    def test_stop_execution(self, af_client_class):
+        patch_af_client(af_client_class)
+
+        idx = self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+
+        response = self.client.post(
+            self.url + str(idx) + "/",
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
         )
 
-    def test_create_delete_instance_load(self):
-        idx = self.create_new_row(self.url + "?run=0", self.model, self.payload)
-        keys_to_check = [
-            "message",
-            "id",
-            "schema",
-            "data_hash",
-            "config",
-            "instance_id",
-            "user_id",
-            "indicators",
-            "description",
-            "name",
-            "created_at",
-            "state",
-        ]
-        execution = self.get_one_row(
-            self.url + idx,
-            payload={**self.payload, **dict(id=idx)},
-            keys_to_check=keys_to_check,
+        self.assertEqual(200, response.status_code)
+        self.assertEqual(response.json["message"], "The execution has been stopped")
+
+    def test_edit_execution(self):
+
+        id_new_instance = self.create_new_row(
+            INSTANCE_URL, InstanceModel, self.instance_payload
         )
-        self.delete_row(self.url + idx + "/")
-        keys_to_check = [
-            "id",
-            "schema",
-            "description",
-            "name",
-            "user_id",
-            "executions",
-            "created_at",
-            "data_hash",
-        ]
-        instance = self.get_one_row(
-            INSTANCE_URL + execution["instance_id"] + "/",
-            payload={},
-            expected_status=200,
+        idx = self.create_new_row(
+            self.url_with_query_arguments(), self.model, self.payload
+        )
+
+        # Extract the data from data/edit_execution_solution.json
+        with open(EDIT_EXECUTION_SOLUTION) as f:
+            data = json.load(f)
+
+        data = {
+            "name": "new_name",
+            "description": "Updated description",
+            "data": data,
+            "instance_id": id_new_instance,
+        }
+        payload_to_check = {
+            "id": idx,
+            "name":"new_name",
+            "description":"Updated description",
+            "data_hash":"74234e98afe7498fb5daf1f36ac2d78acc339464f950703b8c019892f982b90b",
+            "instance_id":"805bad3280c95e45384dc6bd91a41317f9a7858c",
+            }
+        self.update_row(
+            self.url + str(idx) + "/",
+            data,
+            payload_to_check,
+        )
+
+    @patch("cornflow.endpoints.execution.Airflow")
+    def test_get_one_status(self, af_client_class):
+        patch_af_client(af_client_class)
+        idx = self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+        payload = dict(self.payload)
+        payload["id"] = idx
+        keys_to_check = ["state", "message", "id", "data_hash"]
+        data = self.get_one_row(
+            EXECUTION_URL + idx + "/status/",
+            payload,
             check_payload=False,
             keys_to_check=keys_to_check,
         )
-        executions = [execution["id"] for execution in instance["executions"]]
-        self.assertFalse(idx in executions)
+        self.assertEqual(data["state"], 1)
 
-    def test_delete_instance_deletes_execution(self):
-        # we create a new instance
-        with open(INSTANCE_PATH) as f:
-            payload = json.load(f)
-        fk_id = self.create_new_row(INSTANCE_URL, InstanceModel, payload)
-        payload = {**self.payload, **dict(instance_id=fk_id)}
-        # we create an execution for that instance
-        idx = self.create_new_row(self.url + "?run=0", self.model, payload)
-        self.get_one_row(self.url + idx, payload={**self.payload, **dict(id=idx)})
-        # we delete the new instance
-        self.delete_row(INSTANCE_URL + fk_id + "/")
-        # we check the execution does not exist
-        self.get_one_row(
-            self.url + idx, payload={}, expected_status=404, check_payload=False
+    @patch("cornflow.endpoints.execution.Airflow")
+    def test_put_one_status(self, af_client_class):
+        patch_af_client(af_client_class)
+
+        idx = self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+        payload = dict(self.payload)
+        payload["id"] = idx
+        response = self.client.put(
+            EXECUTION_URL + idx + "/status/",
+            data=json.dumps({"status": 0}),
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
         )
 
-    def test_update_one_row_data(self):
+        self.assertEqual(200, response.status_code)
+        self.assertEqual(f"execution {idx} updated correctly", response.json["message"])
+
+class TestExecutionsDetailEndpointDatabricks(
+    TestExecutionsDetailEndpointMock, BaseTestCases.DetailEndpoint
+):
+    def setUp(self):
+        super().setUp()
+        self.url = self.url
+        self.query_arguments = {"run": 0}
+
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_stop_execution(self, db_client_class):
+        patch_db_client(db_client_class)
+
+        idx = self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+
+        response = self.client.post(
+            self.url + str(idx) + "/",
+            follow_redirects=True,
+            headers=self.get_header_with_auth(self.token),
+        )
+
+        self.assertEqual(200, response.status_code)
+        self.assertEqual(
+            response.json["message"], "This feature is not available for Databricks"
+        )
+    def test_edit_execution(self):
+
+        id_new_instance = self.create_new_row(
+            INSTANCE_URL, InstanceModel, self.instance_payload
+        )
         idx = self.create_new_row(
             self.url_with_query_arguments(), self.model, self.payload
         )
-        with open(INSTANCE_PATH) as f:
-            payload = json.load(f)
-        payload["data"]["parameters"]["name"] = "NewName"
 
-        url = self.url + str(idx) + "/"
-        payload = {
-            **self.payload,
-            **dict(id=idx, name="new_name", data=payload["data"]),
+        # Extract the data from data/edit_execution_solution.json
+        with open(EDIT_EXECUTION_SOLUTION) as f:
+            data = json.load(f)
+
+        data = {
+            "name": "new_name",
+            "description": "Updated description",
+            "data": data,
+            "instance_id": id_new_instance,
         }
+        payload_to_check = {
+            "id": idx,
+            "name":"new_name",
+            "description":"Updated description",
+            "data_hash":"74234e98afe7498fb5daf1f36ac2d78acc339464f950703b8c019892f982b90b",
+            "instance_id":"805bad3280c95e45384dc6bd91a41317f9a7858c",
+            }
         self.update_row(
-            url,
-            dict(name="new_name", data=payload["data"]),
-            payload,
+            self.url + str(idx) + "/",
+            data,
+            payload_to_check,
         )
 
-        url += "data/"
-        row = self.client.get(
-            url, follow_redirects=True, headers=self.get_header_with_auth(self.token)
-        )
 
-        self.assertEqual(row.json["checks"], None)
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
-    def test_stop_execution(self, af_client_class):
-        patch_af_client(af_client_class)
 
+class TestExecutionsStatusEndpointDatabricks(TestExecutionsDetailEndpointMock):
+    def setUp(self):
+        super().setUp()
+        self.response_items = {"id", "name", "status"}
+        self.items_to_check = []
+
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
+
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_get_one_status(self, db_client_class):
+        patch_db_client(db_client_class)
         idx = self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+        payload = dict(self.payload)
+        payload["id"] = idx
+        keys_to_check = ["state", "message", "id", "data_hash"]
+        data = self.get_one_row(
+            EXECUTION_URL + idx + "/status/",
+            payload,
+            check_payload=False,
+            keys_to_check=keys_to_check,
+        )
+        self.assertEqual(data["state"], -1)
 
-        response = self.client.post(
-            self.url + str(idx) + "/",
+    @patch("cornflow.endpoints.execution.Databricks")
+    def test_put_one_status(self, db_client_class):
+        patch_db_client(db_client_class)
+
+        idx = self.create_new_row(EXECUTION_URL, self.model, payload=self.payload)
+        payload = dict(self.payload)
+        payload["id"] = idx
+        response = self.client.put(
+            EXECUTION_URL + idx + "/status/",
+            data=json.dumps({"status": 0}),
             follow_redirects=True,
             headers=self.get_header_with_auth(self.token),
         )
 
         self.assertEqual(200, response.status_code)
-        self.assertEqual(response.json["message"], "The execution has been stopped")
+        self.assertEqual(f"execution {idx} updated correctly", response.json["message"])
 
 
 class TestExecutionsDataEndpoint(TestExecutionsDetailEndpointMock):
@@ -430,6 +736,52 @@ class TestExecutionsDataEndpoint(TestExecutionsDetailEndpointMock):
         )
 
 
+class TestExecutionsDataEndpointDatabricks(TestExecutionsDetailEndpointMock):
+    def setUp(self):
+        super().setUp()
+        self.response_items = {"id", "name", "data"}
+        self.items_to_check = ["name"]
+        self.keys_to_check = [
+            "created_at",
+            "checks",
+            "instance_id",
+            "schema",
+            "data",
+            "user_id",
+            "message",
+            "data_hash",
+            "log",
+            "config",
+            "description",
+            "state",
+            "name",
+            "id",
+        ]
+
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
+
+    def test_get_one_execution(self):
+        idx = self.create_new_row(EXECUTION_URL_NORUN, self.model, self.payload)
+        self.url = EXECUTION_URL + idx + "/data/"
+        payload = dict(self.payload)
+        payload["id"] = idx
+        self.get_one_row(self.url, payload, keys_to_check=self.keys_to_check)
+
+    def test_get_one_execution_superadmin(self):
+        idx = self.create_new_row(EXECUTION_URL_NORUN, self.model, self.payload)
+        payload = dict(self.payload)
+        payload["id"] = idx
+        token = self.create_service_user()
+        self.get_one_row(
+            EXECUTION_URL + idx + "/data/",
+            payload,
+            token=token,
+            keys_to_check=self.keys_to_check,
+        )
+
+
 class TestExecutionsLogEndpoint(TestExecutionsDetailEndpointMock):
     def setUp(self):
         super().setUp()
@@ -473,45 +825,52 @@ class TestExecutionsLogEndpoint(TestExecutionsDetailEndpointMock):
         )
 
 
-class TestExecutionsStatusEndpoint(TestExecutionsDetailEndpointMock):
+class TestExecutionsLogEndpointDatabricks(TestExecutionsDetailEndpointMock):
     def setUp(self):
         super().setUp()
-        self.response_items = {"id", "name", "status"}
-        self.items_to_check = []
+        self.response_items = {"id", "name", "log", "indicators"}
+        self.items_to_check = ["name"]
+        self.keys_to_check = [
+            "created_at",
+            "id",
+            "log_text",
+            "instance_id",
+            "state",
+            "message",
+            "description",
+            "data_hash",
+            "name",
+            "log",
+            "schema",
+            "user_id",
+            "config",
+            "indicators",
+        ]
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
-    def test_get_one_status(self, af_client_class):
-        patch_af_client(af_client_class)
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
 
-        idx = self.create_new_row(EXECUTION_URL, self.model, self.payload)
+    def test_get_one_execution(self):
+        idx = self.create_new_row(EXECUTION_URL_NORUN, self.model, self.payload)
         payload = dict(self.payload)
         payload["id"] = idx
-        keys_to_check = ["state", "message", "id", "data_hash"]
-        data = self.get_one_row(
-            EXECUTION_URL + idx + "/status/",
-            payload,
-            check_payload=False,
-            keys_to_check=keys_to_check,
+        self.get_one_row(
+            EXECUTION_URL + idx + "/log/", payload, keys_to_check=self.keys_to_check
         )
-        self.assertEqual(data["state"], 1)
 
-    @patch("cornflow.endpoints.execution_databricks.Airflow")
-    def test_put_one_status(self, af_client_class):
-        patch_af_client(af_client_class)
-
-        idx = self.create_new_row(EXECUTION_URL, self.model, self.payload)
+    def test_get_one_execution_superadmin(self):
+        idx = self.create_new_row(EXECUTION_URL_NORUN, self.model, self.payload)
         payload = dict(self.payload)
         payload["id"] = idx
-        response = self.client.put(
-            EXECUTION_URL + idx + "/status/",
-            data=json.dumps({"status": 0}),
-            follow_redirects=True,
-            headers=self.get_header_with_auth(self.token),
+        token = self.create_service_user()
+        self.get_one_row(
+            EXECUTION_URL + idx + "/log/",
+            payload,
+            token=token,
+            keys_to_check=self.keys_to_check,
         )
 
-        self.assertEqual(200, response.status_code)
-        self.assertEqual(f"execution {idx} updated correctly", response.json["message"])
-
 
 class TestExecutionsModel(TestExecutionsDetailEndpointMock):
     def test_repr_method(self):
@@ -521,3 +880,17 @@ class TestExecutionsModel(TestExecutionsDetailEndpointMock):
     def test_str_method(self):
         idx = self.create_new_row(self.url + "?run=0", self.model, self.payload)
         self.str_method(idx, f"<Execution {idx}>")
+
+
+class TestExecutionsModelDatabricks(TestExecutionsDetailEndpointMock):
+    def create_app(self):
+        app = create_app("testing-databricks")
+        return app
+
+    def test_repr_method(self):
+        idx = self.create_new_row(self.url + "?run=0", self.model, self.payload)
+        self.repr_method(idx, f"<Execution {idx}>")
+
+    def test_str_method(self):
+        idx = self.create_new_row(self.url + "?run=0", self.model, self.payload)
+        self.str_method(idx, f"<Execution {idx}>")