This commit is contained in:
Nick Sweeting
2026-03-23 03:58:32 -07:00
parent 268856bcfb
commit b749b26c5d
286 changed files with 21704 additions and 13480 deletions

View File

@@ -3,7 +3,7 @@ import unittest
from pathlib import Path
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'archivebox.settings')
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "archivebox.settings")
from archivebox.machine.models import Process
@@ -13,26 +13,25 @@ class TestProcessRuntimePaths(unittest.TestCase):
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'],
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')
expected_dir = Path("/tmp/archive/example/chrome/.hooks/on_Snapshot__11_chrome_wait.js")
self.assertEqual(process.runtime_dir, expected_dir)
self.assertEqual(process.stdout_file, expected_dir / 'stdout.log')
self.assertEqual(process.stderr_file, expected_dir / 'stderr.log')
self.assertEqual(process.pid_file, expected_dir / 'process.pid')
self.assertEqual(process.stdout_file, expected_dir / "stdout.log")
self.assertEqual(process.stderr_file, expected_dir / "stderr.log")
self.assertEqual(process.pid_file, expected_dir / "process.pid")
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'],
pwd="/tmp/archive/example",
cmd=["archivebox", "run", "--snapshot-id", "123"],
)
expected_dir = Path('/tmp/archive/example')
expected_dir = Path("/tmp/archive/example")
self.assertEqual(process.runtime_dir, expected_dir)
self.assertEqual(process.stdout_file, expected_dir / 'stdout.log')
self.assertEqual(process.stderr_file, expected_dir / 'stderr.log')
self.assertEqual(process.pid_file, expected_dir / 'process.pid')
self.assertEqual(process.stdout_file, expected_dir / "stdout.log")
self.assertEqual(process.stderr_file, expected_dir / "stderr.log")
self.assertEqual(process.pid_file, expected_dir / "process.pid")