mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2026-04-04 23:07:56 +10:00
234 lines
8.3 KiB
Python
234 lines
8.3 KiB
Python
"""
|
|
Integration tests for forumdl plugin
|
|
|
|
Tests verify:
|
|
1. Hook script exists
|
|
2. Dependencies installed via validation hooks
|
|
3. Verify deps with abx-pkg
|
|
4. Forum extraction works on forum URLs
|
|
5. JSONL output is correct
|
|
6. Config options work
|
|
7. Handles non-forum URLs gracefully
|
|
"""
|
|
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import uuid
|
|
from pathlib import Path
|
|
import pytest
|
|
|
|
PLUGIN_DIR = Path(__file__).parent.parent
|
|
PLUGINS_ROOT = PLUGIN_DIR.parent
|
|
FORUMDL_HOOK = PLUGIN_DIR / 'on_Snapshot__53_forumdl.py'
|
|
FORUMDL_VALIDATE_HOOK = PLUGIN_DIR / 'on_Crawl__00_validate_forumdl.py'
|
|
TEST_URL = 'https://example.com'
|
|
|
|
# Module-level cache for installed binary path
|
|
_forumdl_binary_path = None
|
|
|
|
def get_forumdl_binary_path():
|
|
"""Get the installed forum-dl binary path from cache or by running validation/installation."""
|
|
global _forumdl_binary_path
|
|
if _forumdl_binary_path:
|
|
return _forumdl_binary_path
|
|
|
|
# Run validation hook to find or install binary
|
|
result = subprocess.run(
|
|
[sys.executable, str(FORUMDL_VALIDATE_HOOK)],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=300
|
|
)
|
|
|
|
# Check if binary was found
|
|
for line in result.stdout.strip().split('\n'):
|
|
if line.strip():
|
|
try:
|
|
record = json.loads(line)
|
|
if record.get('type') == 'InstalledBinary' and record.get('name') == 'forum-dl':
|
|
_forumdl_binary_path = record.get('abspath')
|
|
return _forumdl_binary_path
|
|
elif record.get('type') == 'Dependency' and record.get('bin_name') == 'forum-dl':
|
|
# Need to install via pip hook
|
|
pip_hook = PLUGINS_ROOT / 'pip' / 'on_Dependency__install_using_pip_provider.py'
|
|
dependency_id = str(uuid.uuid4())
|
|
|
|
# Build command with overrides if present
|
|
cmd = [
|
|
sys.executable, str(pip_hook),
|
|
'--dependency-id', dependency_id,
|
|
'--bin-name', record['bin_name']
|
|
]
|
|
if 'overrides' in record:
|
|
cmd.extend(['--overrides', json.dumps(record['overrides'])])
|
|
|
|
install_result = subprocess.run(
|
|
cmd,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=300
|
|
)
|
|
|
|
# Parse InstalledBinary from pip installation
|
|
for install_line in install_result.stdout.strip().split('\n'):
|
|
if install_line.strip():
|
|
try:
|
|
install_record = json.loads(install_line)
|
|
if install_record.get('type') == 'InstalledBinary' and install_record.get('name') == 'forum-dl':
|
|
_forumdl_binary_path = install_record.get('abspath')
|
|
return _forumdl_binary_path
|
|
except json.JSONDecodeError:
|
|
pass
|
|
|
|
# Installation failed - print debug info
|
|
if not _forumdl_binary_path:
|
|
print(f"\n=== forum-dl installation failed ===", file=sys.stderr)
|
|
print(f"stdout: {install_result.stdout}", file=sys.stderr)
|
|
print(f"stderr: {install_result.stderr}", file=sys.stderr)
|
|
print(f"returncode: {install_result.returncode}", file=sys.stderr)
|
|
return None
|
|
except json.JSONDecodeError:
|
|
pass
|
|
|
|
return None
|
|
|
|
def test_hook_script_exists():
|
|
"""Verify on_Snapshot hook exists."""
|
|
assert FORUMDL_HOOK.exists(), f"Hook not found: {FORUMDL_HOOK}"
|
|
|
|
|
|
def test_forumdl_validate_hook():
|
|
"""Test forum-dl validate hook checks for forum-dl."""
|
|
# Run forum-dl validate hook
|
|
result = subprocess.run(
|
|
[sys.executable, str(FORUMDL_VALIDATE_HOOK)],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=30
|
|
)
|
|
|
|
# Hook exits 0 if all binaries found, 1 if any not found
|
|
# Parse output for InstalledBinary and Dependency records
|
|
found_binary = False
|
|
found_dependency = False
|
|
|
|
for line in result.stdout.strip().split('\n'):
|
|
if line.strip():
|
|
try:
|
|
record = json.loads(line)
|
|
if record.get('type') == 'InstalledBinary':
|
|
if record['name'] == 'forum-dl':
|
|
assert record['abspath'], "forum-dl should have abspath"
|
|
found_binary = True
|
|
elif record.get('type') == 'Dependency':
|
|
if record['bin_name'] == 'forum-dl':
|
|
found_dependency = True
|
|
except json.JSONDecodeError:
|
|
pass
|
|
|
|
# forum-dl should either be found (InstalledBinary) or missing (Dependency)
|
|
assert found_binary or found_dependency, \
|
|
"forum-dl should have either InstalledBinary or Dependency record"
|
|
|
|
|
|
def test_verify_deps_with_abx_pkg():
|
|
"""Verify forum-dl is installed by calling the REAL validation and installation hooks."""
|
|
binary_path = get_forumdl_binary_path()
|
|
assert binary_path, (
|
|
"forum-dl must be installed successfully via validation hook and pip provider. "
|
|
"NOTE: forum-dl has a dependency on cchardet which does not compile on Python 3.14+ "
|
|
"due to removed longintrepr.h header. This is a known compatibility issue with forum-dl."
|
|
)
|
|
assert Path(binary_path).is_file(), f"Binary path must be a valid file: {binary_path}"
|
|
|
|
|
|
def test_handles_non_forum_url():
|
|
"""Test that forum-dl extractor handles non-forum URLs gracefully via hook."""
|
|
import os
|
|
|
|
binary_path = get_forumdl_binary_path()
|
|
assert binary_path, "Binary must be installed for this test"
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
tmpdir = Path(tmpdir)
|
|
|
|
env = os.environ.copy()
|
|
env['FORUMDL_BINARY'] = binary_path
|
|
|
|
# Run forum-dl extraction hook on non-forum URL
|
|
result = subprocess.run(
|
|
[sys.executable, str(FORUMDL_HOOK), '--url', 'https://example.com', '--snapshot-id', 'test789'],
|
|
cwd=tmpdir,
|
|
capture_output=True,
|
|
text=True,
|
|
env=env,
|
|
timeout=60
|
|
)
|
|
|
|
# Should exit 0 even for non-forum URL (graceful handling)
|
|
assert result.returncode == 0, f"Should handle non-forum URL gracefully: {result.stderr}"
|
|
|
|
# Verify JSONL output
|
|
assert 'STATUS=' in result.stdout, "Should report status"
|
|
assert 'RESULT_JSON=' in result.stdout, "Should output RESULT_JSON"
|
|
|
|
# Parse JSONL result
|
|
result_json = None
|
|
for line in result.stdout.split('\n'):
|
|
if line.startswith('RESULT_JSON='):
|
|
result_json = json.loads(line.split('=', 1)[1])
|
|
break
|
|
|
|
assert result_json, "Should have RESULT_JSON"
|
|
assert result_json['extractor'] == 'forumdl'
|
|
|
|
|
|
def test_config_save_forumdl_false_skips():
|
|
"""Test that SAVE_FORUMDL=False causes skip."""
|
|
import os
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
env = os.environ.copy()
|
|
env['SAVE_FORUMDL'] = 'False'
|
|
|
|
result = subprocess.run(
|
|
[sys.executable, str(FORUMDL_HOOK), '--url', TEST_URL, '--snapshot-id', 'test999'],
|
|
cwd=tmpdir,
|
|
capture_output=True,
|
|
text=True,
|
|
env=env,
|
|
timeout=30
|
|
)
|
|
|
|
assert result.returncode == 0, f"Should exit 0 when skipping: {result.stderr}"
|
|
assert 'STATUS=' in result.stdout
|
|
|
|
|
|
def test_config_timeout():
|
|
"""Test that FORUMDL_TIMEOUT config is respected."""
|
|
import os
|
|
|
|
binary_path = get_forumdl_binary_path()
|
|
assert binary_path, "Binary must be installed for this test"
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
env = os.environ.copy()
|
|
env['FORUMDL_BINARY'] = binary_path
|
|
env['FORUMDL_TIMEOUT'] = '5'
|
|
|
|
result = subprocess.run(
|
|
[sys.executable, str(FORUMDL_HOOK), '--url', 'https://example.com', '--snapshot-id', 'testtimeout'],
|
|
cwd=tmpdir,
|
|
capture_output=True,
|
|
text=True,
|
|
env=env,
|
|
timeout=30
|
|
)
|
|
|
|
assert result.returncode == 0, "Should complete without hanging"
|
|
|
|
if __name__ == '__main__':
|
|
pytest.main([__file__, '-v'])
|