@@ -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
131131def 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