mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2026-09-14 02:56:11 +05:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "archivebox.settings")
|
|
|
|
|
|
from archivebox.machine.models import Process
|
|
|
|
|
|
class TestProcessRuntimePaths:
|
|
def test_hook_processes_use_isolated_runtime_dir(self):
|
|
process = Process(
|
|
process_type=Process.TypeChoices.HOOK,
|
|
pwd="/tmp/archive/example/chrome",
|
|
cmd=["node", "/plugins/chrome/on_Snapshot__11_chrome_wait.js", "--url=https://example.com"],
|
|
)
|
|
|
|
expected_dir = Path("/tmp/archive/example/chrome/.hooks/on_Snapshot__11_chrome_wait.js")
|
|
assert process.runtime_dir == expected_dir
|
|
assert process.stdout_file == expected_dir / "stdout.log"
|
|
assert process.stderr_file == expected_dir / "stderr.log"
|
|
|
|
def test_non_hook_processes_keep_runtime_files_in_pwd(self):
|
|
process = Process(
|
|
process_type=Process.TypeChoices.WORKER,
|
|
pwd="/tmp/archive/example",
|
|
cmd=["archivebox", "run", "--snapshot-id", "123"],
|
|
)
|
|
|
|
expected_dir = Path("/tmp/archive/example")
|
|
assert process.runtime_dir == expected_dir
|
|
assert process.stdout_file == expected_dir / "stdout.log"
|
|
assert process.stderr_file == expected_dir / "stderr.log"
|