Feature: htmlfile fixture

pylover · pylover · commit aedb3d453423 · 2024-08-26T03:17:00.000+03:30
diff --git a/yhttp/dev/__init__.py b/yhttp/dev/__init__.py
@@ -1 +1 @@
-__version__ = '3.1.3'
+__version__ = '3.2.0'
diff --git a/yhttp/dev/fixtures.py b/yhttp/dev/fixtures.py
@@ -3,6 +3,7 @@
 import socket
 import tempfile
 import datetime
+import contextlib
 from unittest.mock import patch
 
 import pytest
@@ -135,3 +136,28 @@ def close(self):
 
     with patch('redis.Redis', new=RedisMock) as p:
         yield p
+
+
+@pytest.fixture(scope='session')
+def htmlfile():
+    @contextlib.contextmanager
+    def create(filename, title, cssfile=None):
+        with open(filename, 'a') as file:
+            file.truncate(0)
+            file.write(
+                '<!DOCTYPE html>\n'
+                '<html lang="en">\n'
+                '<head>\n'
+                '<meta charset="utf-8" />\n'
+                f'<title>{title}</title>\n'
+            )
+
+            if cssfile:
+                file.write(f'<link rel="stylesheet" href="{cssfile}" '
+                           'type="text/css" />\n')
+
+            file.write('</head><body>\n')
+            yield file
+            file.write('</body></html>')
+
+    return create
