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()