From 5a549792eaf7347303af4a6079a75ff0f2ec1255 Mon Sep 17 00:00:00 2001
From: 7000pctAUTO <lukems3823@gmail.com>
Date: Sun, 22 Mar 2026 21:59:14 +0000
Subject: [PATCH] Add mockapi source files and tests

---
 src/mockapi/core/server_generator.py | 67 ++++------------------------
 1 file changed, 9 insertions(+), 58 deletions(-)

diff --git a/src/mockapi/core/server_generator.py b/src/mockapi/core/server_generator.py
index aabd744..bbb67b2 100644
--- a/src/mockapi/core/server_generator.py
+++ b/src/mockapi/core/server_generator.py
@@ -16,23 +16,12 @@ class MockOperationResolver(Resolver):
     """Custom operation resolver for mock API."""
 
     def __init__(self, mock_server_generator):
-        """Initialize the resolver.
-
-        Args:
-            mock_server_generator: The MockServerGenerator instance
-        """
+        """Initialize the resolver."""
         super().__init__()
         self.mock_server_generator = mock_server_generator
 
     def resolve(self, operation):
-        """Resolve an operation to a mock function.
-
-        Args:
-            operation: The operation object from connexion
-
-        Returns:
-            Resolution object with the mock function
-        """
+        """Resolve an operation to a mock function."""
         operation_id = self.resolve_operation_id(operation)
         for path, path_item in self.mock_server_generator.spec.get("paths", {}).items():
             for method, op_def in path_item.items():
@@ -47,12 +36,7 @@ class MockServerGenerator:
     """Generates a mock server from an OpenAPI specification."""
 
     def __init__(self, spec: Dict[str, Any], config: Optional[Config] = None):
-        """Initialize the mock server generator.
-
-        Args:
-            spec: The OpenAPI specification dictionary
-            config: Configuration object
-        """
+        """Initialize the mock server generator."""
         self.spec = spec
         self.config = config or Config()
         schemas = spec.get("components", {}).get("schemas", {})
@@ -60,11 +44,7 @@ class MockServerGenerator:
         self.app: Optional[App] = None
 
     def generate(self) -> App:
-        """Generate the connexion application.
-
-        Returns:
-            Configured connexion App instance
-        """
+        """Generate the connexion application."""
         self.app = connexion.App(__name__, specification_dir=".")
 
         self.app.add_api(
@@ -84,14 +64,7 @@ class MockServerGenerator:
         return self.app
 
     def _create_mock_function(self, operation: Dict[str, Any]):
-        """Create a mock function for an operation.
-
-        Args:
-            operation: The OpenAPI operation definition
-
-        Returns:
-            Function that generates mock responses
-        """
+        """Create a mock function for an operation."""
 
         def mock_function(*args, **kwargs):
             mock_config = operation.get("x-mock-config", {})
@@ -113,26 +86,12 @@ class MockServerGenerator:
         return mock_function
 
     def _should_return_error(self, mock_config: Dict[str, Any]) -> bool:
-        """Determine if we should return an error response.
-
-        Args:
-            mock_config: x-mock-config extension data
-
-        Returns:
-            True if error should be returned
-        """
+        """Determine if we should return an error response."""
         error_probability = mock_config.get("errorProbability", 0)
         return random.random() < error_probability
 
     def _generate_error_response(self, mock_config: Dict[str, Any]) -> tuple:
-        """Generate an error response.
-
-        Args:
-            mock_config: x-mock-config extension data
-
-        Returns:
-            Tuple of (response_body, status_code)
-        """
+        """Generate an error response."""
         status_code = mock_config.get("errorCode", 500)
         error_message = mock_config.get("errorMessage", "Mock error")
         return {"error": error_message}, status_code
@@ -142,14 +101,6 @@ def create_mock_server(
     spec: Dict[str, Any],
     config: Optional[Config] = None,
 ) -> App:
-    """Create a mock server from an OpenAPI spec.
-
-    Args:
-        spec: OpenAPI specification dictionary
-        config: Configuration object
-
-    Returns:
-        Configured connexion App
-    """
+    """Create a mock server from an OpenAPI spec."""
     generator = MockServerGenerator(spec, config)
-    return generator.generate()
\ No newline at end of file
+    return generator.generate()