ArchiveBox/tests/test_oneshot.py

from pathlib import Path

from .fixtures import *

def test_oneshot_command_exists(tmp_path, disable_extractors_dict):
    os.chdir(tmp_path)
    process = subprocess.run(['archivebox', 'oneshot'], capture_output=True, env=disable_extractors_dict)
    assert not "invalid choice: 'oneshot'" in process.stderr.decode("utf-8")

def test_oneshot_command_saves_page_in_right_folder(tmp_path, disable_extractors_dict):
    disable_extractors_dict.update({"SAVE_DOM": "true"})
    process = subprocess.run(
        [
            "archivebox",
            "oneshot",
            f"--out-dir={tmp_path}",
            "--extract=title,favicon,dom",
            "https://example.com",
        ],
        capture_output=True,
        env=disable_extractors_dict,
    )
    items = ' '.join([str(x) for x in tmp_path.iterdir()])
    current_path = ' '.join([str(x) for x in Path.cwd().iterdir()])
    assert "index.json" in items
    assert not "index.sqlite3" in current_path

def test_oneshot_command_succeeds(tmp_path, disable_extractors_dict):
    disable_extractors_dict.update({"SAVE_DOM": "true"})
    process = subprocess.run(
        [
            "archivebox",
            "oneshot",
            f"--out-dir={tmp_path}",
            "--extract=title,favicon,dom",
            "https://example.com",
        ],
        capture_output=True,
        env=disable_extractors_dict,
    )

    assert process.returncode == 0