def testSucceedsWithNoSamples(self):
stream = six.StringIO()
instance = publisher.PrettyPrintStreamPublisher(stream)
instance.PublishSamples([])
self.assertRegexpMatches(
stream.getvalue(),
r'^\s*-+PerfKitBenchmarker\sResults\sSummary-+\s*$')
def testWritesToStream(self):
stream = six.StringIO()
instance = publisher.PrettyPrintStreamPublisher(stream)
samples = [{
'test': 'testb',
'metric': '1',
'value': 1.0,
'unit': 'MB',
'metadata': {}
}, {
'test': 'testb',
'metric': '2',
'value': 14.0,
'unit': 'MB',
'metadata': {}
}, {
'test': 'testa',
'metric': '3',
'value': 47.0,
'unit': 'us',
'metadata': {}
}]
instance.PublishSamples(samples)
value = stream.getvalue()
self.assertRegexpMatches(value, re.compile(r'TESTA.*TESTB', re.DOTALL))
def testDefaultsToStdout(self):
with mock.patch('sys.stdout') as mock_stdout:
instance = publisher.PrettyPrintStreamPublisher()
self.assertEqual(mock_stdout, instance.stream)
