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