def test_collect_audit_emails(self):
"""Test _collect_audit_emails function."""
audit_captains = factories.PersonFactory.create_batch(3)
auditors = factories.PersonFactory.create_batch(3)
audit_captain_role_id = all_models.AccessControlRole.query.filter(
all_models.AccessControlRole.name == "Audit Captains",
all_models.AccessControlRole.object_type == "Audit",
).one().id
auditor_role_id = all_models.AccessControlRole.query.filter(
all_models.AccessControlRole.name == "Auditors",
all_models.AccessControlRole.object_type == "Audit",
).one().id
acl_data = self._prepare_acl({
audit_captain_role_id: audit_captains,
auditor_role_id: auditors
})
with app.app_context():
# pylint: disable=protected-access
reporter_email, cc_list = assessment_integration._collect_audit_emails(
acl_data)
audit_reporter = audit_captains[0].email
audit_ccs = set(captain.email for captain in audit_captains[1:])
self.assertEquals(reporter_email, audit_reporter)
self.assertEquals(set(cc_list), audit_ccs)
def test_audit_emails_wh_captains(self):
"""Test _collect_audit_emails without Audit Captains."""
auditors = factories.PersonFactory.create_batch(3)
auditor_role_id = all_models.AccessControlRole.query.filter(
all_models.AccessControlRole.name == "Auditors",
all_models.AccessControlRole.object_type == "Audit",
).one().id
acl_data = self._prepare_acl({
auditor_role_id: auditors,
})
with app.app_context():
# pylint: disable=protected-access
reporter_email, cc_list = assessment_integration._collect_audit_emails(
acl_data)
self.assertEquals(reporter_email, "")
self.assertEquals(cc_list, [])