Skip to content

Commit 08bef72

Browse files
authored
Fix create API key test (#1970)
2 parents 777bfc0 + d83a0d2 commit 08bef72

File tree

1 file changed

+14
-15
lines changed

1 file changed

+14
-15
lines changed

libs/labelbox/tests/integration/test_api_keys.py

+14-15
Original file line numberDiff line numberDiff line change
@@ -231,23 +231,22 @@ def test_create_api_key_invalid_time_unit(client):
231231

232232
@pytest.mark.skipif(
233233
condition=os.environ["LABELBOX_TEST_ENVIRON"] == "prod",
234-
reason="Accounts with sdmin permission can create API keys",
234+
reason="Accounts with admin permission can create API keys",
235235
)
236236
def test_create_api_key_insufficient_permissions(client):
237237
"""Test that creating an API key fails when the user has insufficient permissions."""
238238
user_email = client.get_user().email
239239

240-
assert client.get_user().org_role().name == "Admin"
241-
242-
# Attempt to create another API key using the limited permissions client
243-
# This should fail due to insufficient permissions
244-
with pytest.raises(LabelboxError) as excinfo:
245-
client.create_api_key(
246-
name=f"Test Key {uuid.uuid4()}",
247-
user=user_email,
248-
role="Admin",
249-
validity=5,
250-
time_unit=TimeUnit.MINUTE,
251-
)
252-
253-
assert "192" in str(excinfo.value)
240+
if client.get_user().org_role().name != "Admin":
241+
# Attempt to create another API key using the limited permissions client
242+
# This should fail due to insufficient permissions
243+
with pytest.raises(LabelboxError) as excinfo:
244+
client.create_api_key(
245+
name=f"Test Key {uuid.uuid4()}",
246+
user=user_email,
247+
role="Admin",
248+
validity=5,
249+
time_unit=TimeUnit.MINUTE,
250+
)
251+
252+
assert "192" in str(excinfo.value)

0 commit comments

Comments
 (0)