Factored test code somewhat

validation/graylog.py

@@ -51,12 +51,13 @@ def create_gelf_input(self):
     def create_correlation_count(self, *args, **kwargs):
         self._api.create_correlation_count(*args, **kwargs)
 
-    def get_events_count(self):
-        return self._api.get_events_count()
-
     def get_events(self):
         return self._api.get_events()
 
+    def get_events_count(self):
+        events = self.get_events()
+        return events['total_events']
+
     def _has_event(self):
         events_count = self.get_events_count()
         return events_count == 1

validation/graylog_rest_api.py

@@ -119,9 +119,4 @@ def events_search_is_available(self):
 
     def get_events(self):
         response = self._search_events()
-        body = response.json()
-        return body
-
-    def get_events_count(self):
-        body = self.get_events()
-        return body['total_events']
+        return response.json()