def test_federated_happy_path_and_correlation_id(self):
util.setup_expected_user_realm_response_common(True)
util.setup_expected_mex_wstrust_request_common()
response = util.create_response()
assertion = self.setup_expected_oauth_assertion_request(response)
buffer = StringIO()
handler = logging.StreamHandler(buffer)
util.turn_on_logging(level='DEBUG', handler=handler)
authorityUrl = response['authority']
context = adal.AuthenticationContext(authorityUrl)
correlation_id = '12300002-0000-0000-c000-000000000000'
context.correlation_id = correlation_id
#action
token_response = context.acquire_token_with_username_password(response['resource'], cp['username'], cp['password'], cp['clientId'])
self.assertTrue(util.is_match_token_response(response['cachedResponse'], token_response),
'Response did not match expected: ' + json.dumps(token_response))
#assert
log_content = buffer.getvalue()
self.assertTrue(correlation_id in log_content, 'Logging was turned on but no messages were recieved.')
self.assertNotIn(cp['clientId'], log_content, "Should not log ClientID")
self.assertNotIn(
cp['username'].split('@')[0], log_content, "Should not contain PII")
def test_federated_happy_path_and_correlation_id(self):
correlationId = '12300002-0000-0000-c000-000000000000'
util.set_correlation_id(correlationId)
util.setup_expected_user_realm_response_common(True)
util.setup_expected_mex_wstrust_request_common()
response = util.create_response()
assertion = self.setup_expected_oauth_assertion_request(response)
logFunctionCalled = False
def testCorrelationIdLog(level, message):
logFunctionCalled = True
self.assertIsNotNone(message)
logOptions = {
'level' : 3,
'log' : testCorrelationIdLog
}
oldOptions = log.get_logging_options()
log.set_logging_options(logOptions)
authorityUrl = response['authority'] + '/' + cp['tenant']
token_response = adal.acquire_token_with_username_password(authorityUrl, cp['username'], cp['password'], cp['clientId'], response['resource'])
self.assertTrue(util.isMatchTokenResponse(response['cachedResponse'], token_response), 'Response did not match expected: ' + JSON.stringify(token_response))
log.set_logging_options(oldOptions)
util.set_correlation_id()
self.assertTrue(util.isMatchTokenResponse(response['cachedResponse'], token_response), 'The response did not match what was expected')
self.assertTrue(logFunctionCalled, 'Logging was turned on but no messages were recieved.')
def test_federated_happy_path_and_correlation_id(self):
util.setup_expected_user_realm_response_common(True)
util.setup_expected_mex_wstrust_request_common()
response = util.create_response()
assertion = self.setup_expected_oauth_assertion_request(response)
buffer = StringIO()
handler = logging.StreamHandler(buffer)
util.turn_on_logging(level='DEBUG', handler=handler)
authorityUrl = response['authority']
context = adal.AuthenticationContext(authorityUrl)
correlation_id = '12300002-0000-0000-c000-000000000000'
context.correlation_id = correlation_id
#action
token_response = context.acquire_token_with_username_password(response['resource'], cp['username'], cp['password'], cp['clientId'])
self.assertTrue(util.is_match_token_response(response['cachedResponse'], token_response),
'Response did not match expected: ' + json.dumps(token_response))
#assert
log_content = buffer.getvalue()
self.assertTrue(correlation_id in log_content, 'Logging was turned on but no messages were recieved.')
