hugo
e27cc22cfb
Derived from pengchujin/jzsub at 222a90265d2a8797ca258eb1a980cee0863a8311; preserve the upstream MIT license and attribution.
120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
from __future__ import annotations
|
|
|
|
import hashlib
|
|
import importlib.util
|
|
import json
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
|
|
SCRIPT = Path(__file__).resolve().parents[1] / "scripts" / "citation_watermark.py"
|
|
SPEC = importlib.util.spec_from_file_location("citation_watermark", SCRIPT)
|
|
assert SPEC is not None and SPEC.loader is not None
|
|
citation = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(citation)
|
|
|
|
|
|
class CitationWatermarkTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.temporary = tempfile.TemporaryDirectory()
|
|
self.root = Path(self.temporary.name)
|
|
self.manifest = self.root / "download-manifest.json"
|
|
self.manifest.write_text(
|
|
json.dumps(
|
|
{
|
|
"source": {
|
|
"url": "https://www.jove.com/v/69844/a-protocol-for-harvesting-single-cell-suspension-from-mouse-corneas"
|
|
}
|
|
}
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
def tearDown(self) -> None:
|
|
self.temporary.cleanup()
|
|
|
|
def test_attaches_exact_normalized_citation_without_url_by_default(self) -> None:
|
|
path = citation.attach_citation(
|
|
self.manifest,
|
|
"Huang, C., Jiang, N. A Protocol. J. Vis. Exp. doi:10.3791/69844 (2026).",
|
|
)
|
|
|
|
self.assertEqual(
|
|
path.read_text(encoding="utf-8"),
|
|
"Huang, C., Jiang, N. A Protocol. J. Vis. Exp. doi:10.3791/69844 (2026).\n",
|
|
)
|
|
manifest = json.loads(self.manifest.read_text(encoding="utf-8"))
|
|
record = manifest["citation_watermark"]
|
|
self.assertFalse(record["include_source_url"])
|
|
self.assertEqual(record["position"], "top-left")
|
|
self.assertEqual(record["citation_sha256"], hashlib.sha256(path.read_bytes()).hexdigest())
|
|
|
|
def test_optional_source_url_uses_manifest_canonical_url(self) -> None:
|
|
path = citation.attach_citation(
|
|
self.manifest,
|
|
"Formal citation.",
|
|
include_source_url=True,
|
|
)
|
|
|
|
self.assertIn("\nSource: https://www.jove.com/v/69844/", path.read_text(encoding="utf-8"))
|
|
|
|
def test_structured_citation_and_internal_notice_preserve_four_line_layout(self) -> None:
|
|
citation_text = (
|
|
"Huang, C., Jiang, N., Zheng, X., Gu, H., Zhang, L., Ou, S.\n"
|
|
"A Protocol for Harvesting Single-cell Suspension from Mouse Corneas.\n"
|
|
"J. Vis. Exp. (230), e69844, doi:10.3791/69844 (2026)."
|
|
)
|
|
|
|
path = citation.attach_citation(
|
|
self.manifest,
|
|
citation_text,
|
|
notice=citation.DEFAULT_INTERNAL_NOTICE,
|
|
)
|
|
|
|
self.assertEqual(
|
|
path.read_text(encoding="utf-8"),
|
|
"内容引自网络，仅供内部交流\n\n" + citation_text + "\n",
|
|
)
|
|
manifest = json.loads(self.manifest.read_text(encoding="utf-8"))
|
|
record = manifest["citation_watermark"]
|
|
self.assertEqual(record["layout"], "notice-plus-three-line-citation")
|
|
self.assertEqual(record["notice"], citation.DEFAULT_INTERNAL_NOTICE)
|
|
|
|
def test_cli_accepts_three_structured_fields_and_default_notice(self) -> None:
|
|
result = citation.main(
|
|
[
|
|
str(self.manifest),
|
|
"--authors",
|
|
"Authors",
|
|
"--title",
|
|
"Title",
|
|
"--publication",
|
|
"Journal. doi:test.",
|
|
"--notice",
|
|
]
|
|
)
|
|
|
|
self.assertEqual(result, 0)
|
|
self.assertEqual(
|
|
(self.root / "citation-watermark.txt").read_text(encoding="utf-8"),
|
|
"内容引自网络，仅供内部交流\n\nAuthors\nTitle\nJournal. doi:test.\n",
|
|
)
|
|
|
|
def test_refuses_signed_or_credential_like_source_url(self) -> None:
|
|
self.manifest.write_text(
|
|
json.dumps({"source": {"url": "https://cdn.example/video?token=secret"}}),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
with self.assertRaisesRegex(citation.CitationError, "signed or credential"):
|
|
citation.attach_citation(
|
|
self.manifest,
|
|
"Formal citation.",
|
|
include_source_url=True,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|