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
def test_empty(self):
with self.assertNotRaises():
with prepared([]):
pass
