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 setUpClass(cls):
fakesleep.monkey_patch()
def _setUp(self):
monkey_patch()
self.addCleanup(monkey_restore)