handle custom __new__ arguments in Minio mock fixture and few adjustments

pytest_minio_mock/plugin.py

@@ -793,6 +793,10 @@ def __init__(
             credentials (optional): The credentials object. Defaults to None.
             cert_check (optional): Whether to check certificates. Defaults to True.
         """
+        if "://" not in endpoint:
+            scheme = "https" if secure else "http"
+            endpoint = f"{scheme}://{endpoint}"
+
         self._base_url = endpoint
         self._access_key = access_key
         self._secret_key = secret_key
@@ -816,7 +820,7 @@ def _health_check(self):
         if not self._base_url:
             raise ValueError("base_url is empty")
         if not validators.hostname(self._base_url) and not validators.url(
-            self._base_url
+            self._base_url, simple_host=True
         ):
             raise ValueError(f"base_url {self._base_url} is not valid")
 
@@ -1453,5 +1457,9 @@ def minio_mock_init(
         client.connect(minio_mock_servers)
         return client
 
-    patched = mocker.patch.object(Minio, "__new__", new=minio_mock_init)
-    yield patched
+    patched = mocker.patch.object(Minio, "__new__", new=minio_mock_init, create=True)
+    try:
+        yield patched
+        mocker.stop(patched)
+    finally:
+        Minio.__new__ = lambda cls, *args, **kw: object.__new__(cls)

tests/test_minio_mock_client.py

@@ -7,9 +7,28 @@
 
 @pytest.mark.UNIT
 class TestsMockMinioClient:
+    @pytest.mark.parametrize("is_secure", [True, False])
+    @pytest.mark.UNIT
+    def test_mock_minio_client_init_without_endpoint_schema_and_with_minimal_parameters(
+        self, is_secure: bool
+    ):
+        endpoint = "localhost:9000"
+        client = MockMinioClient(endpoint, secure=is_secure)
+
+        base_url_schema = "https" if is_secure else "http"
+        assert client._base_url == f"{base_url_schema}://{endpoint}"
+        assert client._access_key == None
+        assert client._secret_key == None
+        assert client._session_token == None
+        assert client._secure is is_secure
+        assert client._region == None
+        assert client._http_client == None
+        assert client._credentials == None
+        assert client._cert_check is True
+
     @pytest.mark.UNIT
     def test_mock_minio_client_init_with_minimal_parameters(self):
-        endpoint = "http://localhost:9000"
+        endpoint = "https://localhost:9000"
         client = MockMinioClient(endpoint)
         assert client._base_url == endpoint
         assert client._access_key == None
@@ -67,3 +86,17 @@ def test_mock_minio_client_init_error_handling(self):
             TypeError, match="missing 1 required positional argument: 'endpoint'"
         ):
             client = MockMinioClient()  # not passing endpoint should raise an error
+
+    @pytest.mark.parametrize(
+        "endpoint",
+        [
+            "http://localhost:9000",
+            "https://localhost:9000",
+            "localhost:9000",
+            "any-endpoint.local",
+        ],
+    )
+    @pytest.mark.UNIT
+    def test_mock_minio_client_health_check(self, endpoint: str):
+        client = MockMinioClient(endpoint)
+        assert client._health_check() is None