update

Author: fengga

tests/broker-test.py

@@ -6,6 +6,7 @@
 we can use this script to test it with a given version of MSAL Python.
 """
 import msal
+import getpass
 import os
 try:
     from dotenv import load_dotenv  # Use this only in local dev machine
@@ -54,10 +55,9 @@ def interactive_and_silent(scopes, auth_scheme, data, expected_token_type):
 
 def test_broker_username_password(scopes, expected_token_type):
     print("Testing broker username password flows by using accounts in local .env")
-    username = os.getenv("ENV_LAB4_ACCOUNT")
-    password = os.getenv("ENV_LAB4_ACCOUNT_PASSWORD")
-    assert(username!=None and len(username)>0)
-    assert(password!=None and len(password)>0)
+    username = os.getenv("BROKER_TEST_ACCOUNT") or input("Input test account for broker test: ")
+    password = os.getenv("BROKER_TEST_ACCOUNT_PASSWORD") or getpass.getpass("Input test account's password: ")
+    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")