static void SetupTenantDatabases(EndpointBehaviorBuilder <Context> builder)
    {
        EndpointConfiguration cfg = null;

        builder.CustomConfig(c => cfg = c);

        builder.When((session, context) =>
        {
            var tablePrefix = cfg.GetSettings().EndpointName().Replace(".", "_");
            MsSqlConnectionBuilder.MultiTenant.Setup("TenantA");
            MsSqlConnectionBuilder.MultiTenant.Setup("TenantB");
            var helperA     = new ConfigureEndpointHelper(cfg, tablePrefix, () => MsSqlConnectionBuilder.MultiTenant.Build("TenantA"), BuildSqlDialect.MsSqlServer, null);
            var helperB     = new ConfigureEndpointHelper(cfg, tablePrefix, () => MsSqlConnectionBuilder.MultiTenant.Build("TenantB"), BuildSqlDialect.MsSqlServer, null);
            context.Cleanup = () =>
            {
                helperA.Cleanup();
                helperB.Cleanup();
            };
            return(Task.FromResult(0));
        });
    }
 public Task Cleanup()
 {
     return(endpointHelper?.Cleanup());
 }
 public Task Cleanup()
 {
     endpointHelper?.Cleanup();
     return(Task.FromResult(0));
 }