64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
|
import json
|
||
|
|
import subprocess
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
TEMPLATES = Path(__file__).resolve().parents[1] / "orchestrator" / "templates"
|
||
|
|
LOAD = TEMPLATES / "page-load.js"
|
||
|
|
BOOT = TEMPLATES / "page-boot.js"
|
||
|
|
PROBE = """
|
||
|
|
const fs = require("fs");
|
||
|
|
const vm = require("vm");
|
||
|
|
const code = fs.readFileSync(process.argv[1], "utf8");
|
||
|
|
const steps = JSON.parse(process.argv[2]);
|
||
|
|
const loads = [];
|
||
|
|
const page = {now: 0, selection: ""};
|
||
|
|
const context = vm.createContext({
|
||
|
|
ui: {busy: false, playing: false},
|
||
|
|
document: {hidden: false},
|
||
|
|
window: {getSelection: () => page.selection},
|
||
|
|
Date: {now: () => page.now * 1000},
|
||
|
|
});
|
||
|
|
vm.runInContext(code, context);
|
||
|
|
context.load = () => loads.push(page.now);
|
||
|
|
for (const step of steps) {
|
||
|
|
page.now = step.at;
|
||
|
|
page.selection = step.selection || "";
|
||
|
|
context.ui.busy = Boolean(step.busy);
|
||
|
|
context.document.hidden = Boolean(step.hidden);
|
||
|
|
vm.runInContext("refresh()", context);
|
||
|
|
}
|
||
|
|
console.log(JSON.stringify(loads));
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
def loads(steps: list[dict]) -> list[int]:
|
||
|
|
done = subprocess.run(["node", "-e", PROBE, str(LOAD), json.dumps(steps)], capture_output=True, text=True, timeout=60, check=True)
|
||
|
|
return json.loads(done.stdout)
|
||
|
|
|
||
|
|
|
||
|
|
class TestHubRefresh(unittest.TestCase):
|
||
|
|
def test_a_selection_holds_the_poll_back_for_twenty_seconds_only(self):
|
||
|
|
steps = [
|
||
|
|
{"at": 0},
|
||
|
|
{"at": 10, "selection": "git push"},
|
||
|
|
{"at": 20, "selection": "git push"},
|
||
|
|
{"at": 30, "selection": "git push"},
|
||
|
|
{"at": 40, "selection": "git push"},
|
||
|
|
{"at": 50},
|
||
|
|
{"at": 60, "selection": "other"},
|
||
|
|
{"at": 70, "selection": "other"},
|
||
|
|
]
|
||
|
|
self.assertEqual(loads(steps), [0, 30, 40, 50])
|
||
|
|
|
||
|
|
def test_busy_and_hidden_pages_still_skip_the_poll(self):
|
||
|
|
self.assertEqual(loads([{"at": 0, "busy": True}, {"at": 10, "hidden": True}, {"at": 20}]), [20])
|
||
|
|
|
||
|
|
def test_returning_to_the_window_reloads_at_once(self):
|
||
|
|
boot = BOOT.read_text(encoding="utf-8")
|
||
|
|
self.assertIn('window.addEventListener("focus", refresh);', boot)
|
||
|
|
self.assertIn('document.addEventListener("visibilitychange", refresh);', boot)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|