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']
