#!/usr/bin/python

import common
import cStringIO, os, unittest
from autotest_lib.server.hosts.monitors import monitors_util

class _MockWarnFile(object):
    def __init__(self):
        self.warnings = []


    def write(self, data):
        if data == '\n':
            return
        timestamp, type, message = data.split('\t')
        self.warnings.append((type, message))


class ConsolePatternsTestCase(unittest.TestCase):
    def setUp(self):
        self._warnfile = _MockWarnFile()
        patterns_path = os.path.join(os.path.dirname(__file__),
                                     'console_patterns')
        self._alert_hooks = monitors_util.build_alert_hooks_from_path(
                patterns_path, self._warnfile)
        self._logfile = cStringIO.StringIO()


    def _process_line(self, line):
        input_file = cStringIO.StringIO(line + '\n')
        monitors_util.process_input(input_file, self._logfile,
                                    alert_hooks=self._alert_hooks)


    def _assert_warning_fired(self, type, message):
        key = (type, message)
        self.assert_(key in self._warnfile.warnings,
                     'Warning %s not found in: %s' % (key,
                                                      self._warnfile.warnings))


    def _assert_no_warnings_fired(self):
        self.assertEquals(self._warnfile.warnings, [])


class ConsolePatternsTest(ConsolePatternsTestCase):
    def test_oops(self):
        self._process_line('<0>Oops: 0002 [1] SMP ')
        self._assert_warning_fired('BUG', "machine Oops'd (: 0002 [1] SMP)")


if __name__ == '__main__':
    unittest.main()