From d83a0d26502212ca032a133266475b2c1f9fe554 Mon Sep 17 00:00:00 2001 From: paulnoirel <87332996+paulnoirel@users.noreply.github.com> Date: Wed, 16 Apr 2025 00:37:42 +0100 Subject: [PATCH] Fix test_create_api_key_insufficient_permissions --- .../tests/integration/test_api_keys.py | 29 +++++++++---------- 1 file changed, 14 insertions(+), 15 deletions(-) diff --git a/libs/labelbox/tests/integration/test_api_keys.py b/libs/labelbox/tests/integration/test_api_keys.py index 531e10faa..2282994f8 100644 --- a/libs/labelbox/tests/integration/test_api_keys.py +++ b/libs/labelbox/tests/integration/test_api_keys.py @@ -231,23 +231,22 @@ def test_create_api_key_invalid_time_unit(client): @pytest.mark.skipif( condition=os.environ["LABELBOX_TEST_ENVIRON"] == "prod", - reason="Accounts with sdmin permission can create API keys", + reason="Accounts with admin 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 - assert client.get_user().org_role().name == "Admin" - - # Attempt to create another API key using the limited permissions client - # This should fail due to insufficient permissions - with pytest.raises(LabelboxError) as excinfo: - client.create_api_key( - name=f"Test Key {uuid.uuid4()}", - user=user_email, - role="Admin", - validity=5, - time_unit=TimeUnit.MINUTE, - ) - - assert "192" in str(excinfo.value) + if client.get_user().org_role().name != "Admin": + # Attempt to create another API key using the limited permissions client + # This should fail due to insufficient permissions + with pytest.raises(LabelboxError) as excinfo: + client.create_api_key( + name=f"Test Key {uuid.uuid4()}", + user=user_email, + role="Admin", + validity=5, + time_unit=TimeUnit.MINUTE, + ) + + assert "192" in str(excinfo.value)