def test_all(): """Test that a script gets read from file, parsed and returned.""" _save_script(test_gm_script, 'test.user.js') gm_manager = greasemonkey.GreasemonkeyManager() assert (gm_manager.all_scripts()[0].name == "qutebrowser test userscript")
def test_required_scripts_are_included(download_stub, tmpdir): test_require_script = textwrap.dedent(""" // ==UserScript== // @name qutebrowser test userscript // @namespace invalid.org // @include http://localhost:*/data/title.html // @match http://trolol* // @exclude https://badhost.xxx/* // @run-at document-start // @require http://localhost/test.js // ==/UserScript== console.log("Script is running."); """) _save_script(test_require_script, 'requiring.user.js') with open(str(tmpdir / 'test.js'), 'w', encoding='UTF-8') as f: f.write("REQUIRED SCRIPT") gm_manager = greasemonkey.GreasemonkeyManager() assert len(gm_manager._in_progress_dls) == 1 for download in gm_manager._in_progress_dls: download.finished.emit() scripts = gm_manager.all_scripts() assert len(scripts) == 1 assert "REQUIRED SCRIPT" in scripts[0].code() # Additionally check that the base script is still being parsed correctly assert "Script is running." in scripts[0].code() assert scripts[0].excludes
def test_get_scripts_by_url(url, expected_matches): """Check Greasemonkey include/exclude rules work.""" _save_script(test_gm_script, 'test.user.js') gm_manager = greasemonkey.GreasemonkeyManager() scripts = gm_manager.scripts_for(QUrl(url)) assert len(scripts.start + scripts.end + scripts.idle) == expected_matches
def test_bad_scheme(caplog): """qute:// isn't in the list of allowed schemes.""" _save_script("var nothing = true;\n", 'nothing.user.js') with caplog.at_level(logging.WARNING): gm_manager = greasemonkey.GreasemonkeyManager() scripts = gm_manager.scripts_for(QUrl('qute://settings')) assert len(scripts.start + scripts.end + scripts.idle) == 0
def test_no_metadata(caplog): """Run on all sites at document-end is the default.""" _save_script("var nothing = true;\n", 'nothing.user.js') with caplog.at_level(logging.WARNING): gm_manager = greasemonkey.GreasemonkeyManager() scripts = gm_manager.scripts_for(QUrl('http://notamatch.invalid/')) assert len(scripts.start + scripts.end + scripts.idle) == 1 assert len(scripts.end) == 1
def _get_script(self, *, namespace, name): source = textwrap.dedent(""" // ==UserScript== // @namespace {} // @name {} // ==/UserScript== """.format(namespace, name)) _save_script(source, 'force.user.js') gm_manager = greasemonkey.GreasemonkeyManager() scripts = gm_manager.all_scripts() assert len(scripts) == 1 return scripts[0]
def test_regex_includes_scripts_for(url, expected_matches): """Ensure our GM @*clude support supports regular expressions.""" gh_dark_example = textwrap.dedent(r""" // ==UserScript== // @include /^https?://((gist|guides|help|raw|status|developer)\.)?github\.com/((?!generated_pages\/preview).)*$/ // @exclude /https?://github\.com/foo/ // @run-at document-start // ==/UserScript== """) _save_script(gh_dark_example, 'test.user.js') gm_manager = greasemonkey.GreasemonkeyManager() scripts = gm_manager.scripts_for(QUrl(url)) assert len(scripts.start + scripts.end + scripts.idle) == expected_matches
def test_utf8_bom(): """Make sure UTF-8 BOMs are stripped from scripts. If we don't strip them, we'll have a BOM in the middle of the file, causing QtWebEngine to not catch the "// ==UserScript==" line. """ script = textwrap.dedent(""" \N{BYTE ORDER MARK}// ==UserScript== // @name qutebrowser test userscript // ==/UserScript== """.lstrip('\n')) _save_script(script, 'bom.user.js') gm_manager = greasemonkey.GreasemonkeyManager() scripts = gm_manager.all_scripts() assert len(scripts) == 1 script = scripts[0] assert '// ==UserScript==' in script.code().splitlines()
def greasemonkey_manager(monkeypatch, data_tmpdir): gm_manager = greasemonkey.GreasemonkeyManager() monkeypatch.setattr(greasemonkey, 'gm_manager', gm_manager)
def test_load_emits_signal(qtbot): gm_manager = greasemonkey.GreasemonkeyManager() with qtbot.wait_signal(gm_manager.scripts_reloaded): gm_manager.load_scripts()
def greasemonkey_manager(data_tmpdir): gm_manager = greasemonkey.GreasemonkeyManager() objreg.register('greasemonkey', gm_manager) yield objreg.delete('greasemonkey')
def gm_manager(monkeypatch) -> greasemonkey.GreasemonkeyManager: gmm = greasemonkey.GreasemonkeyManager() monkeypatch.setattr(greasemonkey, "gm_manager", gmm) return gmm