From a28d6a2da5debe73e645b6c57ca3868e9c4b53ce Mon Sep 17 00:00:00 2001
From: mohsinenar <mnarmouche@gmail.com>
Date: Thu, 28 Mar 2024 14:32:55 +0000
Subject: [PATCH] Apply ruff.

---
 agent/nebula_agent.py      |  7 +++----
 tests/nebula_agent_test.py | 20 +++++++++-----------
 2 files changed, 12 insertions(+), 15 deletions(-)

diff --git a/agent/nebula_agent.py b/agent/nebula_agent.py
index 4f74cfe..648d856 100644
--- a/agent/nebula_agent.py
+++ b/agent/nebula_agent.py
@@ -4,7 +4,6 @@
 import json
 import logging
 import os
-from datetime import datetime
 from typing import Any
 
 from ostorlab.agent import agent, definitions as agent_definitions
@@ -35,9 +34,9 @@ class NebulaAgent(agent.Agent):
     """Agent responsible for persisting all types of messages to file type specified in the agent definition."""
 
     def __init__(
-        self,
-        agent_definition: agent_definitions.AgentDefinition,
-        agent_settings: runtime_definitions.AgentSettings,
+            self,
+            agent_definition: agent_definitions.AgentDefinition,
+            agent_settings: runtime_definitions.AgentSettings,
     ) -> None:
         super().__init__(agent_definition, agent_settings)
         self._file_type = self.args.get("file_type", "json")
diff --git a/tests/nebula_agent_test.py b/tests/nebula_agent_test.py
index b47e085..6f1c5de 100644
--- a/tests/nebula_agent_test.py
+++ b/tests/nebula_agent_test.py
@@ -5,7 +5,6 @@
 import pathlib
 
 import pytest
-from freezegun import freeze_time
 from ostorlab.agent import definitions as agent_definitions
 from ostorlab.agent.message import message as msg
 from ostorlab.runtimes import definitions as runtime_definitions
@@ -13,7 +12,6 @@
 from pyfakefs import fake_filesystem_unittest
 
 from agent import nebula_agent
-import os
 
 
 def testAgentNebula_whenUnsupportedFileType_raisesValueError() -> None:
@@ -39,9 +37,9 @@ def testAgentNebula_whenUnsupportedFileType_raisesValueError() -> None:
 
 
 def testAgentNebula_whenFileTypeIsJson_persistMessage(
-    agent_definition: agent_definitions.AgentDefinition,
-    agent_settings: runtime_definitions.AgentSettings,
-    link_message: msg.Message,
+        agent_definition: agent_definitions.AgentDefinition,
+        agent_settings: runtime_definitions.AgentSettings,
+        link_message: msg.Message,
 ) -> None:
     """Test that NebulaAgent persists message to json file."""
     os.environ["UNIVERSE"] = "43"
@@ -63,9 +61,9 @@ def testAgentNebula_whenFileTypeIsJson_persistMessage(
 
 
 def testAgentNebula_whenFileTypeIsJson_persistMultipleLinkMessages(
-    agent_definition: agent_definitions.AgentDefinition,
-    agent_settings: runtime_definitions.AgentSettings,
-    multiple_link_messages: list[msg.Message],
+        agent_definition: agent_definitions.AgentDefinition,
+        agent_settings: runtime_definitions.AgentSettings,
+        multiple_link_messages: list[msg.Message],
 ) -> None:
     """Test that NebulaAgent persists multiple link messages to json file."""
     os.environ["UNIVERSE"] = "43"
@@ -93,9 +91,9 @@ def testAgentNebula_whenFileTypeIsJson_persistMultipleLinkMessages(
 
 
 def testAgentNebula_whenFileTypeIsJson_persistMultipleMessages(
-    agent_definition: agent_definitions.AgentDefinition,
-    agent_settings: runtime_definitions.AgentSettings,
-    multiple_messages: list[msg.Message],
+        agent_definition: agent_definitions.AgentDefinition,
+        agent_settings: runtime_definitions.AgentSettings,
+        multiple_messages: list[msg.Message],
 ) -> None:
     """Test that NebulaAgent persists multiple messages of different types to json files."""
     os.environ["UNIVERSE"] = "43"