Feature/refactor tests plugins

.cz.toml

@@ -0,0 +1,9 @@
+[tool]
+[tool.commitizen]
+name = "cz_conventional_commits"
+version = "22.0.0"
+tag_format = "$version"
+version_files = [
+    'pyproject.toml:version',
+    'aiohttpretty/__init__.py:__version__',
+]

.gitattributes

@@ -0,0 +1,2 @@
+
+*       text=auto

Makefile

@@ -0,0 +1,8 @@
+.PHONY: format
+format:
+	black --line-length 79 aiohttpretty tests
+	isort --profile black --line-length 79 aiohttpretty tests
+
+.PHONY: test
+test:
+	pytest tests

README.md

@@ -94,32 +94,31 @@ Purge the registry and the list of intercepted calls.
 
 Checks to see if the given uri was called during the test.  By default, will verify that the query params match up.  Setting `check_params` to `False` will strip params from the *registered* uri, not the passed-in uri.
 
+### `.open(clear: bool = True)`
+Runs `.clear()` if `clear`, `.activate()` on enter and `.deactivate()` on exit. 
+
+### `.async_decorate(func)`
+Decorator to run `with aiohttpretty.open()` before running coroutine function
+
+### `.decorate(func)`
+Same as `.async_decorate()` but for normal functions
 
 ## Other
 
 ### pytest marker
 
-To simplify usage of `aiohttpretty` in tests, you can make a pytest marker that will automatically activate/deactivate `aiohttpretty` for the scope of a test.  To do so, add the following to your `conftest.py`:
+To simplify usage of `aiohttpretty` in tests, it already ships with a pytest plugin. To use, add `@pytest.mark.aiohttpretty` to your test:
 
-```
-import aiohttpretty
+```python
+import pytest
 
-def pytest_configure(config):
-    config.addinivalue_line(
-        'markers',
-        'aiohttpretty: mark tests to activate aiohttpretty'
-    )
-
-def pytest_runtest_setup(item):
-    marker = item.get_marker('aiohttpretty')
-    if marker is not None:
-        aiohttpretty.clear()
-        aiohttpretty.activate()
-
-def pytest_runtest_teardown(item, nextitem):
-    marker = item.get_marker('aiohttpretty')
-    if marker is not None:
-        aiohttpretty.deactivate()
-```
+@pytest.mark.aiohttpretty
+async def my_test():
+    # inside test function aiohttpretty is clean and active
+    assert thing() == other
 
-Then add `@pytest.mark.aiohttpretty` before `async def test_foo`.
+@pytes.mark.aiohttpretty
+def my_other_test(test_client):
+    # aiohttpretty also supports sync functions in case of testing web applications
+    assert test_client.get('/my-route') == other
+```

aiohttpretty/__init__.py

@@ -0,0 +1,4 @@
+from .main import aiohttpretty
+
+__all__ = ['aiohttpretty']
+__version__ = '22.0.0'

aiohttpretty/exc.py

@@ -0,0 +1,21 @@
+class AioHttPrettyError(Exception):
+    """Base class for all AioHttPrettyErrors"""
+
+
+class NoUrlMatching(AioHttPrettyError, KeyError):
+    """No url matches received url with given params and method"""
+
+    def __str__(self) -> str:
+        return Exception.__str__(self)
+
+
+class ExhaustedAllResponses(AioHttPrettyError, IndexError):
+    """No response left for given url"""
+
+
+class InvalidBody(AioHttPrettyError, TypeError):
+    """Received invalid body type"""
+
+
+class InvalidResponses(AioHttPrettyError, ValueError):
+    """Cannot specify params in responses"""

aiohttpretty/helpers.py

@@ -0,0 +1,49 @@
+import asyncio
+import typing
+
+from aiohttpretty import types
+
+DEFAULT_ENCODING = 'utf-8'
+
+
+def encode_string(string: str, encoding: str = DEFAULT_ENCODING):
+    return string.encode(encoding)
+
+
+def is_stream_like(value: typing.Any):
+    return hasattr(value, 'read') and asyncio.iscoroutinefunction(value.read)
+
+
+class StreamLike(typing.Protocol):
+    async def read(self, n: int) -> bytes:
+        ...
+
+    @property
+    def size(self) -> int:
+        ...
+
+
+def wrap_content_stream(content: typing.Union[str, bytes, StreamLike]):
+    if isinstance(content, str):
+        content = encode_string(content)
+
+    if isinstance(content, bytes):
+        return types.MockStream(content)
+
+    return content
+
+
+def build_raw_headers(headers: typing.Mapping[str, str]):
+    """Convert a dict of headers to a tuple of tuples. Mimics the format of ClientResponse.
+    """
+    return tuple(
+        (encode_string(key), encode_string(value))
+        for key, value in headers.items()
+    )
+
+
+def compare_mapping(
+    first: typing.Mapping[str, typing.Any],
+    second: typing.Mapping[str, typing.Any],
+):
+    return all(second.get(key) == value for key, value in first.items())