def test_add(self):
unit = Mock(plan=1, id=2, storage_path=3, new_path=4)
# test
batch = Batch()
batch.add(unit)
# validation
item = batch.items.values()[0]
self.assertEqual(len(batch), 1)
self.assertEqual(batch.items.keys()[0], unit.storage_path)
self.assertEqual(item.plan, unit.plan)
self.assertEqual(item.unit_id, unit.id)
self.assertEqual(item.storage_path, unit.storage_path)
def test_add(self):
unit = Mock(plan=1, id=2, storage_path='3', new_path='4', files=['4', '5'])
# test
batch = Batch()
batch.add(unit)
# validation
item = batch.items[0]
self.assertEqual(len(batch), 1)
self.assertEqual(
sorted(batch.paths.keys()),
sorted([unit.storage_path] + [os.path.join(unit.storage_path, f) for f in unit.files]))
self.assertEqual(batch.paths.values(), [item, item, item])
self.assertEqual(item.plan, unit.plan)
self.assertEqual(item.unit_id, unit.id)
self.assertEqual(item.storage_path, unit.storage_path)
def test_add(self):
unit = Mock(plan=1, id=2, storage_path='3', new_path='4', files=['4', '5'])
# test
batch = Batch()
batch.add(unit)
# validation
item = batch.items[0]
self.assertEqual(len(batch), 1)
self.assertEqual(
sorted(batch.paths.keys()),
sorted([unit.storage_path] + [os.path.join(unit.storage_path, f) for f in unit.files]))
self.assertEqual(batch.paths.values(), [item, item, item])
self.assertEqual(item.plan, unit.plan)
self.assertEqual(item.unit_id, unit.id)
self.assertEqual(item.storage_path, unit.storage_path)
