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)