Renamed python test files

airbus-cyber · Jan 5, 2024 · fa8e553 · fa8e553
1 parent 1cbc77b
commit fa8e553
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/validation/graylog/driver.py b/validation/graylog/driver.py
@@ -1,14 +1,14 @@
 import time
-from graylog.graylog_server import GraylogServer
-from graylog.graylog_rest_api import GraylogRestApi
+from graylog.server import Server
+from graylog.rest_api import RestApi
 from graylog.server_timeout_error import ServerTimeoutError
 
 
 class Driver:
 
     def __init__(self, docker_compose_path):
-        self._server = GraylogServer(docker_compose_path)
-        self._api = GraylogRestApi()
+        self._server = Server(docker_compose_path)
+        self._api = RestApi()
 
     def _wait(self, condition, attempts, sleep_duration=1):
         count = 0

diff --git a/validation/graylog/graylog_rest_api.py → validation/graylog/rest_api.py b/validation/graylog/graylog_rest_api.py → validation/graylog/rest_api.py
@@ -10,7 +10,7 @@
 _DEFAULT_LOG_BODY = 'type: alert\nid: ${logging_alert.id}\nseverity: ${logging_alert.severity}\napp: graylog\nsubject: ${event_definition_title}\nbody: ${event_definition_description}\n${if backlog && backlog[0]} src: ${backlog[0].fields.src_ip}\nsrc_category: ${backlog[0].fields.src_category}\ndest: ${backlog[0].fields.dest_ip}\ndest_category: ${backlog[0].fields.dest_category}\n${end}'
 
 
-class GraylogRestApi:
+class RestApi:
 
     def _build_url(self, path):
         return parse.urljoin('http://127.0.0.1:9000/api/', path)

diff --git a/validation/graylog/graylog_server.py → validation/graylog/server.py b/validation/graylog/graylog_server.py → validation/graylog/server.py
@@ -1,7 +1,7 @@
 import subprocess
 
 
-class GraylogServer:
+class Server:
 
     def __init__(self, docker_compose_path):
         self._docker_compose_path = docker_compose_path