Modify tests depending on environment

Author: paulnoirel

libs/labelbox/tests/integration/test_api_keys.py

@@ -1,29 +1,34 @@
 import uuid
 import pytest
+import os
 
 from labelbox.schema.timeunit import TimeUnit
 from labelbox.schema.api_key import ApiKey
 from lbox.exceptions import LabelboxError
 # The creation of API keys requires a feature flag to be enabled.
 
 
-def test_create_api_key_success(admin_client):
+@pytest.mark.skipif(
+    condition=os.environ["LABELBOX_TEST_ENVIRON"] != "prod",
+    reason="Admin permissions are required to create API keys",
+)
+def test_create_api_key_success(client):
     # Create a test API key
     key_name = f"Test Key {uuid.uuid4()}"
-    user_email = admin_client.get_user().email
+    user_email = client.get_user().email
 
     assert (
-        admin_client.get_user().org_role().name == "Admin"
+        client.get_user().org_role().name == "Admin"
     ), "User must be an admin to create API keys"
 
     # Get available roles and use the first one
-    available_roles = ApiKey._get_available_api_key_roles(admin_client)
+    available_roles = ApiKey._get_available_api_key_roles(client)
     assert (
         len(available_roles) > 0
     ), "No available roles found for API key creation"
 
     # Create the API key with a short validity period
-    api_key_result = admin_client.create_api_key(
+    api_key_result = client.create_api_key(
         name=key_name,
         user=user_email,
         role=available_roles[0],
@@ -223,7 +228,10 @@ def test_create_api_key_invalid_time_unit(client):
         )
     assert "valid TimeUnit" in str(excinfo.value)
 
-
+@pytest.mark.skipif(
+    condition=os.environ["LABELBOX_TEST_ENVIRON"] == "prod",
+    reason="Accounts with sdmin permission can create API keys",
+)
 def test_create_api_key_insufficient_permissions(client):
     """Test that creating an API key fails when the user has insufficient permissions."""
     user_email = client.get_user().email