48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
import sys, unittest
|
|
from pathlib import Path
|
|
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
|
|
from okf_common import (parse_frontmatter, parse_yaml_subset,
|
|
extract_links, concept_id, FrontmatterError)
|
|
|
|
|
|
class TestFrontmatter(unittest.TestCase):
|
|
def test_parse_basic(self):
|
|
text = "---\ntype: BigQuery Table\ntitle: Orders\ntags: [sales, revenue]\n---\n\n# Body\n"
|
|
meta, body = parse_frontmatter(text)
|
|
self.assertEqual(meta["type"], "BigQuery Table")
|
|
self.assertEqual(meta["title"], "Orders")
|
|
self.assertEqual(meta["tags"], ["sales", "revenue"])
|
|
self.assertIn("# Body", body)
|
|
|
|
def test_url_value_with_colons(self):
|
|
meta, _ = parse_frontmatter("---\ntype: X\nresource: https://e.com/a?b=c\n---\nbody")
|
|
self.assertEqual(meta["resource"], "https://e.com/a?b=c")
|
|
|
|
def test_quoted_value(self):
|
|
self.assertEqual(parse_yaml_subset('okf_version: "0.1"')["okf_version"], "0.1")
|
|
|
|
def test_empty_list(self):
|
|
self.assertEqual(parse_yaml_subset("tags: []")["tags"], [])
|
|
|
|
def test_no_frontmatter(self):
|
|
meta, body = parse_frontmatter("# Just markdown\n")
|
|
self.assertIsNone(meta)
|
|
self.assertEqual(body, "# Just markdown\n")
|
|
|
|
def test_unclosed_raises(self):
|
|
with self.assertRaises(FrontmatterError):
|
|
parse_frontmatter("---\ntype: X\n\nbody without close")
|
|
|
|
|
|
class TestLinksAndIds(unittest.TestCase):
|
|
def test_extract_links(self):
|
|
body = "See [a](/tables/a.md) and [b](./b.md) and [ext](https://x.com)."
|
|
self.assertEqual(extract_links(body), ["/tables/a.md", "./b.md", "https://x.com"])
|
|
|
|
def test_concept_id(self):
|
|
self.assertEqual(concept_id("/bundle", "/bundle/tables/users.md"), "tables/users")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|