Add bearer token input for testing, Origin header

Author: v7fasttrack

mcp_testing/http/tester.py

@@ -17,7 +17,7 @@
 class MCPHttpTester:
     """Class to test an MCP HTTP server implementation."""
 
-    def __init__(self, url, debug=False):
+    def __init__(self, url, debug=False, bearer_token = None):
         """
         Initialize the tester with the server URL.
         
@@ -36,16 +36,25 @@ def __init__(self, url, debug=False):
         # Session information
         self.session_id = None
         self.initialized = False
+        self.bearer_token = bearer_token
 
         # Protocol information
         self.protocol_version = "2025-03-26"
 
         # Create a persistent session for all requests
         self.request_session = requests.Session()
-        self.request_session.headers.update({
+        # Common headers
+        headers = {
             "Content-Type": "application/json",
+            "Origin": "http://localhost",
             "Accept": "application/json, text/event-stream"
-        })
+        }
+
+        # Add Authorization header only if bearer_token is provided
+        if self.bearer_token is not None:
+            headers["Authorization"] = f"Bearer {self.bearer_token}"
+
+        self.request_session.headers.update(headers)
 
         self.log(f"MCP HTTP Tester initialized for {url}")
         self.log(f"Host: {self.host}, Path: {self.path}")
@@ -628,4 +637,4 @@ def run_all_tests(self):
             print(f"Error during test execution: {str(e)}")
             import traceback
             traceback.print_exc()
-            return False 
+            return False 

mcp_testing/scripts/http_test.py

@@ -34,6 +34,10 @@ def main():
             default="2025-03-26", 
             help="Protocol version to use (default: 2025-03-26)"
         )
+        parser.add_argument(
+            "--bearer-token", 
+            help="Protocol version to use (default: 2025-03-26)"
+        )
         parser.add_argument(
             "--debug", 
             action="store_true", 
@@ -71,7 +75,7 @@ def main():
             return 1
 
         # Run the HTTP tests
-        tester = MCPHttpTester(args.server_url, args.debug)
+        tester = MCPHttpTester(args.server_url, args.debug, args.bearer_token)
         tester.protocol_version = args.protocol_version
 
         success = tester.run_all_tests()
@@ -103,4 +107,4 @@ def main():
         return 1
 
 if __name__ == "__main__":
-    sys.exit(main()) 
+    sys.exit(main())