blob: 557f32eb27d1a3a1102d169b24395e54564327d7 [file] [log] [blame] [edit]
#!/usr/bin/python
import sys
import unittest
from cStringIO 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()