mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-10 00:28:31 +02:00
113 lines
3.1 KiB
Python
113 lines
3.1 KiB
Python
# Copyright (C) 2012-2021 jrnl contributors
|
|
# License: https://www.gnu.org/licenses/gpl-3.0.html
|
|
|
|
from contextlib import ExitStack
|
|
import os
|
|
from unittest.mock import patch
|
|
|
|
from pytest_bdd import parsers
|
|
from pytest_bdd import when
|
|
from pytest_bdd.parsers import parse
|
|
|
|
from jrnl.cli import cli
|
|
from jrnl.os_compat import split_args
|
|
|
|
|
|
@when(parse('we change directory to "{directory_name}"'))
|
|
def when_we_change_directory(directory_name):
|
|
if not os.path.isdir(directory_name):
|
|
os.mkdir(directory_name)
|
|
|
|
os.chdir(directory_name)
|
|
|
|
|
|
@when(parse('we run "jrnl {command}" and {input_method}\n{user_input}'))
|
|
@when(
|
|
parsers.re(
|
|
'we run "jrnl (?P<command>[^"]+)" and (?P<input_method>enter|pipe) "(?P<user_input>[^"]+)"'
|
|
)
|
|
)
|
|
@when(parse('we run "jrnl" and {input_method} "{user_input}"'))
|
|
@when(parse('we run "jrnl {command}"'))
|
|
@when('we run "jrnl <command>"')
|
|
@when('we run "jrnl"')
|
|
def we_run(
|
|
command,
|
|
config_path,
|
|
user_input,
|
|
cli_run,
|
|
capsys,
|
|
password,
|
|
cache_dir,
|
|
editor,
|
|
keyring,
|
|
input_method,
|
|
mocks,
|
|
):
|
|
assert input_method in ["", "enter", "pipe"]
|
|
is_tty = input_method != "pipe"
|
|
|
|
if cache_dir["exists"]:
|
|
command = command.format(cache_dir=cache_dir["path"])
|
|
|
|
args = split_args(command)
|
|
status = 0
|
|
|
|
if user_input:
|
|
user_input = user_input.splitlines() if is_tty else [user_input]
|
|
|
|
if password:
|
|
password = password.splitlines()
|
|
|
|
if not password and user_input:
|
|
password = user_input
|
|
|
|
with ExitStack() as stack:
|
|
|
|
stack.enter_context(patch("sys.argv", ["jrnl"] + args))
|
|
|
|
mock_stdin = stack.enter_context(
|
|
patch("sys.stdin.read", side_effect=user_input)
|
|
)
|
|
stack.enter_context(patch("sys.stdin.isatty", return_value=is_tty))
|
|
mock_input = stack.enter_context(
|
|
patch("builtins.input", side_effect=user_input)
|
|
)
|
|
mock_getpass = stack.enter_context(
|
|
patch("getpass.getpass", side_effect=password)
|
|
)
|
|
|
|
if "datetime" in mocks:
|
|
stack.enter_context(mocks["datetime"])
|
|
stack.enter_context(mocks["calendar_parse"])
|
|
|
|
# stack.enter_context(patch("datetime.datetime", new=mocks["datetime"]))
|
|
# stack.enter_context(patch("jrnl.time.__get_pdt_calendar", return_value=mocks["calendar_parse"]))
|
|
|
|
stack.enter_context(
|
|
patch("jrnl.install.get_config_path", return_value=config_path)
|
|
)
|
|
stack.enter_context(
|
|
patch("jrnl.config.get_config_path", return_value=config_path)
|
|
)
|
|
mock_editor = stack.enter_context(patch("subprocess.call", side_effect=editor))
|
|
|
|
try:
|
|
cli(args)
|
|
except StopIteration:
|
|
# This happens when input is expected, but don't have any input left
|
|
pass
|
|
except SystemExit as e:
|
|
status = e.code
|
|
|
|
captured = capsys.readouterr()
|
|
|
|
cli_run["status"] = status
|
|
cli_run["stdout"] = captured.out
|
|
cli_run["stderr"] = captured.err
|
|
cli_run["mocks"] = {
|
|
"stdin": mock_stdin,
|
|
"input": mock_input,
|
|
"getpass": mock_getpass,
|
|
"editor": mock_editor,
|
|
}
|