Test case for HEC event

Author: ashah-splunk

splunklib/client.py

@@ -783,7 +783,7 @@ def get(self, path_segment="", owner=None, app=None, sharing=None, **query):
             path = path_segment
         else:
             if not self.path.endswith('/') and path_segment != "":
-                self.path = self.path if path_segment.startswith('/') else self.path + '/'
+                self.path = self.path if path_segment != "" else self.path + '/'
             path = self.service._abspath(self.path + path_segment, owner=owner,
                                          app=app, sharing=sharing)
         # ^-- This was "%s%s" % (self.path, path_segment).
@@ -845,7 +845,7 @@ def post(self, path_segment="", owner=None, app=None, sharing=None, **query):
             path = path_segment
         else:
             if not self.path.endswith('/') and path_segment != "":
-                self.path = self.path if path_segment.startswith('/') else self.path + '/'
+                self.path = self.path if path_segment != "" else self.path + '/'
             path = self.service._abspath(self.path + path_segment, owner=owner, app=app, sharing=sharing)
         return self.service.post(path, owner=owner, app=app, sharing=sharing, **query)
 

tests/test_service.py

@@ -167,6 +167,17 @@ def _create_unauthenticated_service(self):
             'scheme': self.opts.kwargs['scheme']
         })
 
+    #To check the HEC event endpoint using Endpoint instance
+    def test_hec_event(self):
+        import json
+        service_hec = client.connect(host='localhost', scheme='https', port=8088,
+                                     token="11111111-1111-1111-1111-1111111111113")
+        event_collector_endpoint = client.Endpoint(service_hec, "/services/collector/event")
+        msg = {"index": "main", "event": "Hello World"}
+        response = event_collector_endpoint.post("", body=json.dumps(msg))
+        body = response.body.read()
+        self.assertEqual(body.code, 200)
+
 
 class TestCookieAuthentication(unittest.TestCase):
     def setUp(self):