def setUpClass(cls): fakesleep.monkey_patch() SetTestMode() logger = logging.getLogger('openmotics') logger.setLevel(logging.DEBUG) logger.propagate = False handler = logging.StreamHandler() handler.setLevel(logging.DEBUG) handler.setFormatter(logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")) logger.addHandler(handler)
def setUpClass(cls): SetTestMode() cls._db_filename = tempfile.mktemp() cls.test_db = SqliteDatabase(cls._db_filename) fakesleep.monkey_patch()
def auto_fakesleep(): fakesleep.monkey_patch() yield fakesleep.monkey_restore()
def setUpClass(cls): SetTestMode() fakesleep.monkey_patch() cls.test_db = SqliteDatabase(':memory:')
def setUpClass(cls): super(SchedulingTest, cls).setUpClass() fakesleep.monkey_patch()
def setUpClass(cls): SetTestMode() fakesleep.monkey_patch()
def setUp(self): fakesleep.monkey_patch()
def setUpClass(cls): fakesleep.monkey_patch() SetTestMode() Logs.setup_logger(log_level=logging.DEBUG)
def test_heartbeat(self): events = [] VPNServiceTest._get_task_executor(events) heartbeat = HeartbeatService(url='https://foobar') with mock.patch.object(Config, 'get_entry', return_value=False): heartbeat._beat() tasks = heartbeat._task_executor._tasks.pop() self.assertEqual( { 'events': [(OMBusEvents.VPN_OPEN, False), (OMBusEvents.CLOUD_REACHABLE, False)], 'open_vpn': False }, tasks) self.assertEqual(DEFAULT_SLEEP_TIME, heartbeat._sleep_time) with mock.patch.object(Config, 'get_entry', return_value=True): try: fakesleep.monkey_patch() now = time.time() response = {'success': False} with mock.patch.object( requests, 'post', return_value=VPNServiceTest._fake_response(response)): heartbeat._beat() tasks = heartbeat._task_executor._tasks.pop() self.assertEqual( { 'connectivity': now, 'events': [(OMBusEvents.CLOUD_REACHABLE, True)], 'open_vpn': True }, tasks) self.assertEqual(DEFAULT_SLEEP_TIME, heartbeat._sleep_time) self.assertEqual(now, heartbeat._last_successful_heartbeat) time.sleep(5) now += 5 response = {'success': True} with mock.patch.object( requests, 'post', return_value=VPNServiceTest._fake_response(response)): heartbeat._beat() tasks = heartbeat._task_executor._tasks.pop() self.assertEqual( { 'connectivity': now, 'events': [(OMBusEvents.CLOUD_REACHABLE, True)], 'open_vpn': True }, tasks) self.assertEqual(DEFAULT_SLEEP_TIME, heartbeat._sleep_time) self.assertEqual(now, heartbeat._last_successful_heartbeat) time.sleep(5) now += 5 for collector in heartbeat._collectors.values(): collector._data = collector._collector_thread._name debug_collector = mock.Mock(DebugDumpDataCollector) debug_collector.data = {}, [123] heartbeat._debug_collector = debug_collector response = { 'success': True, 'sleep_time': 2, 'intervals': { 'foo': 0 }, 'configuration': { 'foo': 1 } } with mock.patch.object( requests, 'post', return_value=VPNServiceTest._fake_response( response)) as post: heartbeat._beat() tasks = heartbeat._task_executor._tasks.pop() self.assertEqual( { 'connectivity': now, 'events': [(OMBusEvents.CLOUD_REACHABLE, True)], 'intervals': { 'foo': 0 }, 'configuration': { 'foo': 1 }, 'open_vpn': True }, tasks) self.assertEqual(2, heartbeat._sleep_time) self.assertEqual(now, heartbeat._last_successful_heartbeat) post.assert_called_once_with( 'https://foobar', data={ 'extra_data': json.dumps( { 'errors': 'errorscoll', 'local_ip': 'ip addresscoll', 'debug': {} }, sort_keys=True) }, timeout=10.0) debug_collector.clear.assert_called_once_with([123]) finally: fakesleep.monkey_restore()
def setUpClass(cls): fakesleep.monkey_patch()
def setUpClass(cls): SetTestMode() fakesleep.monkey_patch() fakesleep.reset(seconds=0)
def _setUp(self): monkey_patch() self.addCleanup(monkey_restore)