edit

Author: Ugonnaak1

tests/broker-test.py

@@ -18,7 +18,6 @@
 import msal
 import getpass
 import os
-import warnings
 try:
     from dotenv import load_dotenv  # Use this only in local dev machine
     load_dotenv()  # take environment variables from .env.
@@ -75,10 +74,12 @@ def test_broker_username_password(scopes, expected_token_type):
     assert username and password, "You need to provide a test account and its password"
     result = pca.acquire_token_by_username_password(username, password, scopes)
     _assert(result, expected_token_type)
+    assert result.get("token_source") == "broker"
     print("Username password test succeeds.")
 
 def _assert(result, expected_token_type):
     assert result.get("access_token"), f"We should obtain a token. Got {result} instead."
+    assert result.get("token_source") == "broker", "Token should be obtained via broker"
     assert result.get("token_type").lower() == expected_token_type.lower(), f"{expected_token_type} not found"
 
 for i in range(2):  # Mimic Azure CLI's issue report