def test_given_extension_does_not_exist_then_returns_nothing(self, context_find_by_extension_id): context_find_by_extension_id.return_value = None result = line_extension_service.find_by_extension_id(2) assert_that(result, none()) context_find_by_extension_id.assert_called_once_with(2)
def _find_line_id_for_extension(channel, extension_id): from xivo_dao.data_handler.line_extension import services as line_extension_services line_extension = line_extension_services.find_by_extension_id(extension_id) if line_extension: channel.send(line_extension.line_id) else: channel.send(None)
def test_given_incall_extension_then_returns_line_extension(self, context_find_by_extension_id, find_line_extension_by_extension_id): context_find_by_extension_id.return_value = Mock(Context, type='incall') line_extension = find_line_extension_by_extension_id.return_value = Mock(LineExtension, line_id=1, extension_id=2) result = line_extension_service.find_by_extension_id(2) assert_that(result, equal_to(line_extension)) context_find_by_extension_id.assert_called_once_with(2) find_line_extension_by_extension_id.assert_called_once_with(2)