[IMP] Use create_case in graphql tests

dfir-iris · Oct 2, 2024 · 4e8ca75 · 4e8ca75
1 parent 957503a
commit 4e8ca75
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/tests/tests_graphql.py b/tests/tests_graphql.py
@@ -361,9 +361,7 @@ def test_graphql_case_should_return_error_log_uuid_when_permission_denied(self):
 
     def test_graphql_case_should_return_error_ioc_when_permission_denied(self):
         user = self._subject.create_dummy_user()
-        payload = {'query': ' mutation { caseCreate(name: "case", description: "Some description", clientId: 1) { case { caseId } } }'}
-        body = self._subject.execute_graphql_query(payload)
-        case_identifier = body['data']['caseCreate']['case']['caseId']
+        case_identifier = self._create_case()
         ioc_value = self._generate_new_dummy_ioc_value()
         payload = {
             'query': f'''mutation {{
@@ -613,10 +611,7 @@ def test_graphql_update_case_with_optional_parameter_reviewStatusId(self):
         self.assertEqual(1, body['data']['caseUpdate']['case']['reviewStatusId'])
 
     def test_graphql_query_ioc_should_not_fail(self):
-        payload = {'query': 'mutation { caseCreate(name: "case2", description: "Some description", clientId: 1, socId: "1", classificationId : 1) {case { '
-                            'caseId } } }'}
-        body = self._subject.execute_graphql_query(payload)
-        case_identifier = body['data']['caseCreate']['case']['caseId']
+        case_identifier = self._create_case()
         ioc_value = self._generate_new_dummy_ioc_value()
         payload = {
             'query': f'''mutation {{