mirror of
https://github.com/Screenly/Anthias.git
synced 2026-01-24 22:18:38 -05:00
127 lines
3.6 KiB
Python
127 lines
3.6 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
import mock
|
|
import unittest
|
|
import os
|
|
import viewer
|
|
from time import sleep
|
|
|
|
|
|
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.SERVER_WAIT_TIMEOUT', 0)
|
|
@mock.patch('viewer.start_loop', side_effect=noop)
|
|
@mock.patch('viewer.view_image', side_effect=noop)
|
|
@mock.patch('viewer.view_webpage', side_effect=noop)
|
|
@mock.patch('viewer.setup', side_effect=noop)
|
|
def test_empty(
|
|
self,
|
|
mock_setup,
|
|
mock_view_webpage,
|
|
mock_view_image,
|
|
mock_start_loop,
|
|
):
|
|
m_asset_list = mock.Mock()
|
|
m_asset_list.return_value = ([], None)
|
|
|
|
with mock.patch.object(self.u, 'generate_asset_list', m_asset_list):
|
|
self.u.main()
|
|
|
|
m_asset_list.assert_called_once()
|
|
mock_setup.assert_called_once()
|
|
mock_view_webpage.assert_called_once()
|
|
self.assertEqual(mock_view_image.call_count, 2)
|
|
mock_start_loop.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 TestSignalHandlers(ViewerTestCase):
|
|
@mock.patch('vlc.Instance', mock.MagicMock())
|
|
@mock.patch(
|
|
'lib.media_player.get_device_type',
|
|
return_value='pi4'
|
|
)
|
|
def test_usr1(self, lookup_mock):
|
|
self.p_killall.start()
|
|
self.assertEqual(None, self.u.sigusr1(None, None))
|
|
self.p_killall.stop()
|
|
|
|
|
|
class TestWatchdog(ViewerTestCase):
|
|
def test_watchdog_should_create_file_if_not_exists(self):
|
|
try:
|
|
os.remove(self.u.WATCHDOG_PATH)
|
|
except OSError:
|
|
pass
|
|
self.u.watchdog()
|
|
self.assertEqual(os.path.exists(self.u.WATCHDOG_PATH), True)
|
|
|
|
def test_watchdog_should_update_mtime(self):
|
|
# for watchdog file creation
|
|
self.u.watchdog()
|
|
mtime = os.path.getmtime(self.u.WATCHDOG_PATH)
|
|
|
|
# Python is too fast?
|
|
sleep(0.01)
|
|
|
|
self.u.watchdog()
|
|
mtime2 = os.path.getmtime(self.u.WATCHDOG_PATH)
|
|
self.assertGreater(mtime2, mtime)
|