def test_email_has_role(self):
self._register_get_group(body=self.COURSE_DATA)
self._register_get_membership(body=self.MEMBERSHIP_DATA)
test_membership = Membership(self.CERT, self.URLBASE)
has_role = test_membership.email_has_role(
self.EMAIL, self.ROLE, uuid=self.CUUID
)
assert has_role is True
has_role = test_membership.email_has_role(
self.EMAIL, 'hacker', uuid=self.CUUID
)
assert has_role is False
def test_email_has_role_errors(self, mock_log):
test_membership = Membership(self.CERT, self.URLBASE)
# KeyError
test_body = {u'response': {u'foo': []}}
self._register_get_group(body=self.COURSE_DATA)
self._register_get_membership(body=test_body)
with self.assertRaises(PyLmodUnexpectedData):
test_membership.email_has_role(
self.EMAIL, self.ROLE, uuid=self.CUUID
)
mock_log.exception.assert_called_with(
"KeyError in membership data - "
"got {u'response': {u'foo': []}}"
)
def test_email_has_role_docs(self):
# len(docs) == 0
test_body = {u'response': {u'docs': []}}
self._register_get_group(body=self.COURSE_DATA)
self._register_get_membership(body=test_body)
test_membership = Membership(self.CERT, self.URLBASE)
has_role = test_membership.email_has_role(
self.EMAIL, self.ROLE, uuid=self.CUUID
)
assert has_role is False
