mirror of
https://github.com/Screenly/Anthias.git
synced 2026-01-03 03:38:42 -05:00
115 lines
3.5 KiB
Python
115 lines
3.5 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from nose.tools import eq_
|
|
import mock
|
|
import unittest
|
|
import os
|
|
from time import sleep
|
|
|
|
|
|
class ViewerTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
import viewer
|
|
|
|
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
|
|
|
|
|
|
class TestEmptyPl(ViewerTestCase):
|
|
noop = lambda *a, **k: None
|
|
|
|
@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_hotspot', side_effect=noop)
|
|
@mock.patch('viewer.setup', side_effect=noop)
|
|
def test_empty(
|
|
self,
|
|
mock_setup,
|
|
mock_setup_hotspot,
|
|
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_setup_hotspot.assert_called_once()
|
|
mock_view_webpage.assert_called_once()
|
|
mock_view_image.assert_called_once()
|
|
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 = '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):
|
|
def test_usr1(self):
|
|
self.p_killall.start()
|
|
eq_(None, self.u.sigusr1(None, None))
|
|
self.m_killall.assert_called_once_with('omxplayer.bin', _ok_code=[1])
|
|
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:
|
|
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)
|