feat(re org tests):

2026-07-06 03:37:49 +02:00 · 2025-03-08 12:38:54 +02:00
parent a934ee243e
commit 7fe93e8a14
3 changed files with 0 additions and 0 deletions
@@ -0,0 +1,118 @@
+import pytest
+
+from agentic_security.http_spec import LLMSpec, parse_http_spec
+
+
+class TestParseHttpSpec:
+    # Should correctly parse a simple HTTP spec with headers and body
+    def test_parse_simple_http_spec(self):
+        http_spec = (
+            'GET http://example.com\nContent-Type: application/json\n\n{"key": "value"}'
+        )
+        expected_spec = LLMSpec(
+            method="GET",
+            url="http://example.com",
+            headers={"Content-Type": "application/json"},
+            body='{"key": "value"}',
+        )
+        assert parse_http_spec(http_spec) == expected_spec
+
+    # Should correctly parse a HTTP spec with headers containing special characters
+    def test_parse_http_spec_with_special_characters(self):
+        http_spec = 'POST http://example.com\nX-Auth-Token: abcdefg1234567890!@#$%^&*\n\n{"key": "value"}'
+        expected_spec = LLMSpec(
+            method="POST",
+            url="http://example.com",
+            headers={"X-Auth-Token": "abcdefg1234567890!@#$%^&*"},
+            body='{"key": "value"}',
+        )
+        assert parse_http_spec(http_spec) == expected_spec
+
+    # Should correctly parse a spec with no headers and no body
+    def test_parse_http_spec_with_no_headers_and_no_body(self):
+        # Arrange
+        http_spec = "GET http://example.com"
+
+        # Act
+        result = parse_http_spec(http_spec)
+
+        # Assert
+        assert result.method == "GET"
+        assert result.url == "http://example.com"
+        assert result.headers == {}
+        assert result.body == ""
+
+    def test_parse_http_spec_with_headers_no_body(self):
+        # Arrange
+        http_spec = "GET http://example.com\nContent-Type: application/json\n\n"
+
+        # Act
+        result = parse_http_spec(http_spec)
+
+        # Assert
+        assert result.method == "GET"
+        assert result.url == "http://example.com"
+        assert result.headers == {"Content-Type": "application/json"}
+        assert result.body == ""
+
+
+class TestLLMSpec:
+    def test_validate_raises_error_for_missing_files(self):
+        spec = LLMSpec(
+            method="POST", url="http://example.com", headers={}, body="", has_files=True
+        )
+        with pytest.raises(ValueError, match="Files are required for this request."):
+            spec.validate(prompt="", encoded_image="", encoded_audio="", files={})
+
+    def test_validate_raises_error_for_missing_image(self):
+        spec = LLMSpec(
+            method="POST", url="http://example.com", headers={}, body="", has_image=True
+        )
+        with pytest.raises(ValueError, match="An image is required for this request."):
+            spec.validate(prompt="", encoded_image="", encoded_audio="", files={})
+
+    @pytest.mark.asyncio
+    async def test_probe_sends_request(self, httpx_mock):
+        httpx_mock.add_response(
+            method="POST", url="http://example.com", status_code=200
+        )
+        spec = LLMSpec(
+            method="POST",
+            url="http://example.com",
+            headers={},
+            body='{"prompt": "<<PROMPT>>"}',
+        )
+        response = await spec.probe(prompt="test")
+        assert response.status_code == 200
+
+    @pytest.mark.asyncio
+    async def test_probe_with_files(self, httpx_mock):
+        httpx_mock.add_response(
+            method="POST", url="http://example.com", status_code=200
+        )
+        spec = LLMSpec(
+            method="POST",
+            url="http://example.com",
+            headers={"Content-Type": "multipart/form-data"},
+            body='{"prompt": "<<PROMPT>>"}',
+            has_files=True,
+        )
+        files = {"file": ("filename.txt", "file content")}
+        response = await spec.probe(prompt="test", files=files)
+        assert response.status_code == 200
+
+    @pytest.mark.asyncio
+    async def test_probe_with_image(self, httpx_mock):
+        httpx_mock.add_response(
+            method="POST", url="http://example.com", status_code=200
+        )
+        spec = LLMSpec(
+            method="POST",
+            url="http://example.com",
+            headers={},
+            body='{"image": "<<BASE64_IMAGE>>"}',
+            has_image=True,
+        )
+        encoded_image = "base64encodedstring"
+        response = await spec.probe(prompt="test", encoded_image=encoded_image)
+        assert response.status_code == 200