Skip to content

Commit 6ec97a4

Browse files
Revert "Revert "fix(check-stream): prefix internal config override key""
This reverts commit 190f8af.
1 parent 190f8af commit 6ec97a4

2 files changed

Lines changed: 10 additions & 10 deletions

File tree

airbyte_cdk/sources/declarative/checks/check_stream.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from airbyte_cdk.sources.streams.core import Stream
1414
from airbyte_cdk.sources.streams.http.availability_strategy import HttpAvailabilityStrategy
1515

16-
CHECK_STREAM_NAMES_CONFIG_KEY = "airbyte_check_stream_names"
16+
CHECK_STREAM_NAMES_CONFIG_KEY = "__airbyte_check_stream_names"
1717

1818

1919
def evaluate_availability(

unit_tests/sources/declarative/checks/test_check_stream.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def test_check_stream_names_can_be_overridden_from_config():
9393
check_stream = CheckStream(["static_stream"], parameters={})
9494

9595
assert check_stream.check_connection(
96-
source, logger, {"airbyte_check_stream_names": ["selected_stream"]}
96+
source, logger, {"__airbyte_check_stream_names": ["selected_stream"]}
9797
) == (True, None)
9898
static_stream.stream_slices.assert_not_called()
9999

@@ -107,7 +107,7 @@ def test_check_stream_names_override_accepts_empty_list():
107107

108108
check_stream = CheckStream(["static_stream"], parameters={})
109109

110-
assert check_stream.check_connection(source, logger, {"airbyte_check_stream_names": []}) == (
110+
assert check_stream.check_connection(source, logger, {"__airbyte_check_stream_names": []}) == (
111111
True,
112112
None,
113113
)
@@ -124,8 +124,8 @@ def test_check_stream_names_override_requires_list_of_strings(override):
124124

125125
check_stream = CheckStream(["selected_stream"], parameters={})
126126

127-
with pytest.raises(ValueError, match="airbyte_check_stream_names must be a list of strings."):
128-
check_stream.check_connection(source, logger, {"airbyte_check_stream_names": override})
127+
with pytest.raises(ValueError, match="__airbyte_check_stream_names must be a list of strings."):
128+
check_stream.check_connection(source, logger, {"__airbyte_check_stream_names": override})
129129

130130

131131
def test_check_stream_names_override_rejects_unknown_stream():
@@ -139,7 +139,7 @@ def test_check_stream_names_override_rejects_unknown_stream():
139139

140140
with pytest.raises(ValueError, match="unknown_stream is not part of the catalog."):
141141
check_stream.check_connection(
142-
source, logger, {"airbyte_check_stream_names": ["unknown_stream"]}
142+
source, logger, {"__airbyte_check_stream_names": ["unknown_stream"]}
143143
)
144144

145145

@@ -154,7 +154,7 @@ def test_check_stream_names_override_returns_unavailable_stream_message():
154154
check_stream = CheckStream(["other_stream"], parameters={})
155155

156156
stream_is_available, reason = check_stream.check_connection(
157-
source, logger, {"airbyte_check_stream_names": ["selected_stream"]}
157+
source, logger, {"__airbyte_check_stream_names": ["selected_stream"]}
158158
)
159159
assert not stream_is_available
160160
assert "no stream slices were found, likely because the parent stream is empty" in reason
@@ -164,9 +164,9 @@ def test_check_stream_names_override_validates_before_stream_discovery():
164164
source = MagicMock()
165165
check_stream = CheckStream(["selected_stream"], parameters={})
166166

167-
with pytest.raises(ValueError, match="airbyte_check_stream_names must be a list of strings."):
167+
with pytest.raises(ValueError, match="__airbyte_check_stream_names must be a list of strings."):
168168
check_stream.check_connection(
169-
source, logger, {"airbyte_check_stream_names": "selected_stream"}
169+
source, logger, {"__airbyte_check_stream_names": "selected_stream"}
170170
)
171171

172172
source.streams.assert_not_called()
@@ -810,7 +810,7 @@ def test_check_empty_static_stream_override_still_checks_dynamic_streams():
810810
}
811811
},
812812
}
813-
check_config = {**_CONFIG, "airbyte_check_stream_names": []}
813+
check_config = {**_CONFIG, "__airbyte_check_stream_names": []}
814814

815815
with HttpMocker() as http_mocker:
816816
static_stream_request = HttpRequest(url="https://api.test.com/static")

0 commit comments

Comments
 (0)