[Identity] Fix device code tests (#35846)

A recent change in MSAL made it so it is required that
"device_authorization_endpoint" is set in the discovery response.

This populates the value in the mock discovery response.

Signed-off-by: Paul Van Eck <[email protected]>

Author: pvaneck

sdk/identity/azure-identity/tests/helpers.py

@@ -162,7 +162,12 @@ def get_discovery_response(endpoint="https://a/b"):
     sufficient. MSAL will send token requests to "{endpoint}/oauth2/v2.0/token_endpoint" after receiving a tenant
     discovery response created by this method.
     """
-    aad_metadata_endpoint_names = ("authorization_endpoint", "token_endpoint", "tenant_discovery_endpoint")
+    aad_metadata_endpoint_names = (
+        "authorization_endpoint",
+        "token_endpoint",
+        "tenant_discovery_endpoint",
+        "device_authorization_endpoint",
+    )
     payload = {name: endpoint + "/oauth2/v2.0/" + name for name in aad_metadata_endpoint_names}
     payload["metadata"] = ""
     return mock_response(json_payload=payload)

sdk/identity/azure-identity/tests/test_device_code_credential.py

@@ -178,7 +178,7 @@ def test_device_code_credential():
         requests=[Request()] * 3,  # not validating requests because they're formed by MSAL
         responses=[
             # expected requests: discover tenant, start device code flow, poll for completion
-            mock_response(json_payload={"authorization_endpoint": "https://a/b", "token_endpoint": "https://a/b"}),
+            get_discovery_response(),
             mock_response(
                 json_payload={
                     "device_code": "_",
@@ -237,7 +237,7 @@ def test_tenant_id():
         requests=[Request()] * 3,  # not validating requests because they're formed by MSAL
         responses=[
             # expected requests: discover tenant, start device code flow, poll for completion
-            mock_response(json_payload={"authorization_endpoint": "https://a/b", "token_endpoint": "https://a/b"}),
+            get_discovery_response(),
             mock_response(
                 json_payload={
                     "device_code": "_",