| import importlib.util |
| import pathlib |
| import unittest |
|
|
|
|
| ROOT = pathlib.Path(__file__).resolve().parents[1] |
| MODULE_PATH = ROOT / "app" / "utils" / "screenshot_marker.py" |
| spec = importlib.util.spec_from_file_location("screenshot_marker", MODULE_PATH) |
| if spec is None or spec.loader is None: |
| raise ImportError("screenshot_marker module spec not found") |
| screenshot_marker = importlib.util.module_from_spec(spec) |
| spec.loader.exec_module(screenshot_marker) |
| extract_screenshot_timestamps = screenshot_marker.extract_screenshot_timestamps |
| extract_content_timestamps = screenshot_marker.extract_content_timestamps |
| ensure_screenshot_markers = screenshot_marker.ensure_screenshot_markers |
|
|
|
|
| class TestScreenshotMarker(unittest.TestCase): |
| def test_extract_accepts_star_bracket_format(self): |
| markdown = "A\n*Screenshot-[01:02]\nB" |
| matches = extract_screenshot_timestamps(markdown) |
| self.assertEqual(matches, [("*Screenshot-[01:02]", 62)]) |
|
|
| def test_extract_accepts_legacy_formats(self): |
| markdown = "*Screenshot-03:04 and Screenshot-[05:06]" |
| matches = extract_screenshot_timestamps(markdown) |
| self.assertEqual( |
| matches, |
| [ |
| ("*Screenshot-03:04", 184), |
| ("Screenshot-[05:06]", 306), |
| ], |
| ) |
|
|
| def test_extract_content_timestamps_for_fallback(self): |
| markdown = "## A *Content-[00:12]\n## B *Content-[01:03]\n## C *Content-[01:03]" |
| matches = extract_content_timestamps(markdown) |
| self.assertEqual(matches, [12, 63]) |
|
|
| def test_ensure_screenshot_markers_adds_duration_fallback(self): |
| markdown = "## A\ncontent" |
| with_markers = ensure_screenshot_markers(markdown, 120) |
| self.assertIn("*Screenshot-[00:30]", with_markers) |
| self.assertIn("*Screenshot-[01:00]", with_markers) |
| self.assertIn("*Screenshot-[01:30]", with_markers) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
