Add a pytest fixture for pook (#134)

* Add a pytest fixture for pook

* use context-manager

History.rst

@@ -33,6 +33,8 @@ Draft v2.0.0 / Unreleased
 
     * These classes never implemented regex matching
 
+  * Add a pytest fixture to the package.
+
 v1.4.3 / 2024-02-23
 -------------------
 

examples/pytest_example.py

@@ -36,3 +36,9 @@ def test_no_match_exception():
     pook.get("server.com/bar", reply=204)
     with pytest.raises(Exception):
         requests.get("http://server.com/baz")
+
+
+def test_fixture(pook):
+    pook.get("https://example.org/").reply(204)
+    res = requests.get("https://example.org/")
+    assert res.status_code == 204

pyproject.toml

@@ -38,6 +38,9 @@ requires-python = ">=3.8"
 [project.urls]
 Homepage = "https://github.com/h2non/pook"
 
+[project.entry-points.pytest11]
+pook = "pook.pytest_fixture"
+
 [tool.hatch.version]
 path = "src/pook/__init__.py"
 

src/pook/pytest_fixture.py

@@ -0,0 +1,10 @@
+import pytest
+
+import pook as pook_mod
+
+
+@pytest.fixture
+def pook():
+    """Pytest fixture for HTTP traffic mocking and testing"""
+    with pook_mod.use():
+        yield pook_mod