""" Integration tests for singlefile plugin Tests verify: 1. Hook scripts exist with correct naming 2. CLI-based singlefile extraction works 3. Dependencies available via abx-pkg 4. Output contains valid HTML 5. Connects to Chrome session via CDP when available 6. Works with extensions loaded (ublock, etc.) """ import json import os import subprocess import tempfile from pathlib import Path import pytest from archivebox.plugins.chrome.tests.chrome_test_helpers import ( get_test_env, get_plugin_dir, get_hook_script, chrome_session, cleanup_chrome, ) PLUGIN_DIR = get_plugin_dir(__file__) SNAPSHOT_HOOK = get_hook_script(PLUGIN_DIR, 'on_Snapshot__*_singlefile.py') TEST_URL = "https://example.com" def test_snapshot_hook_exists(): """Verify snapshot extraction hook exists""" assert SNAPSHOT_HOOK is not None and SNAPSHOT_HOOK.exists(), f"Snapshot hook not found in {PLUGIN_DIR}" def test_snapshot_hook_priority(): """Test that snapshot hook has correct priority (50)""" filename = SNAPSHOT_HOOK.name assert "50" in filename, "SingleFile snapshot hook should have priority 50" assert filename.startswith("on_Snapshot__50_"), "Should follow priority naming convention" def test_verify_deps_with_abx_pkg(): """Verify dependencies are available via abx-pkg.""" from abx_pkg import Binary, EnvProvider EnvProvider.model_rebuild() # Verify node is available node_binary = Binary(name='node', binproviders=[EnvProvider()]) node_loaded = node_binary.load() assert node_loaded and node_loaded.abspath, "Node.js required for singlefile plugin" def test_singlefile_cli_archives_example_com(): """Test that singlefile CLI archives example.com and produces valid HTML.""" with tempfile.TemporaryDirectory() as tmpdir: tmpdir = Path(tmpdir) env = get_test_env() env['SINGLEFILE_ENABLED'] = 'true' # Run singlefile snapshot hook result = subprocess.run( ['python', str(SNAPSHOT_HOOK), f'--url={TEST_URL}', '--snapshot-id=test789'], cwd=tmpdir, capture_output=True, text=True, env=env, timeout=120 ) assert result.returncode == 0, f"Hook execution failed: {result.stderr}" # Verify output file exists output_file = tmpdir / 'singlefile.html' assert output_file.exists(), f"singlefile.html not created. stdout: {result.stdout}, stderr: {result.stderr}" # Verify it contains real HTML html_content = output_file.read_text() assert len(html_content) > 500, "Output file too small to be valid HTML" assert '' in html_content or ' 500, "Output file too small" assert 'Example Domain' in html_content, "Should contain example.com content" else: # If singlefile couldn't connect to Chrome, it may have failed # Check if it mentioned browser-server in its args (indicating it tried to use CDP) assert result.returncode == 0 or 'browser-server' in result.stderr or 'cdp' in result.stderr.lower(), \ f"Singlefile should attempt CDP connection. stderr: {result.stderr}" def test_singlefile_disabled_skips(): """Test that SINGLEFILE_ENABLED=False exits without JSONL.""" with tempfile.TemporaryDirectory() as tmpdir: tmpdir = Path(tmpdir) env = get_test_env() env['SINGLEFILE_ENABLED'] = 'False' result = subprocess.run( ['python', str(SNAPSHOT_HOOK), f'--url={TEST_URL}', '--snapshot-id=test-disabled'], cwd=tmpdir, capture_output=True, text=True, env=env, timeout=30 ) assert result.returncode == 0, f"Should exit 0 when disabled: {result.stderr}" # Should NOT emit JSONL when disabled jsonl_lines = [line for line in result.stdout.strip().split('\n') if line.strip().startswith('{')] assert len(jsonl_lines) == 0, f"Should not emit JSONL when disabled, but got: {jsonl_lines}" if __name__ == '__main__': pytest.main([__file__, '-v'])