def test_fail_inside_context(self): log = [] images = [FakeImage('img', log)] expected = [('prepare', 'img'), ('teardown', 'img')] with pytest.raises(InjectedFailure): with prepared(images): raise InjectedFailure() assert expected == log
def test_one_image(self): log = [] images = [ FakeImage('img', log)] expected = [ ('prepare', 'img'), ('teardown', 'img')] with prepared(images): self.assertEqual(expected[:1], log) self.assertEqual(expected, log)
def test_prepare_failure(self): log = [] images = [ FakeImage('img1', log), FakeImage('img2', log, prepare=InjectedFailure) ] expected = [('prepare', 'img1'), ('teardown', 'img1')] with pytest.raises(InjectedFailure): with prepared(images): pass assert expected == log
def test_fail_inside_context_with_teardown_failure(self): log = [] images = [ FakeImage('img', log, teardown=InjectedFailure)] expected = [ ('prepare', 'img')] with self.assertRaises(RuntimeError): with prepared(images): raise RuntimeError() self.assertEqual(expected, log)
def test_fail_inside_context(self): log = [] images = [ FakeImage('img', log)] expected = [ ('prepare', 'img'), ('teardown', 'img')] with self.assertRaises(InjectedFailure): with prepared(images): raise InjectedFailure() self.assertEqual(expected, log)
def test_prepare_failure(self): log = [] images = [ FakeImage('img1', log), FakeImage('img2', log, prepare=InjectedFailure)] expected = [ ('prepare', 'img1'), ('teardown', 'img1')] with self.assertRaises(InjectedFailure): with prepared(images): pass self.assertEqual(expected, log)
def test_two_images(self): log = [] images = [ FakeImage('img1', log), FakeImage('img2', log)] expected = [ ('prepare', 'img1'), ('prepare', 'img2'), ('teardown', 'img2'), ('teardown', 'img1')] with prepared(images): self.assertEqual(expected[:2], log) self.assertEqual(expected, log)
def test_teardown_failure(self): log = [] images = [ FakeImage('img1', log), FakeImage('img2', log, teardown=InjectedFailure)] expected = [ ('prepare', 'img1'), ('prepare', 'img2'), ('teardown', 'img1')] with self.assertRaises(TeardownError): with prepared(images): pass self.assertEqual(expected, log)
def _run(self): with prepared(self._images): vol_paths = [image_info.path for image_info in self._images] virtsysprep.sysprep(vol_paths)
def test_empty(self): with self.assertNotRaises(): with prepared([]): pass