add yaml_parser tests, small refactor

3 files changed, 53 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
index 58416ea..1beffc6 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,3 +1,4 @@
+import os
 import sys
 from typing import Callable
 import pytest
@@ -7,6 +8,11 @@ from pyssg.arg_parser import get_parser
 from pyssg.custom_logger import setup_logger
 
 
+@pytest.fixture
+def test_dir():
+    return os.path.dirname(os.path.abspath(__file__))
+
+
 @pytest.fixture(scope='session')
 def arg_parser():
     return get_parser()
diff --git a/tests/io_files/simple.yaml b/tests/io_files/simple.yaml
new file mode 100644
index 0000000..0b722a6
--- /dev/null
+++ b/tests/io_files/simple.yaml
@@ -0,0 +1,28 @@
+%YAML 1.2
+---
+define: &root "$HOME/pyssg/site_example/"
+
+title: "Example site"
+path:
+  src: !join [*root, "src"]
+  dst: !join [*root, "dst"]
+  plt: !join [*root, "plt"]
+  db: !join [*root, ".files"]
+url:
+  main: "https://example.com"
+  static: "https://static.example.com"
+  default_image: "images/default.png"
+fmt:
+  date: "%a, %b %d, %Y @ %H:%M %Z"
+  list_date: "%b %d"
+  list_sep_date: "%B %Y"
+dirs:
+  /:
+    cfg:
+      plt: "page.html"
+      tags: False
+      index: False
+      rss: False
+      sitemap: False
+      exclude_dirs: []
+...
\ No newline at end of file
diff --git a/tests/test_yaml_parser.py b/tests/test_yaml_parser.py
new file mode 100644
index 0000000..9a1b3c3
--- /dev/null
+++ b/tests/test_yaml_parser.py
@@ -0,0 +1,19 @@
+from importlib.resources import path as rpath
+from pyssg.yaml_parser import get_parsed_yaml
+
+
+def test_yaml_resource_read() -> None:
+    yaml: list[dict] = get_parsed_yaml('simple.yaml', 'tests.io_files')
+    assert len(yaml) == 1
+
+
+def test_yaml_path_read(test_dir: str) -> None:
+    yaml: list[dict] = get_parsed_yaml(f'{test_dir}/io_files/simple.yaml')
+    assert len(yaml) == 1
+
+
+def test_yaml_join() -> None:
+    yaml: dict = get_parsed_yaml('simple.yaml', 'tests.io_files')[0]
+    define_str: str = '$HOME/pyssg/site_example/'
+    assert yaml['define'] == define_str
+    assert yaml['path']['src'] == f'{define_str}src'