99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:
|
|
|
|
# Copyright 2016-2017 Florian Bruhin (The Compiler) <mail@qutebrowser.org>
|
|
#
|
|
# This file is part of qutebrowser.
|
|
#
|
|
# qutebrowser is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# qutebrowser is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with qutebrowser. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import sys
|
|
import json
|
|
import textwrap
|
|
import os
|
|
import signal
|
|
|
|
import pytest_bdd as bdd
|
|
bdd.scenarios('editor.feature')
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse('I set up a fake editor replacing "{text}" by '
|
|
'"{replacement}"'))
|
|
def set_up_editor_replacement(quteproc, server, tmpdir, text, replacement):
|
|
"""Set up editor.command to a small python script doing a replacement."""
|
|
text = text.replace('(port)', str(server.port))
|
|
script = tmpdir / 'script.py'
|
|
script.write(textwrap.dedent("""
|
|
import sys
|
|
|
|
with open(sys.argv[1], encoding='utf-8') as f:
|
|
data = f.read()
|
|
|
|
data = data.replace("{text}", "{replacement}")
|
|
|
|
with open(sys.argv[1], 'w', encoding='utf-8') as f:
|
|
f.write(data)
|
|
""".format(text=text, replacement=replacement)))
|
|
editor = json.dumps([sys.executable, str(script), '{}'])
|
|
quteproc.set_setting('editor.command', editor)
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse('I set up a fake editor returning "{text}"'))
|
|
def set_up_editor(quteproc, server, tmpdir, text):
|
|
"""Set up editor.command to a small python script inserting a text."""
|
|
script = tmpdir / 'script.py'
|
|
script.write(textwrap.dedent("""
|
|
import sys
|
|
|
|
with open(sys.argv[1], 'w', encoding='utf-8') as f:
|
|
f.write({text!r})
|
|
""".format(text=text)))
|
|
editor = json.dumps([sys.executable, str(script), '{}'])
|
|
quteproc.set_setting('editor.command', editor)
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse('I set up a fake editor returning empty text'))
|
|
def set_up_editor_empty(quteproc, server, tmpdir):
|
|
"""Set up editor.command to a small python script inserting empty text."""
|
|
set_up_editor(quteproc, server, tmpdir, "")
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse('I set up a fake editor that waits'))
|
|
def set_up_editor_wait(quteproc, server, tmpdir):
|
|
"""Set up editor.command to a small python script inserting a text."""
|
|
pidfile = tmpdir / 'editor_pid'
|
|
script = tmpdir / 'script.py'
|
|
script.write(textwrap.dedent("""
|
|
import os
|
|
import sys
|
|
import time
|
|
import signal
|
|
|
|
with open(r'{pidfile}', 'w') as f:
|
|
f.write(str(os.getpid()))
|
|
|
|
signal.signal(signal.SIGTERM, lambda s, f: sys.exit(0))
|
|
|
|
time.sleep(100)
|
|
""".format(pidfile=pidfile)))
|
|
editor = json.dumps([sys.executable, str(script), '{}'])
|
|
quteproc.set_setting('editor.command', editor)
|
|
|
|
|
|
@bdd.when(bdd.parsers.parse('I kill the waiting editor'))
|
|
def kill_editor_wait(quteproc, server, tmpdir):
|
|
"""Kill the waiting editor."""
|
|
pidfile = tmpdir / 'editor_pid'
|
|
pid = int(pidfile.read())
|
|
os.kill(pid, signal.SIGTERM)
|