def setUp(self):
self.instance = publisher.SampleCollector(publishers=[])
self.sample = sample.Sample('widgets', 100, 'oz', {'foo': 'bar'})
self.benchmark = 'test!'
self.benchmark_spec = mock.MagicMock()
p = mock.patch(publisher.__name__ + '.FLAGS')
self.mock_flags = p.start()
self.addCleanup(p.stop)
self.mock_flags.product_name = 'PerfKitBenchmarker'
def testSpecPickledEvenWithException(self):
# Make provision have an exception.
self.enter_context(
mock.patch.object(pkb, 'DoProvisionPhase',
side_effect=Exception()))
# Make benchmark_spec avoid pickling for test.
test_benchmark_spec = (
pkb_common_test_case.CreateBenchmarkSpecFromYaml(
pkb_common_test_case.SIMPLE_CONFIG, 'cluster_boot'))
self.enter_context(mock.patch.object(test_benchmark_spec, 'Pickle'))
mock_freeze = self.enter_context(
mock.patch.object(test_benchmark_spec, 'Freeze'))
collector = publisher.SampleCollector()
# Run the benchmark loop.
with self.assertRaises(Exception):
pkb.RunBenchmark(test_benchmark_spec, collector)
# PKB should still attempt to freeze benchmark spec.
mock_freeze.assert_called_once()
def setUp(self):
self.instance = publisher.SampleCollector(publishers=[])
self.sample = sample.Sample('widgets', 100, 'oz', {'foo': 'bar'})
self.benchmark = 'test!'
self.benchmark_spec = mock.MagicMock()
