sqlit/tests/cli/test_cli_main.py at 33b16c4dcd63f98b4668b9ab040e6eb91e9cbe24 · Maxteabag/sqlit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
from __future__ import annotations

import subprocess
import sys
from pathlib import Path

from tests.conftest import run_cli


def _run_cli_with_stdin(*args: str, stdin: str, env_config_dir: Path) -> subprocess.CompletedProcess:
    """Invoke the sqlit CLI with a piped stdin payload."""
    cmd = [sys.executable, "-m", "sqlit.cli", *args]
    return subprocess.run(
        cmd,
        input=stdin,
        capture_output=True,
        text=True,
        cwd=str(env_config_dir),
        env={
            "SQLIT_CONFIG_DIR": str(env_config_dir),
            "PATH": __import__("os").environ.get("PATH", ""),
            "PYTHONPATH": __import__("os").environ.get("PYTHONPATH", ""),
        },
    )


def test_cli_connections_list_empty(tmp_path: Path, monkeypatch):
    settings_path = tmp_path / "settings.json"
    settings_path.write_text('{"allow_plaintext_credentials": true}', encoding="utf-8")

    monkeypatch.setenv("SQLIT_CONFIG_DIR", str(tmp_path))

    result = run_cli("connections", "list", check=False)

    assert result.returncode == 0
    assert "No saved connections." in result.stdout


def test_url_stdin_creates_connection(tmp_path: Path):
    settings_path = tmp_path / "settings.json"
    settings_path.write_text('{"allow_plaintext_credentials": true}', encoding="utf-8")

    result = _run_cli_with_stdin(
        "connections", "add", "--url-stdin", "--name", "StdinURL",
        stdin="sqlite:///tmp/sqlit-stdin-test.db\n",
        env_config_dir=tmp_path,
    )

    assert result.returncode == 0, result.stderr
    assert "StdinURL" in result.stdout


def test_url_stdin_rejects_when_url_also_provided(tmp_path: Path):
    settings_path = tmp_path / "settings.json"
    settings_path.write_text('{"allow_plaintext_credentials": true}', encoding="utf-8")

    result = _run_cli_with_stdin(
        "connections", "add",
        "--url", "sqlite:///tmp/a.db",
        "--url-stdin",
        "--name", "X",
        stdin="sqlite:///tmp/b.db\n",
        env_config_dir=tmp_path,
    )

    assert result.returncode != 0
    assert "mutually exclusive" in (result.stderr + result.stdout)


def test_password_stdin_mutex_with_password(tmp_path: Path):
    settings_path = tmp_path / "settings.json"
    settings_path.write_text('{"allow_plaintext_credentials": true}', encoding="utf-8")

    result = _run_cli_with_stdin(
        "connect", "postgresql",
        "--name", "X",
        "--server", "localhost",
        "--port", "5432",
        "--database", "d",
        "--username", "u",
        "--password", "cleartext",
        "--password-stdin",
        stdin="frompipe\n",
        env_config_dir=tmp_path,
    )

    assert result.returncode != 0
    assert "mutually exclusive" in (result.stderr + result.stdout)


def test_multiple_stdin_flags_rejected(tmp_path: Path):
    settings_path = tmp_path / "settings.json"
    settings_path.write_text('{"allow_plaintext_credentials": true}', encoding="utf-8")

    result = _run_cli_with_stdin(
        "connections", "edit", "Nonexistent",
        "--password-stdin",
        "--ssh-password-stdin",
        stdin="x\n",
        env_config_dir=tmp_path,
    )

    assert result.returncode != 0
    output = result.stderr + result.stdout
    assert "only one" in output and "stdin" in output


def test_password_stdin_eof_errors_cleanly(tmp_path: Path):
    settings_path = tmp_path / "settings.json"
    settings_path.write_text('{"allow_plaintext_credentials": true}', encoding="utf-8")

    result = _run_cli_with_stdin(
        "connect", "postgresql",
        "--name", "X",
        "--server", "localhost",
        "--port", "5432",
        "--database", "d",
        "--username", "u",
        "--password-stdin",
        stdin="",
        env_config_dir=tmp_path,
    )

    assert result.returncode != 0
    assert "EOF" in (result.stderr + result.stdout)


def test_cli_version_option(tmp_path: Path):
    result = _run_cli_with_stdin(
        "--version",
        stdin="",
        env_config_dir=tmp_path,
    )
    assert result.returncode == 0
    assert "sqlit" in result.stdout


def test_cli_version_short_option(tmp_path: Path):
    result = _run_cli_with_stdin(
        "-v",
        stdin="",
        env_config_dir=tmp_path,
    )
    assert result.returncode == 0
    assert "sqlit" in result.stdout