ArchiveBox/tests/test_cli_shell.py

#!/usr/bin/env python3
"""
Tests for archivebox shell command.
Verify shell command starts Django shell (basic smoke tests only).
"""

import os
import subprocess

from .fixtures import *


def test_shell_command_exists(tmp_path, process):
    """Test that shell command is recognized."""
    os.chdir(tmp_path)

    # Test that the command exists (will fail without input but should recognize command)
    result = subprocess.run(
        ['archivebox', 'shell', '--help'],
        capture_output=True,
        text=True,
        timeout=10,
    )

    # Should show shell help or recognize command
    assert result.returncode in [0, 1, 2]