Add tests for bulk aggregation

equinor · Feb 8, 2024 · af17a97 · af17a97
1 parent 47885ca
commit af17a97
Show file tree

Hide file tree

Showing 4 changed files with 79 additions and 4 deletions.
diff --git a/tests/test_access/tst_access_drogon_manage_login.py b/tests/test_access/tst_access_drogon_manage_login.py
@@ -92,6 +92,26 @@ def test_read_restricted_classification_data(explorer: Explorer):
     print("Hits on restricted:", hits)
     assert hits > 0 
 
-# TODO: aggregate bulk operation should fail 
+
+def test_aggregate_bulk(explorer: Explorer):
+    """Test a bulk aggregation method"""
+    print("Running test:", inspect.currentframe().f_code.co_name)
+    cases = explorer.cases
+    print("Number of cases: ", len(cases))
+    assert len(cases) > 0
+    case = cases[0]
+    case_uuid = case.metadata.get("fmu").get("case").get("uuid")
+    print("About to trigger aggregation on case", case_uuid)
+    body = {
+        "operations": [ "min" ],
+        "case_uuid": case_uuid, 
+        "class": "surface", 
+        "iteration_name": "iter-0"
+    }
+    response = explorer._sumo.post(f"/aggregations", json=body)
+    print(response.status_code)
+    assert response.status_code == 200
+    assert response.json().get("batchType") == "batch"
+
 
 # TODO: FAST aggregate operation should succeed
diff --git a/tests/test_access/tst_access_drogon_read_login.py b/tests/test_access/tst_access_drogon_read_login.py
@@ -93,6 +93,25 @@ def test_read_restricted_classification_data(explorer: Explorer):
     print("Hits on restricted:", hits)
     assert hits == 0
 
-# TODO: aggregate bulk operation should fail 
+def test_aggregate_bulk(explorer: Explorer):
+    """Test a bulk aggregation method"""
+    print("Running test:", inspect.currentframe().f_code.co_name)
+    cases = explorer.cases
+    print("Number of cases: ", len(cases))
+    assert len(cases) > 0
+    case = cases[0]
+    case_uuid = case.metadata.get("fmu").get("case").get("uuid")
+    print("About to trigger aggregation on case", case_uuid)
+    body = {
+        "operations": [ "min" ],
+        "case_uuid": case_uuid, 
+        "class": "surface", 
+        "iteration_name": "iter-0"
+    }
+    response = explorer._sumo.post(f"/aggregations", json=body)
+    print(response.status_code)
+    assert response.status_code == 200
+    assert response.json().get("batchType") == "batch"
+    # TODO SHould fail
 
 # TODO: FAST aggregate operation should succeed
diff --git a/tests/test_access/tst_access_drogon_write_login.py b/tests/test_access/tst_access_drogon_write_login.py
@@ -92,6 +92,25 @@ def test_read_restricted_classification_data(explorer: Explorer):
     print("Hits on restricted:", hits)
     assert hits > 0 
 
-# TODO: aggregate bulk operation should fail 
+def test_aggregate_bulk(explorer: Explorer):
+    """Test a bulk aggregation method"""
+    print("Running test:", inspect.currentframe().f_code.co_name)
+    cases = explorer.cases
+    print("Number of cases: ", len(cases))
+    assert len(cases) > 0
+    case = cases[0]
+    case_uuid = case.metadata.get("fmu").get("case").get("uuid")
+    print("About to trigger aggregation on case", case_uuid)
+    body = {
+        "operations": [ "min" ],
+        "case_uuid": case_uuid, 
+        "class": "surface", 
+        "iteration_name": "iter-0"
+    }
+    response = explorer._sumo.post(f"/aggregations", json=body)
+    print(response.status_code)
+    assert response.status_code == 200
+    assert response.json().get("batchType") == "batch"
 
+
 # TODO: FAST aggregate operation should succeed
diff --git a/tests/test_access/tst_access_no_access_login.py b/tests/test_access/tst_access_no_access_login.py
@@ -99,6 +99,23 @@ def test_read_restricted_classification_data(explorer: Explorer):
     assert hits == 0
 
 
-# TODO: aggregate bulk operation should fail
+def test_aggregate_bulk(explorer: Explorer):
+    """Test a bulk aggregation method"""
+    print("Running test:", inspect.currentframe().f_code.co_name)
+    # Fixed test case ("Drogon_AHM_2023-02-22") in Sumo/DEV
+    TESTCASE_UUID = "10f41041-2c17-4374-a735-bb0de62e29dc"
+    print("About to trigger aggregation on case", TESTCASE_UUID)
+    body = {
+        "operations": [ "min" ],
+        "case_uuid": TESTCASE_UUID, 
+        "class": "surface", 
+        "iteration_name": "iter-0"
+    }
+    response = explorer._sumo.post(f"/aggregations", json=body)
+    print(response.status_code)
+    assert response.status_code == 200
+    assert response.json().get("batchType") == "batch"
+    # TODO: Should fail
+
 
 # TODO: FAST aggregate operation should succeed