@@ -115,8 +115,7 @@ def test_create_metadata_item_rule(self, mock_add_metadata_rule):
115115
116116 @patch .object (clone_metadata_utils , 'create_metadata_items' )
117117 @patch .object (clone_metadata_utils , 'list_metadata_items' )
118- @patch ('builtins.input' , return_value = 'true' )
119- def test_compare_create_metadata_items_new_fields (sel , mock_list , mock_create , ):
118+ def test_compare_create_metadata_items_new_fields (self , mock_list , mock_create ):
120119 """Test comparing and creating new metadata fields"""
121120 metadata_fields = {
122121 'metadata_fields' : [
@@ -139,7 +138,8 @@ def test_compare_create_metadata_items_new_fields(sel, mock_list, mock_create, )
139138 'metadata_fields' : []
140139 }
141140
142- clone_metadata_utils .compare_create_metadata_items (mock_source_fields , mock_destination_fields , key = "metadata_fields" , ** self .mock_target_config )
141+ with patch ('builtins.input' , return_value = 'y' ):
142+ clone_metadata_utils .compare_create_metadata_items (mock_source_fields , mock_destination_fields , key = "metadata_fields" , ** self .mock_target_config )
143143
144144 # Both fields should be created
145145 self .assertEqual (mock_create .call_count , 2 )
@@ -152,7 +152,6 @@ def test_compare_create_metadata_items_new_fields(sel, mock_list, mock_create, )
152152
153153 @patch .object (clone_metadata_utils , 'create_metadata_items' )
154154 @patch .object (clone_metadata_utils , 'list_metadata_items' )
155- @patch ('builtins.input' , return_value = 'y' )
156155 def test_compare_create_metadata_items_new_rules (self , mock_list , mock_create ):
157156 """Test comparing and creating new metadata rules"""
158157 metadata_rules = {
@@ -185,7 +184,8 @@ def test_compare_create_metadata_items_new_rules(self, mock_list, mock_create):
185184 'metadata_rules' : []
186185 }
187186
188- clone_metadata_utils .compare_create_metadata_items (mock_source_metadata_rules , mock_destination_metadata_rules , key = "metadata_rules" , ** self .mock_target_config )
187+ with patch ('builtins.input' , return_value = 'y' ):
188+ clone_metadata_utils .compare_create_metadata_items (mock_source_metadata_rules , mock_destination_metadata_rules , key = "metadata_rules" , ** self .mock_target_config )
189189 # Both rules should be created
190190 self .assertEqual (mock_create .call_count , 2 )
191191 mock_create .assert_any_call ('add_metadata_rule' , mock_source_metadata_rules ['metadata_rules' ][0 ], ** self .mock_target_config )
@@ -196,7 +196,6 @@ def test_compare_create_metadata_items_new_rules(self, mock_list, mock_create):
196196 self .assertEqual (result , mock_list .return_value )
197197
198198 @patch .object (clone_metadata_utils , 'create_metadata_items' )
199- @patch ('builtins.input' , return_value = 'true' )
200199 def test_compare_create_metadata_items_existing_fields (self , mock_create ):
201200 """Test comparing when fields already exist"""
202201 mock_source_fields = {
@@ -220,13 +219,13 @@ def test_compare_create_metadata_items_existing_fields(self, mock_create):
220219 ]
221220 }
222221
223- clone_metadata_utils .compare_create_metadata_items (mock_source_fields , mock_destination_fields , key = "metadata_fields" , ** self .mock_target_config )
222+ with patch ('builtins.input' , return_value = 'y' ):
223+ clone_metadata_utils .compare_create_metadata_items (mock_source_fields , mock_destination_fields , key = "metadata_fields" , ** self .mock_target_config )
224224
225225 # No fields should be created
226226 mock_create .assert_not_called ()
227227
228228 @patch .object (clone_metadata_utils , 'create_metadata_items' )
229- @patch ('builtins.input' , return_value = 'true' )
230229 def test_compare_create_metadata_items_existing_rules (self , mock_create ):
231230 """Test comparing when rules already exist"""
232231
@@ -259,14 +258,14 @@ def test_compare_create_metadata_items_existing_rules(self, mock_create):
259258 ]
260259 }
261260
262- clone_metadata_utils .compare_create_metadata_items (mock_source_metadata_rules , mock_destination_metadata_rules , key = "metadata_rules" , ** self .mock_target_config )
261+ with patch ('builtins.input' , return_value = 'y' ):
262+ clone_metadata_utils .compare_create_metadata_items (mock_source_metadata_rules , mock_destination_metadata_rules , key = "metadata_rules" , ** self .mock_target_config )
263263
264264 # No rules should be created
265265 mock_create .assert_not_called ()
266266
267267 @patch .object (clone_metadata_utils , 'create_metadata_items' )
268268 @patch .object (clone_metadata_utils , 'list_metadata_items' )
269- @patch ('builtins.input' , return_value = 'true' )
270269 def test_compare_create_metadata_items_mixed_scenario (self , mock_list , mock_create ):
271270 """Test comparing with mix of new and existing fields"""
272271 metadata_fields = {
@@ -298,7 +297,8 @@ def test_compare_create_metadata_items_mixed_scenario(self, mock_list, mock_crea
298297 mock_source_fields = metadata_fields
299298 mock_list .return_value = metadata_fields
300299
301- clone_metadata_utils .compare_create_metadata_items (mock_source_fields , mock_destination_fields , key = "metadata_fields" , ** self .mock_target_config )
300+ with patch ('builtins.input' , return_value = 'y' ):
301+ clone_metadata_utils .compare_create_metadata_items (mock_source_fields , mock_destination_fields , key = "metadata_fields" , ** self .mock_target_config )
302302
303303 # Only new_field should be created
304304 mock_create .assert_called_once_with ('add_metadata_field' , mock_source_fields ['metadata_fields' ][1 ], ** self .mock_target_config )
@@ -309,7 +309,6 @@ def test_compare_create_metadata_items_mixed_scenario(self, mock_list, mock_crea
309309
310310 @patch .object (clone_metadata_utils , 'create_metadata_items' )
311311 @patch .object (clone_metadata_utils , 'list_metadata_items' )
312- @patch ('builtins.input' , return_value = 'true' )
313312 def test_compare_create_metadata_items_mixed_rules_scenario (self , mock_list , mock_create ):
314313 """Test comparing with mix of new and existing rules"""
315314 metadata_rules = {
@@ -353,7 +352,8 @@ def test_compare_create_metadata_items_mixed_rules_scenario(self, mock_list, moc
353352 mock_source_metadata_rules = metadata_rules
354353 mock_list .return_value = metadata_rules
355354
356- clone_metadata_utils .compare_create_metadata_items (mock_source_metadata_rules , mock_destination_metadata_rules , key = "metadata_rules" , ** self .mock_target_config )
355+ with patch ('builtins.input' , return_value = 'y' ):
356+ clone_metadata_utils .compare_create_metadata_items (mock_source_metadata_rules , mock_destination_metadata_rules , key = "metadata_rules" , ** self .mock_target_config )
357357
358358 # Only new_rule should be created
359359 mock_create .assert_called_once_with ('add_metadata_rule' , mock_source_metadata_rules ['metadata_rules' ][1 ], ** self .mock_target_config )
0 commit comments