def test_get_message_mime(self, authenticate_mock, get_mock):
""" Test Get Message Mime"""
print("Test Get Message Mime\n")
content = {"displayName": "Tester"}
try:
authenticate_mock.return_value = True
MS_graph_helper = MSGraphHelper(
MOCKED_OPTS.get("microsoft_graph_token_url"),
MOCKED_OPTS.get("microsoft_graph_url"),
MOCKED_OPTS.get("tenant_id"), MOCKED_OPTS.get("client_id"),
MOCKED_OPTS.get("client_secret"),
MOCKED_OPTS.get("max_messages"), MOCKED_OPTS.get("max_users"),
MOCKED_OPTS.get("max_retries_total"),
MOCKED_OPTS.get("max_retries_backoff_factor"),
MOCKED_OPTS.get("max_batched_requests"), None)
get_mock.return_value = generate_response(content, 200)
response = MS_graph_helper.get_message_mime(
"*****@*****.**", "AAA")
assert response.status_code == 200
assert response.content["displayName"] == "Tester"
get_mock.return_value = generate_response(content, 404)
response = MS_graph_helper.get_message_mime(
"*****@*****.**", "AAA")
assert response.status_code == 404
assert response.content["displayName"] == "Tester"
get_mock.return_value = generate_response(content, 300)
response = MS_graph_helper.get_message_mime(
"*****@*****.**", "AAA")
except IntegrationError as err:
assert True
def _exchange_online_write_message_as_attachment_function(
self, event, *args, **kwargs):
"""Function: This function will get the mime content of an Exchange Online message and write it as an
attachment."""
try:
# Initialize the results payload
rp = ResultPayload(CONFIG_DATA_SECTION, **kwargs)
# Validate fields
validate_fields(
['incident_id', 'exo_email_address', 'exo_messages_id'],
kwargs)
# Get the function parameters:
incident_id = kwargs.get("incident_id") # number
task_id = kwargs.get("task_id") # number
email_address = kwargs.get("exo_email_address") # text
message_id = kwargs.get("exo_messages_id") # text
attachment_name = kwargs.get("exo_attachment_name") # text
LOG.info(u"incident_id: %s", incident_id)
LOG.info(u"task_id: %s", task_id)
LOG.info(u"exo_email_address: %s", email_address)
LOG.info(u"exo_messages_id: %s", message_id)
LOG.info(u"exo_attachment_name: %s", attachment_name)
yield StatusMessage(
u"Starting to get message mime for email address: {}".format(
email_address))
# Get the MS Graph helper class
MS_graph_helper = MSGraphHelper(
self.options.get("microsoft_graph_token_url"),
self.options.get("microsoft_graph_url"),
self.options.get("tenant_id"), self.options.get("client_id"),
self.options.get("client_secret"),
self.options.get("max_messages"),
self.options.get("max_users"),
RequestsCommon(self.opts, self.options).get_proxies())
# Call MS Graph API to get the user profile
response = MS_graph_helper.get_message_mime(
email_address, message_id)
datastream = BytesIO(response.content)
if attachment_name is None:
attachment_name = u"message-{}-{}.eml".format(
email_address, message_id)
LOG.info(u"attachment_name: %s", attachment_name)
# Get the rest client so we can add the attachment to the incident.
rest_client = self.rest_client()
# Write the file as attachement: failures will raise an exception
write_file_attachment(rest_client, attachment_name, datastream,
incident_id, task_id)
results_data = {"attachment_name": attachment_name}
results = rp.done(True, results_data)
yield StatusMessage(
u"Returning results for get message mime for email address: {0}\n attachment name: {1}"
.format(email_address, attachment_name))
# Produce a FunctionResult with the results
yield FunctionResult(results)
except Exception as err:
LOG.error(err)
yield FunctionError(err)