def test_get_group_id_errors(self, mock_log): test_membership = Membership(self.CERT, self.URLBASE) # IndexError test_body = {u'response': {u'docs': []}} self._register_get_group(body=test_body) with self.assertRaises(PyLmodUnexpectedData): test_membership.get_group_id(self.CUUID) mock_log.exception.assert_called_with( "Error in get_group response data - " "got {u'response': {u'docs': []}}" ) # KeyError test_body = {u'response': {u'foo': []}} self._register_get_group(body=test_body) with self.assertRaises(PyLmodUnexpectedData): test_membership.get_group_id(self.CUUID) mock_log.exception.assert_called_with( "Error in get_group response data - " "got {u'response': {u'foo': []}}" )
def test_get_group_id(self): self._register_get_group(body=self.COURSE_DATA) test_membership = Membership(self.CERT, self.URLBASE) group_id = test_membership.get_group_id(self.CUUID) assert group_id == self.COURSE_DATA['response']['docs'][0]['id']