def test_migrate(self):
plan = Mock()
items = [
Item(plan, '1', 'path-1', 'new-path-1', []),
Item(plan, '2', 'path-2', 'new-path-2', []),
Item(plan, '3', 'path-3', 'new-path-3', []),
]
# test
batch = Batch()
batch.items = items
batch._migrate()
# validate
for i in items:
expected_call = call(i.unit_id, i.storage_path, i.new_path)
self.assertTrue(expected_call in plan.migrate.call_args_list)
def test_migrate(self):
plan = Mock()
items = [
Item(plan, '1', 'path-1', 'new-path-1', []),
Item(plan, '2', 'path-2', 'new-path-2', []),
Item(plan, '3', 'path-3', 'new-path-3', []),
]
# test
batch = Batch()
batch.items = items
batch._migrate()
# validate
for i in items:
expected_call = call(i.unit_id, i.storage_path, i.new_path)
self.assertIn(expected_call, plan.migrate.call_args_list)
def test_migrate(self):
plan = Mock()
items = [
Item(plan, '1', 'path-1', 'new-path-1', []),
Item(plan, '2', 'path-2', 'new-path-2', []),
Item(plan, '3', 'path-3', 'new-path-3', []),
]
# test
batch = Batch()
batch.items = items
batch._migrate()
# validate
self.assertEqual(
plan.migrate.call_args_list,
[
call(i.unit_id, i.storage_path, i.new_path) for i in items
])
def test_migrate(self, unit_migrate):
items = [
Item(Mock(), '1', 'path-1', 'new-path-1'),
Item(Mock(), '2', 'path-2', 'new-path-2'),
Item(Mock(), '3', 'path-3', 'new-path-3'),
]
_dict = Mock()
_dict.itervalues.return_value = iter(items)
# test
batch = Batch()
batch.items = _dict
batch._migrate()
# validate
self.assertEqual(
unit_migrate.call_args_list,
[
call(i.plan, i.unit_id, i.storage_path, i.new_path) for i in items
])
