From a359c058f348057839b5b0870ce6636d7e6b6e66 Mon Sep 17 00:00:00 2001
From: Patrick Devaney <patrickbdevaney@gmail.com>
Date: Sun, 22 Dec 2024 23:33:39 -0500
Subject: [PATCH] adding unit test for spreadsheet_swarm

---
 swarms/structs/spreadsheet_swarm.py     | 12 +++-
 tests/structs/test_spreadsheet_swarm.py | 78 +++++++++++++++++++++++++
 2 files changed, 87 insertions(+), 3 deletions(-)
 create mode 100644 tests/structs/test_spreadsheet_swarm.py

diff --git a/swarms/structs/spreadsheet_swarm.py b/swarms/structs/spreadsheet_swarm.py
index 3a934920..fa583dad 100644
--- a/swarms/structs/spreadsheet_swarm.py
+++ b/swarms/structs/spreadsheet_swarm.py
@@ -150,8 +150,14 @@ class SpreadSheetSwarm(BaseSwarm):
         logger.info(f"Running the swarm with task: {task}")
         self.metadata.start_time = time
 
-        # Run the asyncio event loop
-        asyncio.run(self._run_tasks(task, *args, **kwargs))
+        # Check if we're already in an event loop
+        if asyncio.get_event_loop().is_running():
+            # If so, create and run tasks directly using `create_task` without `asyncio.run`
+            task_future = asyncio.create_task(self._run_tasks(task, *args, **kwargs))
+            asyncio.get_event_loop().run_until_complete(task_future)
+        else:
+            # If no event loop is running, run using `asyncio.run`
+            asyncio.run(self._run_tasks(task, *args, **kwargs))
 
         self.metadata.end_time = time
 
@@ -298,4 +304,4 @@ class SpreadSheetSwarm(BaseSwarm):
                         output.result,
                         output.timestamp,
                     ]
-                )
\ No newline at end of file
+                )
diff --git a/tests/structs/test_spreadsheet_swarm.py b/tests/structs/test_spreadsheet_swarm.py
new file mode 100644
index 00000000..2af3b6d8
--- /dev/null
+++ b/tests/structs/test_spreadsheet_swarm.py
@@ -0,0 +1,78 @@
+import os
+from datetime import datetime
+from uuid import uuid4
+
+# Import necessary classes from your swarm module
+from swarms.structs.agent import Agent
+from swarms.structs.base_swarm import BaseSwarm
+from swarms.telemetry.capture_sys_data import log_agent_data
+from swarms.utils.file_processing import create_file_in_folder
+from swarms import SpreadSheetSwarm
+
+# Ensure you have an environment variable or default workspace dir
+workspace_dir = os.getenv("WORKSPACE_DIR", "./workspace")
+
+def create_agents(num_agents: int):
+    """
+    Create a list of agent instances.
+    
+    Args:
+        num_agents (int): The number of agents to create.
+    
+    Returns:
+        List[Agent]: List of created Agent objects.
+    """
+    agents = []
+    for i in range(num_agents):
+        agent_name = f"Agent-{i + 1}"
+        agents.append(Agent(agent_name=agent_name))
+    return agents
+
+def main():
+    # Number of agents to create
+    num_agents = 5
+
+    # Create the agents
+    agents = create_agents(num_agents)
+
+    # Initialize the swarm with agents and other configurations
+    swarm = SpreadSheetSwarm(
+        name="Test-Swarm",
+        description="A swarm for testing purposes.",
+        agents=agents,
+        autosave_on=True,
+        max_loops=2,
+        workspace_dir=workspace_dir
+    )
+
+    # Run a sample task in the swarm (synchronously)
+    task = "process_data"
+    
+    # Ensure the run method is synchronous
+    swarm_metadata = swarm.run(task)  # Assuming this is made synchronous
+
+    # Print swarm metadata after task completion
+    print("Swarm Metadata:")
+    print(swarm_metadata)
+
+    # Check if CSV file has been created and saved
+    if os.path.exists(swarm.save_file_path):
+        print(f"Metadata saved to: {swarm.save_file_path}")
+    else:
+        print(f"Metadata not saved correctly. Check the save path.")
+
+    # Test saving metadata to JSON file
+    swarm.data_to_json_file()
+
+    # Test exporting metadata to JSON
+    swarm_json = swarm.export_to_json()
+    print("Exported JSON metadata:")
+    print(swarm_json)
+
+    # Log agent data
+    print("Logging agent data:")
+    print(log_agent_data(swarm.metadata.model_dump()))
+
+# Run the synchronous main function
+if __name__ == "__main__":
+    main()