mirror of
https://github.com/Screenly/Anthias.git
synced 2025-12-23 22:38:05 -05:00
105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
import os
|
|
import unittest
|
|
from time import sleep
|
|
|
|
import mock
|
|
|
|
import viewer
|
|
from viewer.scheduling import Scheduler
|
|
|
|
logging.disable(logging.CRITICAL)
|
|
|
|
|
|
class ViewerTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.original_splash_delay = viewer.SPLASH_DELAY
|
|
viewer.SPLASH_DELAY = 0
|
|
|
|
self.u = viewer
|
|
|
|
self.m_scheduler = mock.Mock(name='m_scheduler')
|
|
self.p_scheduler = mock.patch.object(
|
|
self.u, 'Scheduler', self.m_scheduler
|
|
)
|
|
|
|
self.m_cmd = mock.Mock(name='m_cmd')
|
|
self.p_cmd = mock.patch.object(self.u.sh, 'Command', self.m_cmd)
|
|
|
|
self.m_killall = mock.Mock(name='killall')
|
|
self.p_killall = mock.patch.object(
|
|
self.u.sh, 'killall', self.m_killall
|
|
)
|
|
|
|
self.m_reload = mock.Mock(name='reload')
|
|
self.p_reload = mock.patch.object(
|
|
self.u, 'load_settings', self.m_reload
|
|
)
|
|
|
|
self.m_sleep = mock.Mock(name='sleep')
|
|
self.p_sleep = mock.patch.object(self.u, 'sleep', self.m_sleep)
|
|
|
|
self.m_loadb = mock.Mock(name='load_browser')
|
|
self.p_loadb = mock.patch.object(self.u, 'load_browser', self.m_loadb)
|
|
|
|
def tearDown(self):
|
|
self.u.SPLASH_DELAY = self.original_splash_delay
|
|
|
|
|
|
def noop(*a, **k):
|
|
return None
|
|
|
|
|
|
class TestEmptyPl(ViewerTestCase):
|
|
@mock.patch('viewer.constants.SERVER_WAIT_TIMEOUT', 0)
|
|
def test_empty(self):
|
|
m_asset_list = mock.Mock()
|
|
m_asset_list.return_value = ([], None)
|
|
|
|
with mock.patch('viewer.scheduling.generate_asset_list', m_asset_list):
|
|
self.u.scheduler = Scheduler()
|
|
|
|
m_asset_list.assert_called_once()
|
|
|
|
|
|
class TestLoadBrowser(ViewerTestCase):
|
|
@mock.patch('pydbus.SessionBus', mock.MagicMock())
|
|
def test_setup(self):
|
|
self.p_loadb.start()
|
|
self.u.setup()
|
|
self.p_loadb.stop()
|
|
|
|
def test_load_browser(self):
|
|
self.m_cmd.return_value.return_value.process.stdout = (
|
|
b'Screenly service start'
|
|
)
|
|
self.p_cmd.start()
|
|
self.u.load_browser()
|
|
self.p_cmd.stop()
|
|
self.m_cmd.assert_called_once_with('ScreenlyWebview')
|
|
|
|
|
|
class TestWatchdog(ViewerTestCase):
|
|
def test_watchdog_should_create_file_if_not_exists(self):
|
|
try:
|
|
os.remove(self.u.utils.WATCHDOG_PATH)
|
|
except OSError:
|
|
pass
|
|
self.u.watchdog()
|
|
self.assertEqual(os.path.exists(self.u.utils.WATCHDOG_PATH), True)
|
|
|
|
def test_watchdog_should_update_mtime(self):
|
|
# for watchdog file creation
|
|
self.u.watchdog()
|
|
mtime = os.path.getmtime(self.u.utils.WATCHDOG_PATH)
|
|
|
|
# Python is too fast?
|
|
sleep(0.01)
|
|
|
|
self.u.watchdog()
|
|
mtime2 = os.path.getmtime(self.u.utils.WATCHDOG_PATH)
|
|
self.assertGreater(mtime2, mtime)
|