fixe test

apache · Dec 20, 2022 · 9bdc015 · 9bdc015
1 parent 3c0e357
commit 9bdc015
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/integration_tests/explore/api_tests.py b/tests/integration_tests/explore/api_tests.py
@@ -112,7 +112,7 @@ def test_no_params_provided(test_client, login_as_admin):
 
 def test_get_from_cache(test_client, login_as_admin, dataset):
     resp = test_client.get(
-        f"api/v1/explore/?form_data_key={FORM_DATA_KEY}&dataset_id={dataset.id}&dataset_type={dataset.type}"
+        f"api/v1/explore/?form_data_key={FORM_DATA_KEY}&datasource_id={dataset.id}&datasource_type={dataset.type}"
     )
     assert resp.status_code == 200
     data = json.loads(resp.data.decode("utf-8"))
@@ -146,7 +146,7 @@ def test_get_from_cache_unknown_key_chart_id(
 def test_get_from_cache_unknown_key_dataset(test_client, login_as_admin, dataset):
     unknown_key = "unknown_key"
     resp = test_client.get(
-        f"api/v1/explore/?form_data_key={unknown_key}&dataset_id={dataset.id}&dataset_type={dataset.type}"
+        f"api/v1/explore/?form_data_key={unknown_key}&datasource_id={dataset.id}&datasource_type={dataset.type}"
     )
     assert resp.status_code == 200
     data = json.loads(resp.data.decode("utf-8"))
@@ -204,15 +204,15 @@ def test_get_dataset_access_denied(
 ):
     message = "Dataset access denied"
     mock_can_access_datasource.side_effect = DatasetAccessDeniedError(
-        message=message, dataset_id=dataset.id, dataset_type=dataset.type
+        message=message, datasource_id=dataset.id, datasource_type=dataset.type
     )
     resp = test_client.get(
-        f"api/v1/explore/?form_data_key={FORM_DATA_KEY}&dataset_id={dataset.id}&dataset_type={dataset.type}"
+        f"api/v1/explore/?form_data_key={FORM_DATA_KEY}&datasource_id={dataset.id}&datasource_type={dataset.type}"
     )
     data = json.loads(resp.data.decode("utf-8"))
     assert resp.status_code == 403
-    assert data["dataset_id"] == dataset.id
-    assert data["dataset_type"] == dataset.type
+    assert data["datasource_id"] == dataset.id
+    assert data["datasource_type"] == dataset.type
     assert data["message"] == message
 
 
@@ -221,7 +221,7 @@ def test_wrong_endpoint(mock_get_datasource, test_client, login_as_admin, datase
     dataset.default_endpoint = "another_endpoint"
     mock_get_datasource.return_value = dataset
     resp = test_client.get(
-        f"api/v1/explore/?dataset_id={dataset.id}&dataset_type={dataset.type}"
+        f"api/v1/explore/?datasource_id={dataset.id}&datasource_type={dataset.type}"
     )
     data = json.loads(resp.data.decode("utf-8"))
     assert resp.status_code == 302