// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Microsoft.AspNetCore.Connections;
using Microsoft.AspNetCore.SignalR.Protocol;
using Microsoft.AspNetCore.SignalR.Tests;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
 
namespace Microsoft.AspNetCore.SignalR.Client.Tests;
 
public partial class HubConnectionTests
{
    private static HubConnection CreateHubConnection(TestConnection connection, IHubProtocol protocol = null, ILoggerFactory loggerFactory = null)
    {
        var builder = new HubConnectionBuilder().WithUrl("http://example.com");
 
        var delegateConnectionFactory = new DelegateConnectionFactory(
            endPoint => connection.StartAsync());
 
        builder.Services.AddSingleton<IConnectionFactory>(delegateConnectionFactory);
 
        if (loggerFactory != null)
        {
            builder.WithLoggerFactory(loggerFactory);
        }
 
        if (protocol != null)
        {
            builder.Services.AddSingleton(protocol);
        }
 
        return builder.Build();
    }
}