class TestLineExtensionService(unittest.TestCase):
def setUp(self):
self.old_service = Mock()
self.line_dao = Mock()
self.extension_dao = Mock()
self.service = LineExtensionService(self.old_service, self.line_dao, self.extension_dao)
def test_when_listing_associations_then_service_is_called(self):
expected_line_extensions = self.old_service.get_all_by_line_id.return_value
result = self.service.list(sentinel.line_id)
self.old_service.get_all_by_line_id.assert_called_once_with(sentinel.line_id)
assert_that(result, equal_to(expected_line_extensions))
def test_when_getting_association_then_model_built(self):
result = self.service.get(sentinel.line_id, sentinel.extension_id)
assert_that(result.line_id, equal_to(sentinel.line_id))
assert_that(result.extension_id, equal_to(sentinel.extension_id))
def test_when_associating_then_service_is_called(self):
association = Mock()
self.service.associate(association)
self.old_service.associate.assert_called_once_with(association)
def test_when_dissociating_then_service_is_called(self):
association = Mock()
self.service.dissociate(association)
self.old_service.dissociate.assert_called_once_with(association)
def test_when_getting_parent_then_service_is_called(self):
expected_line_voicemail = self.old_service.get_by_line_id.return_value
result = self.service.get_by_parent(sentinel.line_id)
self.old_service.get_by_line_id.assert_called_once_with(sentinel.line_id)
assert_that(result, equal_to(expected_line_voicemail))
def test_when_getting_extension_then_service_is_called(self):
expected_line_extension = self.old_service.get_by_extension_id.return_value
result = self.service.get_by_extension_id(sentinel.extension_id)
self.old_service.get_by_extension_id.assert_called_once_with(sentinel.extension_id)
assert_that(result, equal_to(expected_line_extension))
def setUp(self):
self.old_service = Mock()
self.line_dao = Mock()
self.extension_dao = Mock()
self.service = LineExtensionService(self.old_service, self.line_dao, self.extension_dao)
