Source code
Revision control
Copy as Markdown
Other Tools
#!/usr/bin/env vpython3
# Copyright 2015 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Unit tests for instrumentation.InstrumentationParser."""
import unittest
from pylib.instrumentation import instrumentation_parser
class InstrumentationParserTest(unittest.TestCase):
def testInstrumentationParser_nothing(self):
parser = instrumentation_parser.InstrumentationParser([''])
statuses = list(parser.IterStatus())
code, bundle = parser.GetResult()
self.assertEqual(None, code)
self.assertEqual({}, bundle)
self.assertEqual([], statuses)
def testInstrumentationParser_noMatchingStarts(self):
raw_output = [
'',
'this.is.a.test.package.TestClass:.',
'Test result for =.',
'Time: 1.234',
'',
'OK (1 test)',
]
parser = instrumentation_parser.InstrumentationParser(raw_output)
statuses = list(parser.IterStatus())
code, bundle = parser.GetResult()
self.assertEqual(None, code)
self.assertEqual({}, bundle)
self.assertEqual([], statuses)
def testInstrumentationParser_resultAndCode(self):
raw_output = [
'INSTRUMENTATION_RESULT: shortMsg=foo bar',
'INSTRUMENTATION_RESULT: longMsg=a foo',
'walked into',
'a bar',
'INSTRUMENTATION_CODE: -1',
]
parser = instrumentation_parser.InstrumentationParser(raw_output)
statuses = list(parser.IterStatus())
code, bundle = parser.GetResult()
self.assertEqual(-1, code)
self.assertEqual(
{'shortMsg': 'foo bar', 'longMsg': 'a foo\nwalked into\na bar'}, bundle)
self.assertEqual([], statuses)
def testInstrumentationParser_oneStatus(self):
raw_output = [
'INSTRUMENTATION_STATUS: foo=1',
'INSTRUMENTATION_STATUS: bar=hello',
'INSTRUMENTATION_STATUS: world=false',
'INSTRUMENTATION_STATUS: class=this.is.a.test.package.TestClass',
'INSTRUMENTATION_STATUS: test=testMethod',
'INSTRUMENTATION_STATUS_CODE: 0',
]
parser = instrumentation_parser.InstrumentationParser(raw_output)
statuses = list(parser.IterStatus())
expected = [
(0, {
'foo': '1',
'bar': 'hello',
'world': 'false',
'class': 'this.is.a.test.package.TestClass',
'test': 'testMethod',
})
]
self.assertEqual(expected, statuses)
def testInstrumentationParser_multiStatus(self):
raw_output = [
'INSTRUMENTATION_STATUS: class=foo',
'INSTRUMENTATION_STATUS: test=bar',
'INSTRUMENTATION_STATUS_CODE: 1',
'INSTRUMENTATION_STATUS: test_skipped=true',
'INSTRUMENTATION_STATUS_CODE: 0',
'INSTRUMENTATION_STATUS: class=hello',
'INSTRUMENTATION_STATUS: test=world',
'INSTRUMENTATION_STATUS: stack=',
'foo/bar.py (27)',
'hello/world.py (42)',
'test/file.py (1)',
'INSTRUMENTATION_STATUS_CODE: -1',
]
parser = instrumentation_parser.InstrumentationParser(raw_output)
statuses = list(parser.IterStatus())
expected = [
(1, {'class': 'foo', 'test': 'bar',}),
(0, {'test_skipped': 'true'}),
(-1, {
'class': 'hello',
'test': 'world',
'stack': '\nfoo/bar.py (27)\nhello/world.py (42)\ntest/file.py (1)',
}),
]
self.assertEqual(expected, statuses)
def testInstrumentationParser_statusResultAndCode(self):
raw_output = [
'INSTRUMENTATION_STATUS: class=foo',
'INSTRUMENTATION_STATUS: test=bar',
'INSTRUMENTATION_STATUS_CODE: 1',
'INSTRUMENTATION_RESULT: result=hello',
'world',
'',
'',
'INSTRUMENTATION_CODE: 0',
]
parser = instrumentation_parser.InstrumentationParser(raw_output)
statuses = list(parser.IterStatus())
code, bundle = parser.GetResult()
self.assertEqual(0, code)
self.assertEqual({'result': 'hello\nworld\n\n'}, bundle)
self.assertEqual([(1, {'class': 'foo', 'test': 'bar'})], statuses)
if __name__ == '__main__':
unittest.main(verbosity=2)