class TestSingleLineExtensionResource(unittest.TestCase): def setUp(self): self.service = Mock() self.converter = Mock() self.resource = SingleLineExtensionResource(self.service, self.converter) def test_when_getting_an_extension_then_service_called(self, request, url_for): self.resource.get_by_extension(sentinel.parent_id) self.service.get_by_extension_id.assert_called_once_with(sentinel.parent_id) def test_when_getting_an_association_then_model_converted(self, request, url_for): expected_association = self.service.get_by_extension_id.return_value expected_response = self.converter.encode.return_value result = self.resource.get_by_extension(sentinel.parent_id) self.converter.encode.assert_called_once_with(expected_association) url_for.assert_called_once_with('.get_by_extension', extension_id=sentinel.parent_id) assert_that(result, equal_to((expected_response, 200, {'Content-Type': 'application/json', 'Location': url_for.return_value})))
def setUp(self): self.service = Mock() self.converter = Mock() self.resource = SingleLineExtensionResource(self.service, self.converter)