Added mailboxes as local data sources (#1022)

app/Tests/Tools/ToolCalling/ToolRegistryTestBase.cs

@@ -60,7 +60,9 @@ public abstract class ToolRegistryTestBase
         this.rustService.Dispose();
     }
 
-    protected ToolRegistry CreateRegistry(params TestTool[] tools) => new(tools, [new CodeToolDefinitionSource(tools)], this.SettingsManager, this.CreateToolSettingsService(), NullLogger<ToolRegistry>.Instance);
+    protected ToolRegistry CreateRegistry(params TestTool[] tools) => this.CreateRegistry([], tools);
+
+    protected ToolRegistry CreateRegistry(IReadOnlyList<IToolCollection> collections, params TestTool[] tools) => new(tools, [new CodeToolDefinitionSource(tools)], collections, this.SettingsManager, this.CreateToolSettingsService(), NullLogger<ToolRegistry>.Instance);
 
     protected ToolSettingsService CreateToolSettingsService() => new(this.SettingsManager, this.rustService, NullLogger<ToolSettingsService>.Instance);
 
@@ -112,10 +114,25 @@ public abstract class ToolRegistryTestBase
     {
         public int ResolveCount { get; private set; }
 
+        /// <summary>
+        /// Where the tool sends data; when left out, the most open kind, as for every tool which says nothing.
+        /// </summary>
+        public ToolOutboundData OutboundData { get; init; } = ToolOutboundData.MODEL_CHOSEN_ADDRESSES;
+
+        public bool EnforcesOutboundDataRestriction { get; init; }
+
+        /// <summary>
+        /// Whether the tool exists right now. Settable, so a test can switch its preview off and on again.
+        /// </summary>
+        public bool IsAvailable { get; set; } = true;
+
         public string ImplementationKey => definition.ImplementationKey;
 
         public ToolDefinition GetDefinition() => definition;
 
+        // Named after its ID, so a test can tell the tools apart wherever their names appear:
+        public string GetDisplayName() => definition.Id;
+
         public ValueTask<ToolFunctionDefinition?> ResolveFunctionAsync(ToolDefinition registeredDefinition, ToolResolutionContext context, CancellationToken token = default)
         {
             this.ResolveCount++;
@@ -129,4 +146,26 @@ public abstract class ToolRegistryTestBase
 
         public Task<ToolExecutionResult> ExecuteAsync(JsonElement arguments, ToolExecutionContext context, CancellationToken token = default) => Task.FromResult(execute is null ? new ToolExecutionResult() : execute(context));
     }
+
+    /// <summary>
+    /// A tool collection which gathers the tools it is told to.
+    /// </summary>
+    /// <param name="id">The ID of the collection.</param>
+    /// <param name="minimumConfidence">The confidence the collection asks for itself.</param>
+    /// <param name="toolIds">The IDs of its tools.</param>
+    protected sealed class TestCollection(string id, ConfidenceLevel minimumConfidence, params string[] toolIds) : IToolCollection
+    {
+        public ToolCollectionDefinition GetDefinition() => new()
+        {
+            Id = id,
+            ToolIds = toolIds,
+            MinimumProviderConfidence = minimumConfidence,
+        };
+
+        public string Icon => string.Empty;
+
+        public string GetDisplayName() => id;
+
+        public string GetDescription() => id;
+    }
 }