summaryrefslogtreecommitdiffstats
path: root/tests/test_log4jlogster.py
blob: 56715d798b0719b0631b6645a7dbc31365ee9e54 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from logster.parsers.Log4jLogster import Log4jLogster
from logster.logster_helper import LogsterParsingException
import unittest

class TestLog4jLogster(unittest.TestCase):

    def setUp(self):
        self.logster = Log4jLogster()
        self.log_tmpl = '13:55:36 %s com.etsy.LogsterTest - This is a message'

    def test_valid_lines(self):
        self.logster.parse_line(self.log_tmpl % 'WARN')
        self.logster.parse_line(self.log_tmpl % 'ERROR')
        self.logster.parse_line(self.log_tmpl % 'ERROR')
        self.assertEqual(1, self.logster.WARN)
        self.assertEqual(2, self.logster.ERROR)
        self.assertEqual(0, self.logster.FATAL)

        self.assertRaises(LogsterParsingException, self.logster.parse_line, self.log_tmpl % 'DEBUG')
        self.assertFalse(hasattr(self.logster, 'DEBUG'))

    def test_valid_lines_non_default(self):
        debug_logster = Log4jLogster('--log-levels=DEBUG')
        debug_logster.parse_line(self.log_tmpl % 'DEBUG')
        self.assertEqual(1, debug_logster.DEBUG)

        self.assertRaises(LogsterParsingException, debug_logster.parse_line, self.log_tmpl % 'WARN')
        self.assertRaises(LogsterParsingException, debug_logster.parse_line, self.log_tmpl % 'ERROR')
        self.assertFalse(hasattr(debug_logster, 'WARN'))
        self.assertFalse(hasattr(debug_logster, 'ERROR'))
        self.assertFalse(hasattr(debug_logster, 'FATAL'))

    def test_metrics_1sec(self):
        self.test_valid_lines()
        metrics = self.logster.get_state(1)
        self.assertEqual(3, len(metrics))

        expected = {"WARN": 1,
                    "ERROR": 2,
                    "FATAL": 0,
                   }
        for m in metrics:
            self.assertEqual(expected[m.name], m.value)

    def test_metrics_2sec(self):
        self.test_valid_lines()
        metrics = self.logster.get_state(2)
        self.assertEqual(3, len(metrics))

        expected = {"WARN": 0.5,
                    "ERROR": 1,
                    "FATAL": 0,
                   }
        for m in metrics:
            self.assertEqual(expected[m.name], m.value)

    def test_invalid_line(self):
        self.assertRaises(LogsterParsingException, self.logster.parse_line, 'invalid log entry')

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