mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2025-12-27 22:26:00 +00:00
144 lines
3.9 KiB
Python
144 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Integration tests for archivebox config command."""
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
from .fixtures import process, disable_extractors_dict
|
|
|
|
|
|
def test_config_shows_all_config_values(tmp_path, process):
|
|
"""Test that config without args shows all config values."""
|
|
os.chdir(tmp_path)
|
|
|
|
result = subprocess.run(
|
|
['archivebox', 'config'],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
# Should show various config sections
|
|
assert 'TIMEOUT' in result.stdout or 'timeout' in result.stdout.lower()
|
|
# Config should show some output
|
|
assert len(result.stdout) > 100
|
|
|
|
|
|
def test_config_get_specific_key(tmp_path, process):
|
|
"""Test that --get retrieves a specific config value."""
|
|
os.chdir(tmp_path)
|
|
|
|
result = subprocess.run(
|
|
['archivebox', 'config', '--get', 'TIMEOUT'],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
# Should show the TIMEOUT value
|
|
assert 'TIMEOUT' in result.stdout or result.returncode == 0
|
|
|
|
|
|
def test_config_set_value_writes_to_config_file(tmp_path, process):
|
|
"""Test that --set writes config value to ArchiveBox.conf file."""
|
|
os.chdir(tmp_path)
|
|
|
|
# Set a config value
|
|
result = subprocess.run(
|
|
['archivebox', 'config', '--set', 'TIMEOUT=120'],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
# Read the config file directly to verify it was written
|
|
config_file = tmp_path / 'ArchiveBox.conf'
|
|
if config_file.exists():
|
|
config_content = config_file.read_text()
|
|
# Config should contain the set value
|
|
assert 'TIMEOUT' in config_content or 'timeout' in config_content.lower()
|
|
|
|
|
|
def test_config_set_and_get_roundtrip(tmp_path, process):
|
|
"""Test that a value set with --set can be retrieved with --get."""
|
|
os.chdir(tmp_path)
|
|
|
|
# Set a value
|
|
set_result = subprocess.run(
|
|
['archivebox', 'config', '--set', 'TIMEOUT=999'],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
# Verify set was successful
|
|
assert set_result.returncode == 0 or '999' in set_result.stdout
|
|
|
|
# Read the config file directly to verify
|
|
config_file = tmp_path / 'ArchiveBox.conf'
|
|
if config_file.exists():
|
|
config_content = config_file.read_text()
|
|
assert '999' in config_content or 'TIMEOUT' in config_content
|
|
|
|
|
|
def test_config_search_finds_matching_keys(tmp_path, process):
|
|
"""Test that --search finds config keys matching a pattern."""
|
|
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 or result.returncode == 0
|
|
|
|
|
|
def test_config_invalid_key_fails(tmp_path, process):
|
|
"""Test that setting an invalid config key fails."""
|
|
os.chdir(tmp_path)
|
|
|
|
result = subprocess.run(
|
|
['archivebox', 'config', '--set', 'INVALID_KEY_THAT_DOES_NOT_EXIST=value'],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
# Should fail
|
|
assert result.returncode != 0 or 'failed' in result.stdout.lower()
|
|
|
|
|
|
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,
|
|
)
|
|
|
|
# Should fail because there's no = sign
|
|
assert result.returncode != 0
|
|
|
|
|
|
class TestConfigCLI:
|
|
"""Test the CLI interface for config command."""
|
|
|
|
def test_cli_help(self, tmp_path, process):
|
|
"""Test that --help works for config command."""
|
|
os.chdir(tmp_path)
|
|
|
|
result = subprocess.run(
|
|
['archivebox', 'config', '--help'],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
assert result.returncode == 0
|
|
assert '--get' in result.stdout
|
|
assert '--set' in result.stdout
|
|
|
|
|
|
if __name__ == '__main__':
|
|
pytest.main([__file__, '-v'])
|