85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
"""Tests for markdown transport helpers — run with `python test_engine_routing.py`."""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent))
|
|
|
|
import _notion_compat as compat
|
|
import notion_writer
|
|
|
|
|
|
def _fake_client():
|
|
c = mock.MagicMock()
|
|
c.request.return_value = {"object": "page", "id": "p1"}
|
|
return c
|
|
|
|
|
|
def test_create_page_markdown():
|
|
c = _fake_client()
|
|
parent = {"type": "data_source_id", "data_source_id": "ds1"}
|
|
compat.create_page_markdown(c, parent, {"Name": {"title": []}}, "# Hi")
|
|
_, kwargs = c.request.call_args
|
|
assert kwargs["path"] == "pages"
|
|
assert kwargs["method"] == "POST"
|
|
assert kwargs["body"]["markdown"] == "# Hi"
|
|
assert kwargs["body"]["parent"] == parent
|
|
|
|
|
|
def test_append_markdown():
|
|
c = _fake_client()
|
|
compat.append_markdown(c, "page-123", "## More")
|
|
_, kwargs = c.request.call_args
|
|
assert kwargs["path"] == "pages/page-123/markdown"
|
|
assert kwargs["method"] == "PATCH"
|
|
assert kwargs["body"]["type"] == "insert_content"
|
|
assert kwargs["body"]["insert_content"]["content"] == "## More"
|
|
assert kwargs["body"]["insert_content"]["position"] == {"type": "end"}
|
|
|
|
|
|
def test_replace_markdown():
|
|
c = _fake_client()
|
|
compat.replace_markdown(c, "page-123", "# Fresh", allow_deleting=True)
|
|
_, kwargs = c.request.call_args
|
|
assert kwargs["path"] == "pages/page-123/markdown"
|
|
assert kwargs["body"]["type"] == "replace_content"
|
|
assert kwargs["body"]["replace_content"]["new_str"] == "# Fresh"
|
|
assert kwargs["body"]["replace_content"]["allow_deleting_content"] is True
|
|
|
|
|
|
def test_extract_local_images_splits():
|
|
content = ("# Title\n\n\n\n"
|
|
"\n\nbody")
|
|
without, imgs = notion_writer.extract_local_images(content)
|
|
assert imgs == [("local", "./b.png")]
|
|
assert "" not in without
|
|
assert "" in without # remote stays
|
|
|
|
|
|
def test_content_has_local_images():
|
|
assert notion_writer._content_has_local_images("") is True
|
|
assert notion_writer._content_has_local_images("") is False
|
|
assert notion_writer._content_has_local_images("no images") is False
|
|
|
|
|
|
def run_all():
|
|
tests = [
|
|
test_create_page_markdown,
|
|
test_append_markdown,
|
|
test_replace_markdown,
|
|
test_extract_local_images_splits,
|
|
test_content_has_local_images,
|
|
]
|
|
for t in tests:
|
|
print(f"\n{t.__name__}")
|
|
t()
|
|
print("\n" + "=" * 50)
|
|
print(f"✅ All {len(tests)} tests passed")
|
|
print("=" * 50)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_all()
|