mirror of
https://github.com/saphid/frame-control.git
synced 2026-10-25 12:01:48 +01:00
58 lines
2.6 KiB
Python
58 lines
2.6 KiB
Python
"""The real-device check must abort before touching a standby or busy Frame."""
|
|
import sandbox # noqa: F401
|
|
import contextlib
|
|
import importlib.util
|
|
import io
|
|
from pathlib import Path
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
|
|
MODULE = Path(__file__).resolve().parents[1]/'scripts/framey-worn-check.py'
|
|
spec = importlib.util.spec_from_file_location('framey_worn_check', MODULE)
|
|
worn = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(worn)
|
|
|
|
|
|
class WornGate(unittest.TestCase):
|
|
def run_gate(self, state, remote_side_effect=None):
|
|
out = io.StringIO()
|
|
with patch.object(worn, 'probe', return_value=state) as probe, \
|
|
patch.object(worn, 'remote', side_effect=remote_side_effect) as remote, \
|
|
contextlib.redirect_stdout(out):
|
|
result = worn.main()
|
|
return result, out.getvalue(), remote, probe
|
|
|
|
def test_standby_is_read_only_under_owned_lock_and_releases_it(self):
|
|
state = {'battery':[{'percent':99}], 'activity':3, 'panelService':True,
|
|
'plugins':[{'id':14,'name':'ProtonDB Badges'}], 'panel':None}
|
|
code, report, remote, probe = self.run_gate(state)
|
|
self.assertEqual(code, 1)
|
|
self.assertIn('standby', report)
|
|
probe.assert_called_once()
|
|
self.assertEqual(remote.call_count, 2)
|
|
self.assertIn('mkdir /tmp/frame-test.lock', remote.call_args_list[0].args[0])
|
|
self.assertIn('rmdir /tmp/frame-test.lock', remote.call_args_list[1].args[0])
|
|
|
|
def test_low_battery_releases_owned_lock(self):
|
|
state = {'battery':[{'percent':19}], 'activity':0, 'panelService':True,
|
|
'plugins':[{'id':14,'name':'ProtonDB Badges'}], 'panel':None}
|
|
code, report, remote, probe = self.run_gate(state)
|
|
self.assertEqual(code, 1)
|
|
self.assertIn('below 20%', report)
|
|
probe.assert_called_once()
|
|
self.assertEqual(remote.call_count, 2)
|
|
|
|
def test_foreign_lock_prevents_all_sensor_reads_and_is_never_removed(self):
|
|
state = {'battery':[{'percent':99}], 'activity':0, 'panelService':True,
|
|
'plugins':[{'id':14,'name':'ProtonDB Badges'}], 'panel':None}
|
|
code, report, remote, probe = self.run_gate(state, RuntimeError('File exists'))
|
|
self.assertEqual(code, 1)
|
|
self.assertIn('File exists', report)
|
|
probe.assert_not_called()
|
|
self.assertEqual(remote.call_count, 1)
|
|
self.assertIn('mkdir /tmp/frame-test.lock', remote.call_args.args[0])
|
|
|
|
def test_screenshot_palette_counts_distinct_colors(self):
|
|
self.assertEqual(worn.color_count('0 #ff00aa 1 #ff00aa 2 #00ff00'), 2)
|