tests/waitable_presence_test.py

   1 #!/usr/bin/env python3
   2
   3 import re
   4 import unittest
   5 from unittest.mock import MagicMock
   6
   7 import base_presence
   8 import unittest_utils  # Needed for --unittests_ignore_perf flag
   9 import waitable_presence
  10 from type.locations import Location
  11
  12
  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
  20         )
  21         changed = wp.wait()
  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)
  28         wp.reset()
  29         wp.shutdown()
  30
  31
  32 if __name__ == '__main__':
  33     unittest.main()