Skip to content

Commit a626d18

Browse files
tentatively add user input mock - 5
1 parent fe0233e commit a626d18

1 file changed

Lines changed: 13 additions & 13 deletions

File tree

test/test_modules/test_cli_clone.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)