Amend test permissions int to constants

EBI-Metagenomics · Apr 29, 2021 · 37538d4 · 37538d4
1 parent f4b3af6
commit 37538d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/api/test_permissions.py b/tests/api/test_permissions.py
@@ -98,7 +98,7 @@ def setup_method(self, db, pipelines, experiment_type, analysis_status):
         # public run (needeed for auth filtering)
         public_status = baker.make(
             "emgapi.Status",
-            pk=4,
+            pk=emg_models.Status.PUBLIC,
             status="public",
         )
         public_run = baker.make(
@@ -112,7 +112,7 @@ def setup_method(self, db, pipelines, experiment_type, analysis_status):
         # private (needeed for auth filtering)
         private_status = baker.make(
             "emgapi.Status",
-            pk=2,
+            pk=emg_models.Status.PRIVATE,
             status="private",
         )
         run_private = baker.make(