pumpcontrol/triggersTests.py at master · jacrify/pumpcontrol · 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
__author__ = 'john'

import unittest
import datetime
import triggers

class MockFetcher:
    def __init__(self,maxspeed):
        self.maxspeed=maxspeed
        return

    def get_max_speed(self,lookbackMins, now):
        return self.maxspeed


class TestTriggers(unittest.TestCase):
    def test_scheduled_single(self):
        config={}
        config['scheduledStartHour']=[8]
        config['scheduledEndHour']=[10]

        s=triggers.ScheduledTrigger()

        now=datetime.datetime(2015,12,25,7,59,0)
        self.assertEqual(False,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,8,00,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,9,59,0)
        self.assertEqual(True,s.shouldBeOn(config,now))


        now=datetime.datetime(2015,12,25,10,0,0)
        self.assertEqual(False,s.shouldBeOn(config,now))


    def test_scheduled_multiple(self):
        config={}
        config['scheduledStartHour']=[8,10]
        config['scheduledEndHour']=[9,11]

        s=triggers.ScheduledTrigger()

        now=datetime.datetime(2015,12,25,7,59,0)
        self.assertEqual(False,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,8,00,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,8,59,0)
        self.assertEqual(True,s.shouldBeOn(config,now))


        now=datetime.datetime(2015,12,25,9,59,0)
        self.assertEqual(False,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,10,0,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,11,0,0)
        self.assertEqual(False,s.shouldBeOn(config,now))


    def test_manual(self):
        config={}

        config['manualEndTime']=datetime.datetime(2015,12,25,10,00,0)

        s=triggers.ManualTrigger()

        now=datetime.datetime(2014,12,25,7,59,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,7,59,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,8,00,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,9,59,0)
        self.assertEqual(True,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,10,0,0)
        self.assertEqual(False,s.shouldBeOn(config,now))

        now=datetime.datetime(2015,12,25,11,0,0)
        self.assertEqual(False,s.shouldBeOn(config,now))

        now=datetime.datetime(2016,12,25,11,0,0)
        self.assertEqual(False,s.shouldBeOn(config,now))

        config={}
        self.assertEqual(False,s.shouldBeOn(config,now))

    def test_wind_trigger(self):
        config={}

        config['windpollminutes']=15
        config['onLookbackMins']=60
        config['onTriggerSpeed']=10

        #will cause trigger
        f=MockFetcher(11)

        s=triggers.WindTrigger(f)


        now=datetime.datetime(2014,12,25,8,0,0)
        self.assertEqual(True,s.shouldBeOn(config,now))
        f.maxspeed=0
        #shouldn't trigger, as 15 minutes have not passed so new speed won't be checked.
        now=datetime.datetime(2014,12,25,8,14,0)
        self.assertEqual(True,s.shouldBeOn(config,now))


        now=datetime.datetime(2014,12,25,8,15,0)
        self.assertEqual(False,s.shouldBeOn(config,now))


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