unit test fixes

Author: aponcedeleonch

src/codegate/pipeline/cli/commands.py

@@ -156,11 +156,11 @@ async def _add_workspace(self, flags: Dict[str, str], args: List[str]) -> str:
         Add a workspace
         """
         if args is None or len(args) == 0:
-            return "Please provide a name. Use `codegate workspace add <workspace_name>`"
+            return "Please provide a name. Use `codegate workspace add your_workspace_name`"
 
         new_workspace_name = args[0]
         if not new_workspace_name:
-            return "Please provide a name. Use `codegate workspace add <workspace_name>`"
+            return "Please provide a name. Use `codegate workspace add your_workspace_name`"
 
         try:
             _ = await self.workspace_crud.add_workspace(new_workspace_name)
@@ -178,11 +178,11 @@ async def _activate_workspace(self, flags: Dict[str, str], args: List[str]) -> s
         Activate a workspace
         """
         if args is None or len(args) == 0:
-            return "Please provide a name. Use `codegate workspace activate <workspace_name>`"
+            return "Please provide a name. Use `codegate workspace activate workspace_name`"
 
         workspace_name = args[0]
         if not workspace_name:
-            return "Please provide a name. Use `codegate workspace activate <workspace_name>`"
+            return "Please provide a name. Use `codegate workspace activate workspace_name`"
 
         try:
             await self.workspace_crud.activate_workspace(workspace_name)

tests/pipeline/workspace/test_workspace.py

@@ -42,7 +42,7 @@ async def test_list_workspaces(mock_workspaces, expected_output):
     workspace_commands.workspace_crud.get_workspaces = mock_get_workspaces
 
     # Call the method
-    result = await workspace_commands._list_workspaces()
+    result = await workspace_commands._list_workspaces(None, None)
 
     # Check the result
     assert result == expected_output
@@ -54,11 +54,11 @@ async def test_list_workspaces(mock_workspaces, expected_output):
     "args, existing_workspaces, expected_message",
     [
         # Case 1: No workspace name provided
-        ([], [], "Please provide a name. Use `codegate workspace add <workspace_name>`"),
+        ([], [], "Please provide a name. Use `codegate workspace add your_workspace_name`"),
         # Case 2: Workspace name is empty string
-        ([""], [], "Please provide a name. Use `codegate workspace add <workspace_name>`"),
+        ([""], [], "Please provide a name. Use `codegate workspace add your_workspace_name`"),
         # Case 3: Successful add
-        (["myworkspace"], [], "Workspace `myworkspace` has been added"),
+        (["myworkspace"], [], "Workspace **myworkspace** has been added"),
     ],
 )
 async def test_add_workspaces(args, existing_workspaces, expected_message):
@@ -83,7 +83,7 @@ async def test_add_workspaces(args, existing_workspaces, expected_message):
         mock_recorder.add_workspace = AsyncMock()
 
         # Call the method
-        result = await workspace_commands._add_workspace(args)
+        result = await workspace_commands._add_workspace(None, args)
 
         # Assertions
         assert result == expected_message