test: add assertion for remove_server method call in client edit test

- Enhanced test_client_edit_non_interactive_remove_server to verify that
  client_manager.remove_server is called with the correct prefixed server name
- Added proper mock setup for MCPM-managed server configuration
- Ensures the removal operation is actually triggered as intended
- Also includes automatic formatting fixes in test_profile_commands.py

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: niechen

tests/test_client.py

@@ -443,8 +443,13 @@ def test_client_edit_non_interactive_remove_server(monkeypatch):
     mock_client_manager = Mock()
     mock_client_manager.is_client_installed = Mock(return_value=True)
     mock_client_manager.config_path = "/path/to/config.json"
-    mock_client_manager.get_servers.return_value = {"existing-server": Mock()}
+    # Mock an MCPM-managed server in client config
+    existing_mcpm_server = Mock()
+    existing_mcpm_server.command = "mcpm"
+    existing_mcpm_server.args = ["run", "existing-server"]
+    mock_client_manager.get_servers.return_value = {"mcpm_existing-server": existing_mcpm_server}
     mock_client_manager.update_servers.return_value = None
+    mock_client_manager.remove_server.return_value = None
 
     monkeypatch.setattr("mcpm.commands.client.ClientRegistry.get_client_manager", Mock(return_value=mock_client_manager))
     monkeypatch.setattr("mcpm.commands.client.ClientRegistry.get_client_info", Mock(return_value={"name": "Cursor"}))
@@ -463,9 +468,12 @@ def test_client_edit_non_interactive_remove_server(monkeypatch):
         "--remove-server", "existing-server"
     ])
 
-    # The command runs without crashing, even if the server wasn't in the client
+    # The command runs without crashing and removes the server
     assert result.exit_code == 0
     assert "Cursor Configuration Management" in result.output
+    
+    # Verify that remove_server was called with the prefixed server name
+    mock_client_manager.remove_server.assert_called_with("mcpm_existing-server")
 
 
 def test_client_edit_non_interactive_set_servers(monkeypatch):

tests/test_profile_commands.py

@@ -61,7 +61,7 @@ def test_profile_edit_non_interactive_remove_server(monkeypatch):
     # Mock GlobalConfigManager (needed for server validation)
     mock_global_config = Mock()
     mock_global_config.list_servers.return_value = {
-        "server1": server1, 
+        "server1": server1,
         "server2": server2
     }
     monkeypatch.setattr("mcpm.commands.profile.edit.global_config_manager", mock_global_config)
@@ -246,7 +246,7 @@ def test_profile_edit_interactive_fallback(monkeypatch):
     mock_profile_config.get_profile.return_value = [existing_server]
     monkeypatch.setattr("mcpm.commands.profile.edit.profile_config_manager", mock_profile_config)
 
-    # Mock GlobalConfigManager 
+    # Mock GlobalConfigManager
     mock_global_config = Mock()
     mock_global_config.list_servers.return_value = {"existing-server": existing_server}
     monkeypatch.setattr("mcpm.commands.profile.edit.global_config_manager", mock_global_config)
@@ -281,7 +281,7 @@ def test_profile_inspect_non_interactive(monkeypatch):
     class MockCompletedProcess:
         def __init__(self, returncode=0):
             self.returncode = returncode
-            
+
     mock_subprocess = Mock()
     mock_subprocess.run.return_value = MockCompletedProcess(0)
     monkeypatch.setattr("mcpm.commands.profile.inspect.subprocess", mock_subprocess)