Merge pull request #114 from ipa-lab/merge_web_api_testing_development

Merge web api testing development

Author: andreashappe

.gitignore

@@ -25,3 +25,10 @@ scripts/mac_ansible_hosts.ini
 scripts/mac_ansible_id_rsa
 scripts/mac_ansible_id_rsa.pub
 .aider*
+
+src/hackingBuddyGPT/usecases/web_api_testing/documentation/openapi_spec/
+src/hackingBuddyGPT/usecases/web_api_testing/documentation/reports/
+src/hackingBuddyGPT/usecases/web_api_testing/retrieve_spotify_token.py
+config/my_configs/*
+config/configs/*
+config/configs/

pyproject.toml

@@ -45,11 +45,15 @@ dependencies = [
     'uvicorn[standard] == 0.30.6',
     'dataclasses_json == 0.6.7',
     'websockets == 13.1',
-    'langchain-community',
-    'langchain-openai',
+    'pandas',
+    'faker',
+    'fpdf',
+    'langchain_core',
+    'langchain_community',
+    'langchain_chroma',
+    'langchain_openai',
     'markdown',
     'chromadb',
-    'langchain-chroma',
 ]
 
 [project.urls]
@@ -69,7 +73,7 @@ where = ["src"]
 pythonpath = "src"
 addopts = ["--import-mode=importlib"]
 [project.optional-dependencies]
-testing = ['pytest', 'pytest-mock']
+testing = ['pytest', 'pytest-mock', 'pandas', 'faker',     'langchain_core']
 dev = [
 	'ruff',
 ]

src/hackingBuddyGPT/capabilities/http_request.py

@@ -45,18 +45,11 @@ def __call__(
         body_is_base64: Optional[bool] = False,
         headers: Optional[Dict[str, str]] = None,
     ) -> str:
+
         if body is not None and body_is_base64:
             body = base64.b64decode(body).decode()
-        if self.host[-1] != "/":
+        if self.host[-1] != "/" and not path.startswith("/"):
             path = "/" + path
-        resp = self._client.request(
-            method,
-            self.host + path,
-            params=query,
-            data=body,
-            headers=headers,
-            allow_redirects=self.follow_redirects,
-        )
         try:
             resp = self._client.request(
                 method,

src/hackingBuddyGPT/capabilities/parsed_information.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass, field
+from typing import Dict, Any, List, Tuple
+from hackingBuddyGPT.capabilities import Capability
+
+
+from dataclasses import dataclass, field
+from typing import Any, Dict, List, Tuple
+
+@dataclass
+class ParsedInformation(Capability):
+    status_code: str
+    reason_phrase:  Dict[str, Any] = field(default_factory=dict)
+    headers: Dict[str, Any] = field(default_factory=dict)
+    response_body: Dict[str, Any] = field(default_factory=dict)
+    registry: List[Tuple[str, str, str, str]] = field(default_factory=list)
+
+    def describe(self) -> str:
+        """
+        Returns a description of the test case.
+        """
+        return f"Parsed information for {self.status_code}, reason_phrase: {self.reason_phrase}, headers: {self.headers}, response_body: {self.response_body} "
+    def __call__(self, status_code: str, reason_phrase: str, headers: str, response_body:str) -> dict:
+        self.registry.append((status_code, response_body, headers,response_body))
+
+        return {"status_code": status_code, "reason_phrase": reason_phrase, "headers": headers, "response_body": response_body}

src/hackingBuddyGPT/capabilities/python_test_case.py

@@ -0,0 +1,22 @@
+
+from hackingBuddyGPT.capabilities import Capability
+
+
+from dataclasses import dataclass, field
+from typing import Any, Dict, List, Tuple
+
+@dataclass
+class PythonTestCase(Capability):
+    description: str
+    input: Dict[str, Any] = field(default_factory=dict)
+    expected_output: Dict[str, Any] = field(default_factory=dict)
+    registry: List[Tuple[str, dict, dict]] = field(default_factory=list)
+
+    def describe(self) -> str:
+        """
+        Returns a description of the test case.
+        """
+        return f"Test Case: {self.description}\nInput: {self.input}\nExpected Output: {self.expected_output}"
+    def __call__(self, description: str, input: dict, expected_output: dict) -> dict:
+        self.registry.append((description, input, expected_output))
+        return {"description": description, "input": input, "expected_output": expected_output}

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