use a closure for client_id

.evergreen/auth_oidc/azure/test.py

@@ -1,4 +1,5 @@
 from pymongo import MongoClient
+from functools import partial
 import os
 import json
 from urllib.request import urlopen, Request
@@ -8,9 +9,8 @@
 _AUTH_MAP["MONGODB-OIDC"] = _authenticate_oidc
 
 app_id = os.environ['AZUREOIDC_CLIENTID']
-client_id = os.environ['AZUREOIDC_TOKENCLIENT']
 
-def callback(client_info, server_info):
+def callback(client_id, client_info, server_info):
  url = "http://169.254.169.254/metadata/identity/oauth2/token"
  url += "?api-version=2018-02-01"
  url += f"&resource=api://{app_id}"
@@ -41,8 +41,7 @@ def callback(client_info, server_info):
  raise ValueError(msg)
  return dict(access_token=data['access_token'])
 
-
-props = dict(request_token_callback=callback)
+props = dict(request_token_callback=partial(callback(os.environ['AZUREOIDC_TOKENCLIENT'])))
 print('Testing MONGODB-OIDC on azure...')
 print('Testing resource 1...')
 c = MongoClient('mongodb://localhost:27017/?authMechanism=MONGODB-OIDC', authMechanismProperties=props)
@@ -51,7 +50,7 @@ def callback(client_info, server_info):
 print('Testing resource 1... done.')
 
 print('Testing resource 2...')
-client_id = os.environ['AZUREOIDC_TOKENCLIENT2']
+props = dict(request_token_callback=partial(callback(os.environ['AZUREOIDC_TOKENCLIENT2'])))
 c = MongoClient('mongodb://localhost:27017/?authMechanism=MONGODB-OIDC', authMechanismProperties=props)
 c.test.test.find_one({})
 c.close()