feat: dual mode should also support mock

Signed-off-by: vsoch <vsoch@users.noreply.github.com>

Author: vsoch

mcpserver/cli/args.py

@@ -121,13 +121,14 @@ def populate_start_args(start):
     # const=True is what we get if the flag is present but no value is given
     # default=False is what we get if the flag is totally absent
     # THe user can also ask for an archetype (hpc, cloud, standalone)
-    worker_group.add_argument(
-        "--mock",
-        nargs="?",
-        const=True,
-        default=False,
-        help="Start a mock worker. Can optionally select hpc, cloud, or standalone",
-    )
+    for group in worker_group, hub_group:
+        group.add_argument(
+            "--mock",
+            nargs="?",
+            const=True,
+            default=False,
+            help="Start a mock worker. Can optionally select hpc, cloud, or standalone",
+        )
 
     # Agent Reasoning Group
     agent_group = start.add_argument_group("🧠 Agent Reasoning")

mcpserver/core/base.py

@@ -2,7 +2,11 @@
 import json
 import time
 
+from resource_secretary.providers import discover_providers
+from resource_secretary.providers.mock import discover_mock_providers
+
 import mcpserver.utils as utils
+from mcpserver.logger import logger
 
 
 class WorkerBase:
@@ -12,6 +16,17 @@ class WorkerBase:
     its dual mode (acting as worker AND hub.)
     """
 
+    def init_providers(self, mock=False):
+        """
+        Probe the local system on startup. E.g., "we found spack, flux, etc."
+        These can be faux (mock) or real discovered providers
+        """
+        logger.info("📡 Probing local system for resource providers...")
+        if mock:
+            self.catalog = discover_mock_providers(self.worker_id, choice=mock)
+        else:
+            self.catalog = discover_providers()
+
     def register_agent_tools(self):
         """
         Registers the core negotiation tools with the FastMCP instance.

mcpserver/core/hub.py

@@ -397,6 +397,7 @@ def __init__(self, *args, **kwargs):
         # Calls super on the HubManager. WorkerBase has no init
         super().__init__(*args, **kwargs)
         self.setup_dual()
+        self.init_providers(kwargs.get("mock", False))
 
     def setup_dual(self):
         """

mcpserver/core/worker.py

@@ -47,17 +47,6 @@ def __init__(
         # Register MCP Tools automatically
         self.register_agent_tools()
 
-    def init_providers(self, mock=False):
-        """
-        Probe the local system on startup. E.g., "we found spack, flux, etc."
-        These can be faux (mock) or real discovered providers
-        """
-        logger.info("📡 Probing local system for resource providers...")
-        if mock:
-            self.catalog = discover_mock_providers(self.worker_id, choice=mock)
-        else:
-            self.catalog = discover_providers()
-
     def show(self):
         """
         Show providers installed and verbosity.