lsrestclient 1.1.0__tar.gz → 1.2.0__tar.gz

{lsrestclient-1.1.0 → lsrestclient-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lsrestclient
-Version: 1.1.0
+Version: 1.2.0
 Summary: REST Api Client
 Author: mba
 Author-email: bartel@electronic-shop.lu

{lsrestclient-1.1.0 → lsrestclient-1.2.0}/lsrestclient/client.py

@@ -71,8 +71,8 @@ class LsRestClient(Session):
         self._clients[name] = self
 
     @staticmethod
-    def mock_name(method: str, url: str) -> str:
-        return f"{method.upper()}_{url}"
+    def mock_name(client_name: str, method: str, url: str) -> str:
+        return f"{client_name}_{method.upper()}_{url}"
 
     def full_url(
         self,
@@ -99,7 +99,7 @@ class LsRestClient(Session):
 
     def caller(self, method: str, url: str, *args, **kwargs) -> LsRestClientResponse:
         # check mocks
-        mock = self._mocks.get(self.mock_name(method, url), None)
+        mock = self._mocks.get(self.mock_name(self.name, method, url), None)
         func = mock if mock is not None else self.request
         return func(method, url, *args, **kwargs)
 
@@ -251,27 +251,9 @@ class LsRestClient(Session):
         """
         return self.request("HEAD", *args, **kwargs)
 
-    def mock(self, method: str, url: str, *args, **kwargs):
-        mock_name = self.mock_name(method, url)
-        mock = MagicMock(*args, **kwargs)
-        # mock = MagicMock(self, mock_name, *args, **kwargs)
+    def mock(self, mock_name: str, mock: MagicMock):
         self._mocks[mock_name] = mock
-        return LsRestClientMock(self, mock_name, mock)
 
     def unmock(self, mock_name: str):
         if mock_name in self._mocks:
             del self._mocks[mock_name]
-
-
-class LsRestClientMock(object):
-    def __init__(self, client, mock_name, mock):
-        self._client = client
-        self._mock_name = mock_name
-        self.mock = mock
-
-    def unmock(self):
-        self._client.unmock(self._mock_name)
-
-    @property
-    def call_count(self):
-        return self.mock.call_count

lsrestclient-1.2.0/lsrestclient/fixtures.py

@@ -0,0 +1,12 @@
+from lsrestclient.mock import lsrestclient_mock_context
+
+try:
+    import pytest
+
+    @pytest.fixture
+    def lsrestclient_mocker():
+        with lsrestclient_mock_context() as mocker:
+            yield mocker
+
+except ImportError:
+    pass

lsrestclient-1.2.0/lsrestclient/mock.py

@@ -0,0 +1,36 @@
+import dataclasses
+from contextlib import contextmanager
+from unittest.mock import MagicMock
+
+from lsrestclient import LsRestClient
+
+
+@dataclasses.dataclass
+class LsRestClientMockModel:
+    mock_name: str
+    client: LsRestClient
+    mock: MagicMock
+
+
+class LsRestClientMocker(object):
+    def __init__(self):
+        super().__init__()
+        self.mocks = {}
+
+    def mock(self, client: LsRestClient, method: str, url: str, *args, **kwargs):
+        mock_name = LsRestClient.mock_name(client.name, method, url)
+        mock = MagicMock(*args, **kwargs)
+        client.mock(mock_name, mock)
+        self.mocks[mock_name] = LsRestClientMockModel(client=client, mock_name=mock_name, mock=mock)
+        return mock
+
+    def unmock_all(self):
+        for mock_name, mock_model in self.mocks.items():
+            mock_model.client.unmock(mock_name)
+
+
+@contextmanager
+def lsrestclient_mock_context():
+    mocker = LsRestClientMocker()
+    yield mocker
+    mocker.unmock_all()

{lsrestclient-1.1.0 → lsrestclient-1.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lsrestclient"
-version = "1.1.0"
+version = "1.2.0"
 description = "REST Api Client"
 authors = ["mba <bartel@electronic-shop.lu>"]
 readme = "README.md"
@@ -32,3 +32,6 @@ log_cli_format = "[%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
 
 [tool.black]
 line-length = 120
+
+[tool.poetry.plugins.pytest11]
+"lsrestclient" = "lsrestclient"