blob: ff4225c9798ec93a7ad008b60aeabfed033af56f (
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
|
#!/usr/bin/env python3
import re
import unittest
from unittest.mock import MagicMock
import base_presence
import unittest_utils # Needed for --unittests_ignore_perf flag
import waitable_presence
from type.locations import Location
class TestWaitablePresence(unittest.TestCase):
def test_basic_functionality(self):
mock_detector = base_presence.PresenceDetection()
mock_detector.update = MagicMock()
mock_detector.is_anyone_in_location_now = MagicMock(return_value=True)
wp = waitable_presence.WaitablePresenceDetectorWithMemory(
1.0, Location.HOUSE, mock_detector
)
changed = wp.wait()
(someone_is_home, since) = wp.is_someone_home()
mock_detector.update.assert_called_with()
mock_detector.is_anyone_in_location_now.assert_called_with(Location.HOUSE)
self.assertTrue(changed)
self.assertTrue(someone_is_home)
self.assertNotEqual(None, since)
wp.reset()
wp.shutdown()
if __name__ == '__main__':
unittest.main()
|
