5 from unittest.mock import MagicMock
8 import unittest_utils # Needed for --unittests_ignore_perf flag
9 import waitable_presence
10 from type.locations import Location
13 class TestWaitablePresence(unittest.TestCase):
14 def test_basic_functionality(self):
15 mock_detector = base_presence.PresenceDetection()
16 mock_detector.update = MagicMock()
17 mock_detector.is_anyone_in_location_now = MagicMock(return_value=True)
18 wp = waitable_presence.WaitablePresenceDetectorWithMemory(
19 1.0, Location.HOUSE, mock_detector
22 (someone_is_home, since) = wp.is_someone_home()
23 mock_detector.update.assert_called_with()
24 mock_detector.is_anyone_in_location_now.assert_called_with(Location.HOUSE)
25 self.assertTrue(changed)
26 self.assertTrue(someone_is_home)
27 self.assertNotEqual(None, since)
32 if __name__ == '__main__':