dclient/tests/test_alias_v2.py

"""Tests for v2 defaults command: instance and database subcommands."""

from click.testing import CliRunner
from dclient.cli import cli
import json
import pathlib


def test_alias_default_workflow(mocker, tmpdir):
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()

    # Add an alias
    result = runner.invoke(cli, ["alias", "add", "prod", "https://prod.example.com"])
    assert result.exit_code == 0

    # No default yet
    result = runner.invoke(cli, ["default", "instance"])
    assert result.exit_code == 0
    assert "No default instance set" in result.output

    # Set default
    result = runner.invoke(cli, ["default", "instance", "prod"])
    assert result.exit_code == 0

    # Show default
    result = runner.invoke(cli, ["default", "instance"])
    assert result.exit_code == 0
    assert result.output.strip() == "prod"

    # List should show * marker
    result = runner.invoke(cli, ["alias", "list"])
    assert result.exit_code == 0
    assert "* prod" in result.output

    # Clear default
    result = runner.invoke(cli, ["default", "instance", "--clear"])
    assert result.exit_code == 0

    result = runner.invoke(cli, ["default", "instance"])
    assert "No default instance set" in result.output


def test_alias_default_unknown_alias(mocker, tmpdir):
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()
    result = runner.invoke(cli, ["default", "instance", "nonexistent"])
    assert result.exit_code == 1
    assert "No such alias" in result.output


def test_alias_default_db_workflow(mocker, tmpdir):
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()

    # Add an alias
    result = runner.invoke(cli, ["alias", "add", "prod", "https://prod.example.com"])
    assert result.exit_code == 0

    # No default database yet
    result = runner.invoke(cli, ["default", "database", "prod"])
    assert result.exit_code == 0
    assert "No default database set" in result.output

    # Set default database
    result = runner.invoke(cli, ["default", "database", "prod", "main"])
    assert result.exit_code == 0

    # Show default database
    result = runner.invoke(cli, ["default", "database", "prod"])
    assert result.exit_code == 0
    assert result.output.strip() == "main"

    # List should show db info
    result = runner.invoke(cli, ["alias", "list"])
    assert "(db: main)" in result.output

    # Clear default database
    result = runner.invoke(cli, ["default", "database", "prod", "--clear"])
    assert result.exit_code == 0

    result = runner.invoke(cli, ["default", "database", "prod"])
    assert "No default database set" in result.output


def test_alias_default_db_unknown_alias(mocker, tmpdir):
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()
    result = runner.invoke(cli, ["default", "database", "nonexistent", "main"])
    assert result.exit_code == 1
    assert "No such alias" in result.output


def test_alias_remove_clears_default(mocker, tmpdir):
    """Removing the default alias also clears default_instance."""
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()

    runner.invoke(cli, ["alias", "add", "prod", "https://prod.example.com"])
    runner.invoke(cli, ["default", "instance", "prod"])

    # Verify it's set
    config = json.loads((pathlib.Path(tmpdir) / "config.json").read_text())
    assert config["default_instance"] == "prod"

    # Remove alias
    runner.invoke(cli, ["alias", "remove", "prod"])

    config = json.loads((pathlib.Path(tmpdir) / "config.json").read_text())
    assert config["default_instance"] is None
    assert "prod" not in config["instances"]


def test_default_instance_accepts_url(mocker, tmpdir):
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()

    runner.invoke(cli, ["alias", "add", "prod", "https://prod.example.com"])
    result = runner.invoke(cli, ["default", "instance", "https://prod.example.com"])
    assert result.exit_code == 0

    result = runner.invoke(cli, ["default", "instance"])
    assert result.exit_code == 0
    assert result.output.strip() == "prod"


def test_default_database_accepts_url(mocker, tmpdir):
    mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
    runner = CliRunner()

    runner.invoke(cli, ["alias", "add", "prod", "https://prod.example.com"])
    result = runner.invoke(
        cli, ["default", "database", "https://prod.example.com", "main"]
    )
    assert result.exit_code == 0

    result = runner.invoke(cli, ["default", "database", "prod"])
    assert result.exit_code == 0
    assert result.output.strip() == "main"