From c719137df359770aee2ddbba4b4c0f84b02512e9 Mon Sep 17 00:00:00 2001 From: Anatolii Yatsuk <35109939+tolik0@users.noreply.github.com> Date: Tue, 24 Oct 2023 16:15:11 +0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20Airbyte=20CDK:=20Fix=20flake=20e?= =?UTF-8?q?rrors=20in=20file-based=20CDK=20(#31771)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../test_default_file_based_availability_strategy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/airbyte-cdk/python/unit_tests/sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py b/airbyte-cdk/python/unit_tests/sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py index 8077a134b12b..4ea4198ed9c4 100644 --- a/airbyte-cdk/python/unit_tests/sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py +++ b/airbyte-cdk/python/unit_tests/sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py @@ -11,7 +11,7 @@ ) from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig from airbyte_cdk.sources.file_based.config.jsonl_format import JsonlFormat -from airbyte_cdk.sources.file_based.exceptions import CheckAvailabilityError, CustomFileBasedException +from airbyte_cdk.sources.file_based.exceptions import CustomFileBasedException from airbyte_cdk.sources.file_based.file_based_stream_reader import AbstractFileBasedStreamReader from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser from airbyte_cdk.sources.file_based.remote_file import RemoteFile @@ -86,5 +86,5 @@ def test_catching_and_raising_custom_file_based_exception(self) -> None: # Invoke the check_availability_and_parsability method and check if it correctly handles the exception is_available, error_message = self._strategy.check_availability_and_parsability(self._stream, Mock(), Mock()) - assert is_available == False + assert not is_available assert "Custom exception for testing." in error_message