aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--wqflask/tests/unit/wqflask/test_markdown_routes.py18
1 files changed, 8 insertions, 10 deletions
diff --git a/wqflask/tests/unit/wqflask/test_markdown_routes.py b/wqflask/tests/unit/wqflask/test_markdown_routes.py
index 2d403d04..90e0f17c 100644
--- a/wqflask/tests/unit/wqflask/test_markdown_routes.py
+++ b/wqflask/tests/unit/wqflask/test_markdown_routes.py
@@ -3,28 +3,26 @@
import unittest
from unittest import mock
+from dataclasses import dataclass
from wqflask.markdown_routes import render_markdown
+@dataclass
class MockRequests404:
- @property
- def status_code(self):
- return 404
+ status_code: int = 404
-class MockRequests200:
- @property
- def status_code(self):
- return 200
- @property
- def content(self):
- return b"""
+@dataclass
+class MockRequests200:
+ status_code: int = 200
+ content: str = b"""
# Glossary
This is some content
## Sub-heading
This is another sub-heading"""
+
class TestMarkdownRoutesFunctions(unittest.TestCase):
"""Test cases for functions in markdown_routes"""