def test_common_manifest_with_id(self):
message = self.FakeMessage(
event_type=EVENT_MAN_RECEIVED,
timestamp=datetime.now().isoformat(),
user_identifier='*****@*****.**',
metadata={
'sample_custodian': '*****@*****.**',
'manifest_id': 1234,
'deputies': ['*****@*****.**', '*****@*****.**']
},
notifier_info={
'work_plan_id': 1,
'drs_study_code': 1234
})
rule = Rule(env='test', config=config, message=message)
rule._generate_manifest_link = Mock()
rule._generate_manifest_link.return_value = ''
to, data = rule._common_manifest()
self.assertEqual(to, [
message.user_identifier, message.metadata['sample_custodian'],
'*****@*****.**', '*****@*****.**'
]),
self.assertEqual(data, {
'manifest_id': message.metadata['manifest_id'],
'link': ''
})
def test_on_work_order_dispatched_no_drs_study_code(self, mocked_notify):
drs_study_code = 'nil'
message = self.create_fake_generic_work_order_message(
EVENT_WO_DISPATCHED, drs_study_code)
rule = Rule(env='test', config=config, message=message)
rule.check_rules()
self.assertIsInstance(mocked_notify, Notify)
subject = "{0} {1} {2} [Data release:{3}]".format(
SBJ_PREFIX_WO, message.metadata['work_order_id'], 'Dispatched',
message.notifier_info['drs_study_code'])
data = {
'user_identifier': '*****@*****.**',
'link':
self._generate_wo_link(message.notifier_info['work_plan_id']),
'work_order_status': 'dispatched',
'work_order_id': message.metadata['work_order_id']
}
mocked_notify.return_value.send_email.assert_called_once_with(
subject=subject,
from_address=config.email.from_address,
template='wo_event',
to=['*****@*****.**'],
data=data)
def test_common_manifest_called(self, mocked_notify):
message = self.create_fake_generic_manifest_message(EVENT_MAN_RECEIVED)
rule = Rule(env='test', config=config, message=message)
rule._common_manifest = Mock()
rule._common_manifest.return_value = [], {}
rule.check_rules()
rule._common_manifest.assert_called_once()
def test_generate_manifest_link(self):
rule = Rule(env='test', config=config, message='')
path = 'path'
id = 'id'
link = rule._generate_manifest_link(path, id)
self.assertEqual(
link,
'{}://{}:{}/{}/{}'.format(config.link.protocol, config.link.root,
config.link.port, path, id))
def test_common_work_order_called(self, mocked_notify):
message = self.create_fake_generic_work_order_message(
EVENT_WO_DISPATCHED, 1234)
rule = Rule(env='test', config=config, message=message)
rule._common_work_order = Mock()
rule._common_work_order.return_value = [], {}
rule.check_rules()
rule._common_work_order.assert_called_once()
def test_generate_wo_link(self):
rule = Rule(env='test', config=config, message='')
work_plan_id = '1234'
link = rule._generate_wo_link(work_plan_id)
self.assertEqual(
link,
'{}://{}:{}/{}/{}/{}'.format(config.link.protocol,
config.link.root, config.link.port,
PATH_WORK_ORDER_BEGIN, work_plan_id,
PATH_WORK_ORDER_END))
def test_on_catalogue_processed(self, mocked_notify):
rule = Rule(env='test', config=config, message='')
rule._on_catalogue_processed()
mocked_notify.return_value.send_email.assert_called_once_with(
subject=SBJ_CAT_PROCESSED,
from_address=config.email.from_address,
to=[config.contact.email_dev_team],
template='catalogue_processed',
data={})
def test_common_work_order_without_id(self):
message = self.FakeMessage(
event_type=EVENT_WO_DISPATCHED,
timestamp=datetime.now().isoformat(),
user_identifier='*****@*****.**',
metadata={'sample_custodian': '*****@*****.**'},
notifier_info={
'work_plan_id': 1,
'drs_study_code': 1234
})
rule = Rule(env='test', config=config, message=message)
with self.assertRaises(ValueError) as cm:
rule._common_work_order()
def test_init(self):
rule = Rule(env='test_env',
config='test_config',
message=self._fake_message)
self.assertEqual(rule._env, 'test_env')
self.assertEqual(rule._config, 'test_config')
self.assertEqual(rule._message, self._fake_message)
self.assertIsInstance(rule._notify, Notify)
def test_on_catalogue_rejected(self, mocked_notify):
message = self.FakeMessage(event_type=EVENT_CAT_REJECTED,
timestamp=datetime.now().isoformat(),
user_identifier='*****@*****.**',
metadata={'error': 'error message'},
notifier_info={})
rule = Rule(env='test', config=config, message=message)
rule._on_catalogue_rejected()
mocked_notify.return_value.send_email.assert_called_once_with(
subject=SBJ_CAT_REJECTED,
from_address=config.email.from_address,
to=[config.contact.email_dev_team],
template='catalogue_rejected',
data={
'error': message.metadata['error'],
'timestamp': message.timestamp
})
def test_manifest_received_triggered(self):
message = self.create_fake_generic_manifest_message(EVENT_MAN_RECEIVED)
rule = Rule(env='test', config='test', message=message)
rule._on_manifest_create = Mock()
rule._on_manifest_received = Mock()
rule._on_work_order_event = Mock()
rule._on_catalogue_new = Mock()
rule._on_catalogue_processed = Mock()
rule._on_catalogue_rejected = Mock()
rule.check_rules()
rule._on_manifest_create.assert_not_called()
rule._on_manifest_received.assert_called_once()
rule._on_work_order_event.assert_not_called()
rule._on_catalogue_new.assert_not_called()
rule._on_catalogue_processed.assert_not_called()
rule._on_catalogue_rejected.assert_not_called()
def test_on_manifest_received(self, mocked_notify):
message = self.create_fake_generic_manifest_message(EVENT_MAN_RECEIVED)
rule = Rule(env='test', config=config, message=message)
rule.check_rules()
self.assertIsInstance(mocked_notify, Notify)
subject = SBJ_MAN_RECEIVED + ' ' + str(message.metadata['manifest_id'])
data = {
'manifest_id': message.metadata['manifest_id'],
'link':
self._generate_manifest_link(message.metadata['manifest_id'])
}
mocked_notify.return_value.send_email.assert_called_once_with(
subject=subject,
from_address=config.email.from_address,
template='manifest_received',
to=['*****@*****.**', '*****@*****.**'],
data=data)
def test_on_manifest_create_with_hmdmc(self, mocked_notify):
message = self.FakeMessage(event_type=EVENT_MAN_CREATED,
timestamp=datetime.now().isoformat(),
user_identifier='*****@*****.**',
metadata={
'sample_custodian': '*****@*****.**',
'manifest_id': 123,
'hmdmc': 'abc321'
},
notifier_info={})
rule = Rule(env='test', config=config, message=message)
rule.check_rules()
self.assertIsInstance(mocked_notify, Notify)
self.assertEqual(mocked_notify.return_value.send_email.call_count, 2)
exp1 = call(data={
'hmdmc_list': 'abc321',
'manifest_id': 123,
'link':
'http://aker.localhost:80/reception/material_submissions/123',
'user_identifier': '*****@*****.**'
},
from_address=u'*****@*****.**',
subject='Aker | Manifest Created 123',
template='manifest_created',
to=['*****@*****.**', '*****@*****.**'])
exp2 = call(data={
'hmdmc_list': 'abc321',
'manifest_id': 123,
'link':
'http://aker.localhost:80/reception/material_submissions/123',
'user_identifier': '*****@*****.**'
},
from_address=u'*****@*****.**',
subject='Aker | Manifest Created with HMDMC 123',
template='manifest_created_hmdmc',
to=[u'*****@*****.**'])
mocked_notify.return_value.send_email.assert_has_calls([exp1, exp2],
any_order=True)
def test_work_order_concluded_event_triggered(self):
drs_study_code = 1234
message = self.create_fake_generic_work_order_message(
EVENT_WO_CONCLUDED, drs_study_code)
rule = Rule(env='test', config='test', message=message)
rule._on_manifest_create = Mock()
rule._on_manifest_received = Mock()
rule._on_work_order_event = Mock()
rule._on_catalogue_new = Mock()
rule._on_catalogue_processed = Mock()
rule._on_catalogue_rejected = Mock()
rule.check_rules()
rule._on_manifest_create.assert_not_called()
rule._on_manifest_received.assert_not_called()
rule._on_work_order_event.assert_called_once()
rule._on_catalogue_new.assert_not_called()
rule._on_catalogue_processed.assert_not_called()
rule._on_catalogue_rejected.assert_not_called()
def test_common_work_order_with_id(self):
message = self.FakeMessage(event_type=EVENT_WO_DISPATCHED,
timestamp=datetime.now().isoformat(),
user_identifier='*****@*****.**',
metadata={
'sample_custodian': '*****@*****.**',
'work_order_id': 1234
},
notifier_info={
'work_plan_id': 1,
'drs_study_code': 1234
})
rule = Rule(env='test', config=config, message=message)
rule._generate_wo_link = Mock()
rule._generate_wo_link.return_value = ''
to, data = rule._common_work_order()
self.assertEqual(to, [message.user_identifier])
self.assertEqual(data, {
'work_order_id': message.metadata['work_order_id'],
'link': ''
})
def test_common_catalogue(self):
rule = Rule(env='test', config=config, message='')
self.assertEqual(rule._common_catalogue(),
[config.contact.email_dev_team])