Fix session notebook tests

Author: DarthMax

scripts/ci/aura_api_ci.py

@@ -48,9 +48,6 @@ def _update_token(self) -> AuraAuthToken:
 
         return AuraApiCI.AuraAuthToken(response.json())
 
-    def create_ds_instance(self, name: str) -> dict[str, Any]:
-        return self.create_instance(name, memory="8GB", type="gds")
-
     def create_instance(self, name: str, memory: str, type: str) -> dict[str, Any]:
         CREATE_OK_MAX_WAIT_TIME = 10
 
@@ -81,7 +78,13 @@ def create_instance(self, name: str, memory: str, type: str) -> dict[str, Any]:
             if should_retry:
                 logging.debug(f"Error code: {response.status_code} - Retrying in {wait_time} s")
 
-        return response.json()["data"]  # type: ignore
+
+        response_json = response.json()
+        if "errors" in response_json:
+            raise Exception(response_json["errors"])
+
+        return response_json["data"]  # type: ignore
+
 
     def check_running(self, db_id: str) -> None:
         RUNNING_MAX_WAIT_TIME = 60 * 5

scripts/ci/run_targeting_aura_sessions.py

@@ -22,7 +22,7 @@ def main() -> None:
     MAX_INT = 1000000
     instance_name = f"ci-build-{sys.argv[1]}" if len(sys.argv) > 1 else "ci-instance-" + str(rd.randint(0, MAX_INT))
 
-    create_result = aura_api.create_instance(instance_name, memory="1GB", type="professional-db")
+    create_result = aura_api.create_instance(instance_name, memory="2GB", type="enterprise-db")
     instance_id = create_result["id"]
     logging.info(f"Creation of database with id '{instance_id}'")
 
@@ -49,7 +49,7 @@ def handle_signal(sig: int, frame: Optional[FrameType]) -> None:
         else:
             project_id_part = ""
 
-        cmd = f"CLIENT_ID={client_id} CLIENT_SECRET={client_secret} {project_id_part} NEO4J_URI={uri} NEO4J_USERNAME={username} NEO4J_PASSWORD={password} tox -e jupyter-notebook-session-ci"
+        cmd = f"AURA_ENV=staging CLIENT_ID={client_id} CLIENT_SECRET={client_secret} {project_id_part} NEO4J_URI={uri} NEO4J_USERNAME={username} NEO4J_PASSWORD={password} tox -e jupyter-notebook-session-ci"
 
         if os.system(cmd) != 0:
             raise Exception("Failed to run notebooks")