Separated managed identity and service principal tests

Author: vladvildanov

.github/workflows/tests.yml

@@ -32,7 +32,7 @@ jobs:
         pip install -r dev_requirements.txt
     - name: Run tests with Python version ${{ matrix.python-version }}
       run: |
-        pytest --junitxml=test-results.xml
+        pytest --junitxml=test-results.xml --extra-markers="not managed_identity"
     - name: Upload test results
       if: always()
       uses: actions/upload-artifact@v4

pytest.ini

@@ -1,2 +1,4 @@
 [pytest]
-asyncio_mode = auto
+asyncio_mode = auto
+markers =
+    managed_identity: Tests that should be run on Azure VM to be able to reach managed identity service.

tests/test_cred_provider.py

@@ -16,6 +16,18 @@ class TestEntraIdCredentialsProvider:
             {
                 "idp_kwargs": {"auth_type": AuthType.SERVICE_PRINCIPAL},
             },
+        ],
+        ids=["Service principal"],
+        indirect=True,
+    )
+    def test_get_credentials(self, credential_provider: EntraIdCredentialsProvider):
+        credentials = credential_provider.get_credentials()
+        assert len(credentials) == 2
+
+
+    @pytest.mark.parametrize(
+        "credential_provider",
+        [
             {
                 "idp_kwargs": {"auth_type": AuthType.MANAGED_IDENTITY},
             },
@@ -26,20 +38,35 @@ class TestEntraIdCredentialsProvider:
                 },
             }
         ],
-        ids=["Service principal", "Managed Identity (System-assigned)", "Managed Identity (User-assigned)"],
+        ids=["Managed Identity (System-assigned)", "Managed Identity (User-assigned)"],
         indirect=True,
     )
-    def test_get_credentials(self, credential_provider: EntraIdCredentialsProvider):
+    @pytest.mark.managed_identity
+    def test_get_credentials_managed_identity(self, credential_provider: EntraIdCredentialsProvider):
         credentials = credential_provider.get_credentials()
         assert len(credentials) == 2
 
+
     @pytest.mark.parametrize(
         "credential_provider",
         [
             {
                 "cred_provider_kwargs": {"block_for_initial": False},
                 "idp_kwargs": {"auth_type": AuthType.SERVICE_PRINCIPAL},
             },
+        ],
+        ids=["Service principal"],
+        indirect=True,
+    )
+    @pytest.mark.asyncio
+    async def test_get_credentials_async(self, credential_provider: EntraIdCredentialsProvider):
+        credentials = await credential_provider.get_credentials_async()
+        assert len(credentials) == 2
+
+
+    @pytest.mark.parametrize(
+        "credential_provider",
+        [
             {
                 "cred_provider_kwargs": {"block_for_initial": True},
                 "idp_kwargs": {"auth_type": AuthType.MANAGED_IDENTITY},
@@ -51,14 +78,16 @@ def test_get_credentials(self, credential_provider: EntraIdCredentialsProvider):
                 },
             }
         ],
-        ids=["Service principal", "Managed Identity (System-assigned)", "Managed Identity (User-assigned)"],
+        ids=["Managed Identity (System-assigned)", "Managed Identity (User-assigned)"],
         indirect=True,
     )
     @pytest.mark.asyncio
-    async def test_get_credentials_async(self, credential_provider: EntraIdCredentialsProvider):
+    @pytest.mark.managed_identity
+    async def test_get_credentials_async_managed_identity(self, credential_provider: EntraIdCredentialsProvider):
         credentials = await credential_provider.get_credentials_async()
         assert len(credentials) == 2
 
+
     @pytest.mark.parametrize(
         "credential_provider",
         [