munin-plugins/test_pg_multigraph.py at master · exhuma/munin-plugins · GitHub

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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import unittest
from imp import load_source


class TestVersionedQueries(unittest.TestCase):

    def test_min(self):
        pgmg = load_source('pgmg', 'pg_multigraph')
        queries = {
            'foo': {
                (1, 0, 0): 'foo-100',
                (2, 0, 0): 'foo-200',
                (3, 0, 0): 'foo-300',
            }
        }
        result = pgmg.queries_for_version(queries, (1, 0, 0))
        self.assertEqual(result, {'foo': 'foo-100'})

    def test_mid(self):
        pgmg = load_source('pgmg', 'pg_multigraph')
        queries = {
            'foo': {
                (1, 0, 0): 'foo-100',
                (2, 0, 0): 'foo-200',
                (3, 0, 0): 'foo-300',
            }
        }
        result = pgmg.queries_for_version(queries, (2, 0, 0))
        self.assertEqual(result, {'foo': 'foo-200'})

    def test_max(self):
        pgmg = load_source('pgmg', 'pg_multigraph')
        queries = {
            'foo': {
                (1, 0, 0): 'foo-100',
                (2, 0, 0): 'foo-200',
                (3, 0, 0): 'foo-300',
            }
        }
        result = pgmg.queries_for_version(queries, (3, 0, 0))
        self.assertEqual(result, {'foo': 'foo-300'})

    def test_below_min(self):
        pgmg = load_source('pgmg', 'pg_multigraph')
        queries = {
            'foo': {
                (1, 0, 0): 'foo-100',
                (2, 0, 0): 'foo-200',
                (3, 0, 0): 'foo-300',
            }
        }
        result = pgmg.queries_for_version(queries, (0, 5, 0))
        self.assertEqual(result, {})

    def test_between_versions(self):
        pgmg = load_source('pgmg', 'pg_multigraph')
        queries = {
            'foo': {
                (1, 0, 0): 'foo-100',
                (2, 0, 0): 'foo-200',
                (3, 0, 0): 'foo-300',
            }
        }
        result = pgmg.queries_for_version(queries, (2, 1, 0))
        self.assertEqual(result, {'foo': 'foo-200'})

    def test_after_max(self):
        pgmg = load_source('pgmg', 'pg_multigraph')
        queries = {
            'foo': {
                (1, 0, 0): 'foo-100',
                (2, 0, 0): 'foo-200',
                (3, 0, 0): 'foo-300',
            }
        }
        result = pgmg.queries_for_version(queries, (4, 0, 0))
        self.assertEqual(result, {'foo': 'foo-300'})