Home | History | Annotate | Download | only in tko
      1 #!/usr/bin/python
      2 
      3 import sys
      4 import unittest
      5 
      6 from cStringIO import StringIO
      7 
      8 import common
      9 from autotest_lib.tko import db
     10 
     11 
     12 class LogErrorTestCase(unittest.TestCase):
     13     """Tests for _log_error()."""
     14 
     15     def setUp(self):
     16         self._old_stderr = sys.stderr
     17         sys.stderr = self.stderr = StringIO()
     18 
     19 
     20     def tearDown(self):
     21         sys.stderr = self._old_stderr
     22 
     23 
     24     def test_log_error(self):
     25         """Test _log_error()."""
     26         db._log_error('error message')
     27         self.assertEqual(self.stderr.getvalue(), 'error message\n')
     28 
     29 
     30 class FormatOperationalErrorTestCase(unittest.TestCase):
     31     """Tests for _format_operational_error()."""
     32 
     33     def test_format_operational_error(self):
     34         """Test _format_operational_error()."""
     35         got = db._format_operational_error(Exception())
     36         self.assertIn('An operational error occurred', got)
     37 
     38 
     39 if __name__ == "__main__":
     40     unittest.main()
     41