def test_remove_object_raises_error_if_invalid_index(self, sets):
manager = ObjectsManager(sets)
obj_index = manager.create(self.options)
self.assertEqual(len(manager.all()), sets)
with self.assertRaises(ValueError):
manager.remove(100)
self.assertEqual(len(manager.all()), sets)
def test_can_create_manager_from_dump(self, sets):
dump = {ObjectsManager.metadata: [[self.options] for _ in range(sets)]}
manager = ObjectsManager(dump=dump)
self.assertEqual(len(manager), sets)
self.assertEqual(len(manager.all()), sets)
for objs in manager.objects:
self.assertEqual(len(objs), sets)
base_object = Object(self.options)
for obj in manager.all():
self.assertEqual(base_object.to_metadata(), obj.to_metadata())
def test_can_create_object(self, sets):
manager = ObjectsManager(sets)
self.assertEqual(len(manager.all()), 0)
obj_index = manager.create(self.options)
self.assertEqual(len(manager.all()), sets)
for set_index in range(sets):
obj = manager.get(obj_index=obj_index, set_index=set_index)
self.assertEqual(obj.filename, __file__)
self.assertEqual(obj.mode, 'raw')
self.assertEqual(obj['target'], '/dev/sda')
def test_can_create_object_with_different_values(self):
manager = ObjectsManager(2)
self.assertEqual(len(manager.all()), 0)
self.options['target'] = ('/dev/sda', '/dev/sdb')
obj_index = manager.create(self.options)
obj0 = manager.get(obj_index=obj_index, set_index=0)
obj1 = manager.get(obj_index=obj_index, set_index=1)
self.assertEqual(obj0['target'], '/dev/sda')
self.assertEqual(obj1['target'], '/dev/sdb')
def test_can_get_all_objects(self, sets):
manager = ObjectsManager(sets)
obj_index = manager.create(self.options)
expected = [Object(self.options).to_metadata() for _ in range(sets)]
observed = [obj.to_metadata() for obj in manager.all()]
self.assertEqual(observed, expected)
def test_can_remove_object(self, sets):
manager = ObjectsManager(sets)
obj_index = manager.create(self.options)
self.assertEqual(len(manager.all()), 1 * sets)
manager.remove(obj_index)
self.assertEqual(len(manager.all()), 0)
