blob: c5c43129f447a4112663e4ab789ca53278c4364b [file] [log] [blame]
#!/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()