Added imports in __init__

Author: DianaStrauss

src/hackingBuddyGPT/usecases/web_api_testing/__init__.py

@@ -1,2 +1,6 @@
 from .simple_openapi_documentation import SimpleWebAPIDocumentation
 from .simple_web_api_testing import SimpleWebAPITesting
+from . import response_processing
+from . import documentation
+from . import prompt_generation
+from . import testing

src/hackingBuddyGPT/usecases/web_api_testing/simple_web_api_testing.py

@@ -26,7 +26,7 @@
 from hackingBuddyGPT.usecases.web_api_testing.response_processing.response_analyzer_with_llm import \
     ResponseAnalyzerWithLLM
 from hackingBuddyGPT.usecases.web_api_testing.response_processing.response_handler import ResponseHandler
-from hackingBuddyGPT.usecases.web_api_testing.testing.test_handler import TestHandler
+from hackingBuddyGPT.usecases.web_api_testing.testing.test_handler import GenerationTestHandler
 from hackingBuddyGPT.usecases.web_api_testing.utils.configuration_handler import ConfigurationHandler
 from hackingBuddyGPT.usecases.web_api_testing.utils.custom_datatypes import Context, Prompt
 from hackingBuddyGPT.usecases.web_api_testing.utils.llm_handler import LLMHandler
@@ -147,7 +147,7 @@ def _setup_handlers(self):
                                                          prompt_helper=self.prompt_helper)
         self._response_handler.set_response_analyzer(self.response_analyzer)
         self._report_handler = ReportHandler(self.config)
-        self._test_handler = TestHandler(self._llm_handler)
+        self._test_handler = GenerationTestHandler(self._llm_handler)
 
     def _setup_initial_prompt(self) -> None:
         """

src/hackingBuddyGPT/usecases/web_api_testing/testing/__init__.py

@@ -1 +1 @@
-from .test_handler import TestHandler
+from .test_handler import GenerationTestHandler

src/hackingBuddyGPT/usecases/web_api_testing/testing/test_handler.py

@@ -5,7 +5,7 @@
 from typing import Any, Dict, Tuple
 
 
-class TestHandler:
+class GenerationTestHandler:
     """
     A class responsible for parsing, generating, and saving structured API test cases,
     including generating pytest-compatible test functions using an LLM.