45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
import importlib.util
|
|
import subprocess
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
|
|
SOURCE = Path(__file__).resolve().parents[1] / "quickshell/.config/quickshell/labfy-sway/bin/check-updates.py"
|
|
SPEC = importlib.util.spec_from_file_location("quickshell_updates", SOURCE)
|
|
updates = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(updates)
|
|
|
|
|
|
class CountTests(unittest.TestCase):
|
|
def check_count(self, command, returncode, stdout, stderr, expected):
|
|
result = subprocess.CompletedProcess(command, returncode, stdout, stderr)
|
|
with mock.patch.object(updates.subprocess, "run", return_value=result) as run:
|
|
if isinstance(expected, type) and issubclass(expected, Exception):
|
|
with self.assertRaises(expected):
|
|
updates.count(command)
|
|
else:
|
|
self.assertEqual(updates.count(command), expected)
|
|
run.assert_called_once_with(command, capture_output=True, text=True, timeout=120, check=False)
|
|
|
|
def test_checkupdates_counts_lines(self):
|
|
self.check_count(["checkupdates"], 0, "pkg-a 1 -> 2\npkg-b 3 -> 4\n", "", 2)
|
|
|
|
def test_checkupdates_no_updates(self):
|
|
self.check_count(["checkupdates"], 2, "", "", 0)
|
|
|
|
def test_checkupdates_code_one_is_error(self):
|
|
self.check_count(["checkupdates"], 1, "", "", RuntimeError)
|
|
|
|
def test_checkupdates_code_two_with_stderr_is_error(self):
|
|
self.check_count(["checkupdates"], 2, "", "erreur inattendue", RuntimeError)
|
|
|
|
def test_yay_no_updates(self):
|
|
self.check_count(["yay", "-Qua"], 1, "", "", 0)
|
|
|
|
def test_yay_counts_lines(self):
|
|
self.check_count(["yay", "-Qua"], 0, "aur-a 1 -> 2\naur-b 3 -> 4\n", "", 2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|