def test_send_message_dap(self, mock_publish_data,
mock_create_message_data):
self.meta_data.output_type = OutputType.DAP
self.meta_data.tx_id = "9010576d-f3df-4011-aa41-adecd9bee011"
str_dap_message = json.dumps(self.expected)
path = "dap/"
mock_create_message_data.return_value = str_dap_message
send_message(self.meta_data, path)
mock_publish_data.assert_called_with(str_dap_message,
self.meta_data.tx_id, path)
def deliver(meta_data: MetaWrapper, data_bytes: bytes):
"""
Encrypts any unencrypted data, writes to the appropriate location within the outputs GCP bucket and notifies DAP
via PubSub
"""
if meta_data.output_type == OutputType.SEFT:
encrypted_output = data_bytes
else:
logger.info("Encrypting output")
encrypted_output = encrypt_output(data_bytes)
logger.info("Storing to bucket")
path = write_to_bucket(encrypted_output,
filename=meta_data.filename,
output_type=meta_data.output_type)
logger.info("Sending DAP notification")
send_message(meta_data, path)