diff --git a/command_line_assistant/cli.py b/command_line_assistant/cli.py index aa18637..2a21883 100644 --- a/command_line_assistant/cli.py +++ b/command_line_assistant/cli.py @@ -1,5 +1,4 @@ import argparse -import logging from command_line_assistant.config import CONFIG_DEFAULT_PATH from command_line_assistant.utils import read_stdin @@ -37,9 +36,8 @@ def get_args(): args.record, ] input_data = read_stdin() - if not args.query_string and input_data: - logging.debug("stdin detected") - args.query_string = input_data.strip() + if input_data and args.query_string: + args.query_string = f"{args.query_string} {input_data.strip()}" if not any(optional_args) and not args.query_string: parser.error("Query string is required if no optional arguments are provided.") diff --git a/tests/test_cli.py b/tests/test_cli.py index ab64e19..f0d03c9 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -1,5 +1,4 @@ import sys -from unittest import mock import pytest @@ -16,7 +15,7 @@ def mock_cli_arguments(args): def test_get_args(monkeypatch): monkeypatch.setattr(sys, "argv", mock_cli_arguments(["test"])) - monkeypatch.setattr(cli, "read_stdin", mock.Mock()) + monkeypatch.setattr(cli, "read_stdin", lambda: None) parser, args = cli.get_args() assert parser