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)