feat: dual mode should also support mock

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

Author: vsoch

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

@@ -33,14 +33,19 @@ def __init__(
         dual=False,
         hub_id=None,
         path="/mcp",
+        mock=False,
     ):
+        # Probably can simplify some of this between worker and hub
         self.mcp = mcp
         self.host = host
         self.port = port
         self.path = path
         self.secret = secret or secrets.token_urlsafe(32)
         self.workers: Dict[str, Dict[str, Any]] = {}
-        self.hub_id = hub_id or socket.gethostname()
+
+        # For use if we are also a worker.
+        self.worker_id = hub_id or socket.gethostname()
+        self.mock = mock
 
         # Make requests to hub in batches, in serial, or in parallel
         self.set_running_mode(batch, serial, dual)
@@ -103,6 +108,7 @@ def from_args(cls, mcp, args) -> Optional["HubManager"]:
             batch=args.batch,
             serial=args.serial,
             dual=args.dual,
+            mock=args.mock,
             # server path
             path=args.path,
         )
@@ -397,12 +403,13 @@ 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):
         """
         Setup dual mode, which means adding ourselves to the fleet.
         """
-        hub_id = self.hub_id or socket.gethostname()
+        hub_id = self.worker_id or socket.gethostname()
         default_url = f"http://{self.host}:{self.port}{self.path}"
         self.workers[hub_id] = {
             "url": self.registration_url,

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.