ArchiveBox/archivebox/tests/test_cli_config.py

#!/usr/bin/env python3
"""
Comprehensive tests for archivebox config command.
Verify config reads/writes ArchiveBox.conf file correctly.
"""

import os
import subprocess


def test_config_displays_all_config(tmp_path, process):
    """Test that config without args displays all configuration."""
    os.chdir(tmp_path)
    result = subprocess.run(["archivebox", "config"], capture_output=True, text=True)

    assert result.returncode == 0
    output = result.stdout
    # Should show config sections
    assert len(output) > 100
    # Should show at least some standard config keys
    assert "TIMEOUT" in output or "OUTPUT_PERMISSIONS" in output


def test_config_get_specific_key(tmp_path, process):
    """Test that config --get KEY retrieves specific value."""
    os.chdir(tmp_path)
    result = subprocess.run(
        ["archivebox", "config", "--get", "TIMEOUT"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0
    assert "TIMEOUT" in result.stdout


def test_config_set_writes_to_file(tmp_path, process):
    """Test that config --set KEY=VALUE writes to ArchiveBox.conf."""
    os.chdir(tmp_path)

    result = subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=120"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0

    # Verify config file was updated
    config_file = tmp_path / "ArchiveBox.conf"
    assert config_file.exists()

    content = config_file.read_text()
    assert "TIMEOUT" in content or "120" in content


def test_config_set_and_get_roundtrip(tmp_path, process):
    """Test that set value can be retrieved with get."""
    os.chdir(tmp_path)

    # Set a unique value
    subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=987"],
        capture_output=True,
        text=True,
    )

    # Get the value back
    result = subprocess.run(
        ["archivebox", "config", "--get", "TIMEOUT"],
        capture_output=True,
        text=True,
    )

    assert "987" in result.stdout


def test_config_set_multiple_values(tmp_path, process):
    """Test setting multiple config values at once."""
    os.chdir(tmp_path)

    result = subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=111", "YTDLP_TIMEOUT=222"],
        capture_output=True,
        text=True,
    )

    assert result.returncode == 0

    # Verify both were written
    config_file = tmp_path / "ArchiveBox.conf"
    content = config_file.read_text()
    assert "111" in content
    assert "222" in content


def test_config_set_invalid_key_fails(tmp_path, process):
    """Test that setting invalid config key fails."""
    os.chdir(tmp_path)

    result = subprocess.run(
        ["archivebox", "config", "--set", "TOTALLY_INVALID_KEY_XYZ=value"],
        capture_output=True,
        text=True,
    )

    assert result.returncode != 0


def test_config_set_requires_equals_sign(tmp_path, process):
    """Test that set requires KEY=VALUE format."""
    os.chdir(tmp_path)

    result = subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT"],
        capture_output=True,
        text=True,
    )

    assert result.returncode != 0


def test_config_search_finds_keys(tmp_path, process):
    """Test that config --search finds matching keys."""
    os.chdir(tmp_path)

    result = subprocess.run(
        ["archivebox", "config", "--search", "TIMEOUT"],
        capture_output=True,
        text=True,
    )

    # Should find timeout-related config
    assert "TIMEOUT" in result.stdout


def test_config_preserves_existing_values(tmp_path, process):
    """Test that setting new values preserves existing ones."""
    os.chdir(tmp_path)

    # Set first value
    subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=100"],
        capture_output=True,
    )

    # Set second value
    subprocess.run(
        ["archivebox", "config", "--set", "YTDLP_TIMEOUT=200"],
        capture_output=True,
    )

    # Verify both are in config file
    config_file = tmp_path / "ArchiveBox.conf"
    content = config_file.read_text()
    assert "TIMEOUT" in content
    assert "YTDLP_TIMEOUT" in content


def test_config_file_is_valid_toml(tmp_path, process):
    """Test that config file remains valid TOML after set."""
    os.chdir(tmp_path)

    subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=150"],
        capture_output=True,
    )

    config_file = tmp_path / "ArchiveBox.conf"
    content = config_file.read_text()

    # Basic TOML validation - should have sections and key=value pairs
    assert "[" in content or "=" in content


def test_config_updates_existing_value(tmp_path, process):
    """Test that setting same key twice updates the value."""
    os.chdir(tmp_path)

    # Set initial value
    subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=100"],
        capture_output=True,
    )

    # Update to new value
    subprocess.run(
        ["archivebox", "config", "--set", "TIMEOUT=200"],
        capture_output=True,
    )

    # Get current value
    result = subprocess.run(
        ["archivebox", "config", "--get", "TIMEOUT"],
        capture_output=True,
        text=True,
    )

    # Should show updated value
    assert "200" in result.stdout