ci: Add unit tests for parse_videos and FileUpdater (#47)

Author: DenverCoder1

Diff for: action.py

@@ -258,9 +258,9 @@ def update(readme_path: str, comment_tag: str, replace_content: str):
         title_color=args.title_color,
         stats_color=args.stats_color,
         youtube_api_key=args.youtube_api_key,
-        show_duration=args.show_duration == "true",
         theme_context_light=json.loads(args.theme_context_light),
         theme_context_dark=json.loads(args.theme_context_dark),
+        show_duration=args.show_duration == "true",
     )
 
     video_content = video_parser.parse_videos()

Diff for: tests/test_action.py

@@ -1,4 +1,21 @@
-from action import VideoParser
+import os
+import re
+
+from action import FileUpdater, VideoParser
+
+video_parser = VideoParser(
+    base_url="https://ytcards.demolab.com/",
+    channel_id="UCipSxT7a3rn81vGLw9lqRkg",
+    max_videos=6,
+    card_width=250,
+    background_color="#0d1117",
+    title_color="#ffffff",
+    stats_color="#dedede",
+    youtube_api_key="",
+    theme_context_light={},
+    theme_context_dark={},
+    show_duration=False,
+)
 
 
 def test_parse_iso8601_duration():
@@ -8,11 +25,73 @@ def test_parse_iso8601_duration():
     assert VideoParser.parse_iso8601_duration("P1DT2H10M10S") == 94210
 
 
-def test_parse_video():
-    """TODO: Test parsing a video from feedparser video object"""
-    pass
+def test_parse_videos():
+    videos = video_parser.parse_videos()
+
+    assert len(videos.splitlines()) == 6
 
+    line_regex = r"^\[!\[.*\]\(.* \"(.*)\"\)\]\(.*\)$"
+    assert all(re.match(line_regex, line) for line in videos.splitlines())
 
-def test_parse_videos():
-    """TODO: Test parsing a list of videos with feedparser"""
-    pass
+    assert "https://ytcards.demolab.com/?id=" in videos
+    assert "title=" in videos
+    assert "timestamp=" in videos
+    assert "background_color=" in videos
+    assert "title_color=" in videos
+    assert "stats_color=" in videos
+    assert "width=" in videos
+
+
+def test_parse_videos_with_theme_context():
+    video_parser._theme_context_light = {
+        "background_color": "#ffffff",
+        "title_color": "#000000",
+        "stats_color": "#000000",
+    }
+    video_parser._theme_context_dark = {
+        "background_color": "#000000",
+        "title_color": "#ffffff",
+        "stats_color": "#ffffff",
+    }
+    videos = video_parser.parse_videos()
+
+    assert len(videos.splitlines()) == 6
+
+    line_regex = (
+        r"^\[!\[.*\]\(.* \"(.*)\"\)\]\(.*\#gh-dark-mode-only\)"
+        r"\[!\[.*\]\(.* \"(.*)\"\)\]\(.*\#gh-light-mode-only\)$"
+    )
+    assert all(re.match(line_regex, line) for line in videos.splitlines())
+
+
+def test_update_file():
+    path = "./tests/README.md"
+    # create a file to test with
+    with open(path, "w+") as f:
+        f.write(
+            "Test Before\n"
+            "\n"
+            "<!-- BEGIN YOUTUBE-CARDS -->\n"
+            "<!-- END YOUTUBE-CARDS -->\n"
+            "\n"
+            "Test After\n"
+        )
+    try:
+        # update the file
+        FileUpdater.update(path, "YOUTUBE-CARDS", "A\nB\nC")
+        # read the file and assert the contents
+        with open(path, "r") as f:
+            assert f.read() == (
+                "Test Before\n"
+                "\n"
+                "<!-- BEGIN YOUTUBE-CARDS -->\n"
+                "A\n"
+                "B\n"
+                "C\n"
+                "<!-- END YOUTUBE-CARDS -->\n"
+                "\n"
+                "Test After\n"
+            )
+    finally:
+        # remove the file
+        os.remove(path)