def test_process_message(self, process, message_indicates_finish,
get_source_from_user_and_url,
get_import_url_from_message_body):
arbitrary_body = 'I am the very model of a modern major general'
arbitrary_url = 'http://go.com/'
arbitrary_source = models.LocationSource.objects.create(
name='Whatnot',
user=self.user,
type=self.source_type,
)
ok_message = Message.objects.create(
mailbox=self.mailbox,
subject='Whatever',
from_header=self.arbitrary_email,
body=arbitrary_body,
)
get_import_url_from_message_body.return_value = (arbitrary_url)
get_source_from_user_and_url.return_value = (arbitrary_source)
message_indicates_finish.return_value = (False)
RunmeterConsumer.process_message(ok_message)
RunmeterConsumer.get_import_url_from_message_body.assert_called_with(
arbitrary_body, )
RunmeterConsumer.get_source_from_user_and_url.assert_called_with(
self.settings.user,
arbitrary_url,
)
def process_incoming_runmeter_message(sender, message, **kwargs):
from location.consumers.runmeter import RunmeterConsumer
if message.mailbox.name == SETTINGS['runmeter_mailbox']:
try:
RunmeterConsumer.process_message(message)
except LocationConsumerSettings.DoesNotExist:
logger.warning(
'Unable to process message \'%s\': '
'No user is currently assigned to from_address %s',
message.from_address
)
def test_process_message(
self, process, message_indicates_finish, get_source_from_user_and_url, get_import_url_from_message_body
):
arbitrary_body = "I am the very model of a modern major general"
arbitrary_url = "http://go.com/"
arbitrary_source = models.LocationSource.objects.create(name="Whatnot", user=self.user, type=self.source_type)
ok_message = Message.objects.create(
mailbox=self.mailbox, subject="Whatever", from_header=self.arbitrary_email, body=arbitrary_body
)
get_import_url_from_message_body.return_value = arbitrary_url
get_source_from_user_and_url.return_value = arbitrary_source
message_indicates_finish.return_value = False
RunmeterConsumer.process_message(ok_message)
RunmeterConsumer.get_import_url_from_message_body.assert_called_with(arbitrary_body)
RunmeterConsumer.get_source_from_user_and_url.assert_called_with(self.settings.user, arbitrary_url)
