mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-06 20:42:52 +08:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import importlib.util
|
|
import pathlib
|
|
import unittest
|
|
|
|
|
|
ROOT = pathlib.Path(__file__).resolve().parents[1]
|
|
MODULE_PATH = ROOT / "app" / "utils" / "note_helper.py"
|
|
spec = importlib.util.spec_from_file_location("note_helper", MODULE_PATH)
|
|
if spec is None or spec.loader is None:
|
|
raise ImportError("note_helper module spec not found")
|
|
note_helper = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(note_helper)
|
|
|
|
|
|
class TestNoteHelper(unittest.TestCase):
|
|
def test_prepend_source_link_adds_header_at_top(self):
|
|
source_url = "https://www.bilibili.com/video/BV1xx411c7mD"
|
|
markdown = "## 标题\n\n内容"
|
|
|
|
result = note_helper.prepend_source_link(markdown, source_url)
|
|
|
|
self.assertTrue(result.startswith(f"> 来源链接：{source_url}\n\n"))
|
|
self.assertIn("## 标题", result)
|
|
|
|
def test_prepend_source_link_does_not_duplicate_when_header_exists(self):
|
|
source_url = "https://www.youtube.com/watch?v=abc123"
|
|
markdown = f"> 来源链接：{source_url}\n\n## 标题\n\n内容"
|
|
|
|
result = note_helper.prepend_source_link(markdown, source_url)
|
|
|
|
self.assertEqual(result, markdown)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|