fix: correct typing imports in test files

devin-ai-integration[bot] · aaronsteers · devin-ai-integration[bot] · commit ee95d097a788 · 2025-04-13T03:10:16.000Z
Co-Authored-By: Aaron &lt;AJ&gt; Steers &lt;aj@airbyte.io&gt;
diff --git a/unit_tests/connector_builder/test_message_grouper.py b/unit_tests/connector_builder/test_message_grouper.py
@@ -4,7 +4,7 @@
 
 import json
 from typing import Any
-from collections.abc import list
+from typing import List
 from collections.abc import Iterator, Mapping
 from unittest.mock import MagicMock, Mock, patch
 
diff --git a/unit_tests/destinations/test_destination.py b/unit_tests/destinations/test_destination.py
@@ -8,7 +8,7 @@
 from os import PathLike
 from pathlib import Path
 from typing import Any
-from collections.abc import dict, List, Union
+from typing import Dict, List, Union
 from collections.abc import Iterable, Mapping
 from unittest.mock import ANY
 
diff --git a/unit_tests/destinations/vector_db_based/document_processor_test.py b/unit_tests/destinations/vector_db_based/document_processor_test.py
@@ -3,7 +3,7 @@
 #
 
 from typing import Any
-from collections.abc import list, Optional
+from typing import List, Optional
 from collections.abc import Mapping
 from unittest.mock import MagicMock
 
diff --git a/unit_tests/sources/concurrent_source/test_concurrent_source_adapter.py b/unit_tests/sources/concurrent_source/test_concurrent_source_adapter.py
@@ -4,7 +4,7 @@
 
 import logging
 from typing import Any
-from collections.abc import list, Optional, Tuple
+from typing import List, Optional, Tuple
 from collections.abc import Mapping
 from unittest.mock import Mock
 
diff --git a/unit_tests/sources/declarative/async_job/test_integration.py b/unit_tests/sources/declarative/async_job/test_integration.py
@@ -3,7 +3,7 @@
 
 import logging
 from typing import Any
-from collections.abc import list, Optional, Set, Tuple
+from typing import List, Optional, Set, Tuple
 from collections.abc import Iterable, Mapping
 from unittest import TestCase, mock
 
diff --git a/unit_tests/sources/declarative/async_job/test_job_orchestrator.py b/unit_tests/sources/declarative/async_job/test_job_orchestrator.py
@@ -4,7 +4,7 @@
 import sys
 import threading
 import time
-from collections.abc import list, Optional, Set, Tuple
+from typing import List, Optional, Set, Tuple
 from collections.abc import Callable, Mapping
 from unittest import TestCase, mock
 from unittest.mock import MagicMock, Mock, call
diff --git a/unit_tests/sources/declarative/async_job/test_job_tracker.py b/unit_tests/sources/declarative/async_job/test_job_tracker.py
@@ -1,6 +1,6 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 
-from collections.abc import list
+from typing import List
 from unittest import TestCase
 
 import pytest
diff --git a/unit_tests/sources/declarative/extractors/test_dpath_extractor.py b/unit_tests/sources/declarative/extractors/test_dpath_extractor.py
@@ -3,8 +3,8 @@
 #
 import io
 import json
-from collections.abc import dict
-from collections.abc import list, Union
+from typing import Dict
+from typing import List, Union
 
 import pytest
 import requests
diff --git a/unit_tests/sources/declarative/extractors/test_record_filter.py b/unit_tests/sources/declarative/extractors/test_record_filter.py
@@ -1,7 +1,7 @@
 #
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
-from collections.abc import list, Optional
+from typing import List, Optional
 from collections.abc import Mapping
 
 import pytest
diff --git a/unit_tests/sources/declarative/incremental/test_concurrent_perpartitioncursor.py b/unit_tests/sources/declarative/incremental/test_concurrent_perpartitioncursor.py
@@ -3,7 +3,7 @@
 from copy import deepcopy
 from datetime import datetime, timedelta
 from typing import Any
-from collections.abc import list, Optional, Union
+from typing import List, Optional, Union
 from collections.abc import Mapping, MutableMapping
 from unittest.mock import MagicMock, patch
 from urllib.parse import unquote
diff --git a/unit_tests/sources/declarative/parsers/testing_components.py b/unit_tests/sources/declarative/parsers/testing_components.py
@@ -3,7 +3,7 @@
 #
 
 from dataclasses import dataclass, field
-from collections.abc import list, Optional
+from typing import List, Optional
 
 from airbyte_cdk.sources.declarative.extractors import DpathExtractor
 from airbyte_cdk.sources.declarative.partition_routers import SubstreamPartitionRouter
diff --git a/unit_tests/sources/declarative/partition_routers/test_grouping_partition_router.py b/unit_tests/sources/declarative/partition_routers/test_grouping_partition_router.py
@@ -3,7 +3,7 @@
 #
 
 from typing import Any
-from collections.abc import list, Optional, Union
+from typing import List, Optional, Union
 from collections.abc import Iterable, Mapping
 from unittest.mock import MagicMock
 
diff --git a/unit_tests/sources/declarative/partition_routers/test_parent_state_stream.py b/unit_tests/sources/declarative/partition_routers/test_parent_state_stream.py
@@ -2,7 +2,7 @@
 
 import copy
 from typing import Any
-from collections.abc import list, Optional, Union
+from typing import List, Optional, Union
 from collections.abc import Mapping, MutableMapping
 from unittest.mock import MagicMock
 
diff --git a/unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py b/unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py
@@ -5,7 +5,7 @@
 import logging
 from functools import partial
 from typing import Any
-from collections.abc import list, Optional, Union
+from typing import List, Optional, Union
 from collections.abc import Iterable, Mapping, MutableMapping
 
 import pytest as pytest
diff --git a/unit_tests/sources/declarative/requesters/request_options/test_request_options.py b/unit_tests/sources/declarative/requesters/request_options/test_request_options.py
@@ -1,5 +1,5 @@
 from typing import Any
-from collections.abc import dict, List, Optional, Type
+from typing import Dict, List, Optional, Type
 
 import pytest
 
diff --git a/unit_tests/sources/declarative/stream_slicers/test_declarative_partition_generator.py b/unit_tests/sources/declarative/stream_slicers/test_declarative_partition_generator.py
@@ -1,6 +1,6 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 
-from collections.abc import list
+from typing import List
 from unittest import TestCase
 from unittest.mock import Mock
 
diff --git a/unit_tests/sources/declarative/test_concurrent_declarative_source.py b/unit_tests/sources/declarative/test_concurrent_declarative_source.py
@@ -7,7 +7,7 @@
 import math
 from datetime import datetime, timedelta, timezone
 from typing import Any
-from collections.abc import dict, List, Optional, Tuple, Union
+from typing import Dict, List, Optional, Tuple, Union
 from collections.abc import Iterable, Mapping
 from unittest.mock import patch
 
diff --git a/unit_tests/sources/declarative/test_manifest_declarative_source.py b/unit_tests/sources/declarative/test_manifest_declarative_source.py
@@ -9,7 +9,7 @@
 from copy import deepcopy
 from pathlib import Path
 from typing import Any
-from collections.abc import list
+from typing import List
 from collections.abc import Mapping
 from unittest.mock import call, patch
 
diff --git a/unit_tests/sources/declarative/transformations/test_add_fields.py b/unit_tests/sources/declarative/transformations/test_add_fields.py
@@ -3,7 +3,7 @@
 #
 
 from typing import Any
-from collections.abc import list, Optional, Tuple
+from typing import List, Optional, Tuple
 from collections.abc import Mapping
 
 import pytest
diff --git a/unit_tests/sources/declarative/transformations/test_remove_fields.py b/unit_tests/sources/declarative/transformations/test_remove_fields.py
@@ -3,7 +3,7 @@
 #
 
 from typing import Any
-from collections.abc import list
+from typing import List
 from collections.abc import Mapping
 
 import pytest
diff --git a/unit_tests/sources/file_based/file_types/test_csv_parser.py b/unit_tests/sources/file_based/file_types/test_csv_parser.py
@@ -9,7 +9,7 @@
 import unittest
 from datetime import datetime
 from typing import Any
-from collections.abc import dict, List, Set
+from typing import Dict, List, Set
 from collections.abc import Generator
 from unittest import TestCase, mock
 from unittest.mock import Mock
diff --git a/unit_tests/sources/file_based/file_types/test_jsonl_parser.py b/unit_tests/sources/file_based/file_types/test_jsonl_parser.py
@@ -6,7 +6,7 @@
 import io
 import json
 from typing import Any
-from collections.abc import dict
+from typing import Dict
 from unittest.mock import MagicMock, Mock
 
 import pytest
diff --git a/unit_tests/sources/file_based/helpers.py b/unit_tests/sources/file_based/helpers.py
@@ -6,7 +6,7 @@
 from datetime import datetime
 from io import IOBase
 from typing import Any
-from collections.abc import dict, List, Optional
+from typing import Dict, List, Optional
 from collections.abc import Mapping
 
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
diff --git a/unit_tests/sources/file_based/in_memory_files_source.py b/unit_tests/sources/file_based/in_memory_files_source.py
@@ -10,7 +10,7 @@
 from datetime import datetime
 from io import IOBase
 from typing import Any
-from collections.abc import dict, List, Optional
+from typing import Dict, List, Optional
 from collections.abc import Iterable, Mapping
 
 import avro.io as ai
diff --git a/unit_tests/sources/file_based/stream/concurrent/test_file_based_concurrent_cursor.py b/unit_tests/sources/file_based/stream/concurrent/test_file_based_concurrent_cursor.py
@@ -3,7 +3,7 @@
 
 from datetime import datetime
 from typing import Any
-from collections.abc import dict, List, Optional, Tuple
+from typing import Dict, List, Optional, Tuple
 from collections.abc import MutableMapping
 from unittest.mock import MagicMock
 
diff --git a/unit_tests/sources/file_based/stream/test_default_file_based_cursor.py b/unit_tests/sources/file_based/stream/test_default_file_based_cursor.py
@@ -4,7 +4,7 @@
 
 from datetime import datetime, timedelta
 from typing import Any
-from collections.abc import list
+from typing import List
 from collections.abc import Mapping
 from unittest.mock import MagicMock
 
diff --git a/unit_tests/sources/file_based/test_file_based_stream_reader.py b/unit_tests/sources/file_based/test_file_based_stream_reader.py
@@ -6,7 +6,7 @@
 from datetime import datetime
 from io import IOBase
 from typing import Any
-from collections.abc import dict, List, Optional, Set
+from typing import Dict, List, Optional, Set
 from collections.abc import Iterable, Mapping
 
 import pytest
diff --git a/unit_tests/sources/file_based/test_scenarios.py b/unit_tests/sources/file_based/test_scenarios.py
@@ -6,7 +6,7 @@
 import math
 from pathlib import Path, PosixPath
 from typing import Any
-from collections.abc import dict, List, Optional, Union
+from typing import Dict, List, Optional, Union
 from collections.abc import Mapping
 
 import pytest
diff --git a/unit_tests/sources/fixtures/source_test_fixture.py b/unit_tests/sources/fixtures/source_test_fixture.py
@@ -6,7 +6,7 @@
 import logging
 from abc import ABC
 from typing import Any
-from collections.abc import list, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 from collections.abc import Iterable, Mapping
 
 import requests
diff --git a/unit_tests/sources/mock_server_tests/mock_source_fixture.py b/unit_tests/sources/mock_server_tests/mock_source_fixture.py
@@ -6,7 +6,7 @@
 from abc import ABC
 from datetime import datetime, timedelta, timezone
 from typing import Any
-from collections.abc import dict, List, Optional, Tuple
+from typing import Dict, List, Optional, Tuple
 from collections.abc import Iterable, Mapping, MutableMapping
 
 import requests
diff --git a/unit_tests/sources/mock_server_tests/test_helpers/airbyte_message_assertions.py b/unit_tests/sources/mock_server_tests/test_helpers/airbyte_message_assertions.py
@@ -2,7 +2,7 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 #
 
-from collections.abc import list
+from typing import List
 
 import pytest
 
diff --git a/unit_tests/sources/mock_server_tests/test_mock_server_abstract_source.py b/unit_tests/sources/mock_server_tests/test_mock_server_abstract_source.py
@@ -3,7 +3,7 @@
 #
 
 from datetime import datetime, timedelta, timezone
-from collections.abc import list, Optional
+from typing import List, Optional
 from unittest import TestCase
 
 import freezegun
diff --git a/unit_tests/sources/mock_server_tests/test_resumable_full_refresh.py b/unit_tests/sources/mock_server_tests/test_resumable_full_refresh.py
@@ -4,7 +4,7 @@
 
 from datetime import datetime, timezone
 from typing import Any
-from collections.abc import dict, List, Optional
+from typing import Dict, List, Optional
 from unittest import TestCase
 
 import freezegun
diff --git a/unit_tests/sources/streams/concurrent/scenarios/stream_facade_builder.py b/unit_tests/sources/streams/concurrent/scenarios/stream_facade_builder.py
@@ -5,7 +5,7 @@
 import concurrent
 import logging
 from typing import Any
-from collections.abc import list, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 from collections.abc import Mapping
 
 from airbyte_cdk.models import (
diff --git a/unit_tests/sources/streams/concurrent/scenarios/thread_based_concurrent_stream_source_builder.py b/unit_tests/sources/streams/concurrent/scenarios/thread_based_concurrent_stream_source_builder.py
@@ -4,7 +4,7 @@
 import json
 import logging
 from typing import Any
-from collections.abc import list, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 from collections.abc import Iterable, Mapping
 
 from airbyte_cdk.models import (
diff --git a/unit_tests/sources/streams/concurrent/scenarios/utils.py b/unit_tests/sources/streams/concurrent/scenarios/utils.py
@@ -2,7 +2,7 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 from typing import Any
-from collections.abc import list, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 from collections.abc import Iterable, Mapping
 
 from airbyte_cdk.models import SyncMode
diff --git a/unit_tests/sources/streams/concurrent/test_partition_enqueuer.py b/unit_tests/sources/streams/concurrent/test_partition_enqueuer.py
@@ -3,7 +3,7 @@
 #
 import unittest
 from queue import Queue
-from collections.abc import list
+from typing import List
 from collections.abc import Callable, Iterable
 from unittest.mock import Mock, patch
 
diff --git a/unit_tests/sources/streams/concurrent/test_partition_reader.py b/unit_tests/sources/streams/concurrent/test_partition_reader.py
@@ -3,7 +3,7 @@
 #
 import unittest
 from queue import Queue
-from collections.abc import list
+from typing import List
 from collections.abc import Callable, Iterable
 from unittest.mock import Mock
 
diff --git a/unit_tests/sources/streams/http/test_http.py b/unit_tests/sources/streams/http/test_http.py
@@ -6,7 +6,7 @@
 import logging
 from http import HTTPStatus
 from typing import Any
-from collections.abc import list, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 from collections.abc import Callable, Iterable, Mapping, MutableMapping
 from unittest.mock import ANY, MagicMock, patch
 
diff --git a/unit_tests/sources/streams/test_stream_read.py b/unit_tests/sources/streams/test_stream_read.py
@@ -5,7 +5,7 @@
 import logging
 from copy import deepcopy
 from typing import Any
-from collections.abc import dict, List, Optional, Union
+from typing import Dict, List, Optional, Union
 from collections.abc import Iterable, Mapping, MutableMapping
 from unittest.mock import Mock
 
diff --git a/unit_tests/sources/streams/test_streams_core.py b/unit_tests/sources/streams/test_streams_core.py
@@ -4,7 +4,7 @@
 
 import logging
 from typing import Any
-from collections.abc import list, Optional
+from typing import List, Optional
 from collections.abc import Iterable, Mapping, MutableMapping
 from unittest import mock
 
diff --git a/unit_tests/sources/test_config.py b/unit_tests/sources/test_config.py
@@ -2,7 +2,7 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 
-from collections.abc import list, Union
+from typing import List, Union
 
 from pydantic.v1 import BaseModel, Field
 
diff --git a/unit_tests/sources/test_connector_state_manager.py b/unit_tests/sources/test_connector_state_manager.py
@@ -3,7 +3,7 @@
 #
 
 from contextlib import nullcontext as does_not_raise
-from collections.abc import list
+from typing import List
 
 import pytest
 
diff --git a/unit_tests/sources/test_integration_source.py b/unit_tests/sources/test_integration_source.py
@@ -5,7 +5,7 @@
 import json
 import os
 from typing import Any
-from collections.abc import list
+from typing import List
 from collections.abc import Mapping
 from unittest import mock
 from unittest.mock import patch
diff --git a/unit_tests/sources/test_source.py b/unit_tests/sources/test_source.py
diff --git a/unit_tests/sources/test_source_read.py b/unit_tests/sources/test_source_read.py
diff --git a/unit_tests/test/mock_http/test_response_builder.py b/unit_tests/test/mock_http/test_response_builder.py
diff --git a/unit_tests/test/test_entrypoint_wrapper.py b/unit_tests/test/test_entrypoint_wrapper.py
diff --git a/unit_tests/test_entrypoint.py b/unit_tests/test_entrypoint.py
diff --git a/unit_tests/test_logger.py b/unit_tests/test_logger.py
diff --git a/unit_tests/utils/test_datetime_format_inferrer.py b/unit_tests/utils/test_datetime_format_inferrer.py
diff --git a/unit_tests/utils/test_schema_inferrer.py b/unit_tests/utils/test_schema_inferrer.py

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,7 @@`
`3`	`3`	`#`
`4`	`4`
`5`	`5`	`from typing import Any`
`6`		`-from collections.abc import list, Optional`
	`6`	`+from typing import List, Optional`
`7`	`7`	`from collections.abc import Mapping`
`8`	`8`	`from unittest.mock import MagicMock`
`9`	`9`
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`#`
`2`	`2`	`# Copyright (c) 2023 Airbyte, Inc., all rights reserved.`
`3`	`3`	`#`
`4`		`-from collections.abc import list, Optional`
	`4`	`+from typing import List, Optional`
`5`	`5`	`from collections.abc import Mapping`
`6`	`6`
`7`	`7`	`import pytest`