Merge pull request #106 from codefromlani/fix/get-version-from-package

fix: dynamically fetch version from package metadata and update test

Author: Agent-Hellboy

mcp_fuzzer/reports/output_protocol.py

@@ -15,12 +15,19 @@
 
 from ..exceptions import ValidationError
 
+from importlib.metadata import version, PackageNotFoundError
+
+try:
+    TOOL_VERSION = version("mcp-fuzzer")
+except PackageNotFoundError:
+    TOOL_VERSION = "unknown"
+
 
 class OutputProtocol:
     """Handles standardized output format with mini-protocol for MCP Fuzzer."""
 
     PROTOCOL_VERSION = "1.0.0"
-    TOOL_VERSION = "0.1.6"  # TODO: Get from package version
+    TOOL_VERSION = TOOL_VERSION
 
     # Output types
     OUTPUT_TYPES = {

mcp_fuzzer/reports/reporter.py

@@ -26,6 +26,13 @@
 from .output_protocol import OutputManager
 from .safety_reporter import SafetyReporter
 
+from importlib.metadata import version, PackageNotFoundError
+
+try:
+    fuzzer_version = version("mcp-fuzzer")
+except PackageNotFoundError:
+    fuzzer_version = "unknown"
+
 
 class FuzzerReporter:
     """Centralized reporter for all MCP Fuzzer output and reporting."""
@@ -100,7 +107,7 @@ def set_fuzzing_metadata(
             "endpoint": endpoint,
             "runs": runs,
             "runs_per_type": runs_per_type,
-            "fuzzer_version": "1.0.0",  # TODO: Get from package
+            "fuzzer_version": fuzzer_version,
         }
 
     def add_tool_results(self, tool_name: str, results: List[Dict[str, Any]]):

tests/unit/reports/test_output_protocol.py

@@ -12,6 +12,13 @@
 
 from mcp_fuzzer.reports.output_protocol import OutputProtocol, OutputManager
 
+from importlib.metadata import version, PackageNotFoundError
+
+try:
+    expected_version = version("mcp-fuzzer")
+except PackageNotFoundError:
+    expected_version = "unknown"
+
 
 class TestOutputProtocol:
     """Test cases for OutputProtocol class."""
@@ -26,7 +33,7 @@ def test_create_base_output(self):
         output = self.protocol.create_base_output("fuzzing_results", data)
 
         assert output["protocol_version"] == "1.0.0"
-        assert output["tool_version"] == "0.1.6"
+        assert output["tool_version"] == expected_version
         assert output["session_id"] == "test-session-123"
         assert output["output_type"] == "fuzzing_results"
         assert output["data"] == data