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)
