| #!/usr/bin/python3 |
| |
| import sys |
| import unittest |
| |
| from six import StringIO |
| |
| import common |
| from autotest_lib.tko import db |
| |
| |
| class LogErrorTestCase(unittest.TestCase): |
| """Tests for _log_error().""" |
| |
| def setUp(self): |
| self._old_stderr = sys.stderr |
| sys.stderr = self.stderr = StringIO() |
| |
| |
| def tearDown(self): |
| sys.stderr = self._old_stderr |
| |
| |
| def test_log_error(self): |
| """Test _log_error().""" |
| db._log_error('error message') |
| self.assertEqual(self.stderr.getvalue(), 'error message\n') |
| |
| |
| class FormatOperationalErrorTestCase(unittest.TestCase): |
| """Tests for _format_operational_error().""" |
| |
| def test_format_operational_error(self): |
| """Test _format_operational_error().""" |
| got = db._format_operational_error(Exception()) |
| self.assertIn('An operational error occurred', got) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |