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)