aspnetcore/test/Microsoft.AspNetCore.SignalR.Client.Tests/HubConnectionBuilderExtensionsTests.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Net;
using Microsoft.AspNetCore.Http.Connections;
using Microsoft.AspNetCore.Http.Connections.Client;
using Microsoft.AspNetCore.SignalR.Protocol;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using Moq;
using Xunit;

namespace Microsoft.AspNetCore.SignalR.Client.Tests
{
    public class HubConnectionBuilderExtensionsTests
    {
        [Fact]
        public void WithHttpConnectionSetsUrl()
        {
            var connectionBuilder = new HubConnectionBuilder();
            connectionBuilder.WithUrl("http://tempuri.org");

            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var value = serviceProvider.GetService<IOptions<HttpConnectionOptions>>().Value;

            Assert.Equal(new Uri("http://tempuri.org"), value.Url);
        }

        [Fact]
        public void WithHttpConnectionSetsTransport()
        {
            var connectionBuilder = new HubConnectionBuilder();
            connectionBuilder.WithUrl("http://tempuri.org", HttpTransportType.LongPolling);

            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var value = serviceProvider.GetService<IOptions<HttpConnectionOptions>>().Value;

            Assert.Equal(HttpTransportType.LongPolling, value.Transports);
        }

        [Fact]
        public void WithHttpConnectionCallsConfigure()
        {
            var proxy = Mock.Of<IWebProxy>();

            var connectionBuilder = new HubConnectionBuilder();
            connectionBuilder.WithUrl("http://tempuri.org", options => { options.Proxy = proxy; });

            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var value = serviceProvider.GetService<IOptions<HttpConnectionOptions>>().Value;

            Assert.Same(proxy, value.Proxy);
        }

        [Fact]
        public void DefaultLoggerFactoryExists()
        {
            var connectionBuilder = new HubConnectionBuilder();
            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var loggerFactory = serviceProvider.GetService<ILoggerFactory>();
            Assert.NotNull(loggerFactory);
        }

        [Fact]
        public void WithHubProtocolAddsProtocol()
        {
            var hubProtocol = Mock.Of<IHubProtocol>();

            var connectionBuilder = new HubConnectionBuilder();
            connectionBuilder.WithHubProtocol(hubProtocol);

            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var resolvedHubProtocol = serviceProvider.GetService<IHubProtocol>();

            Assert.Same(hubProtocol, resolvedHubProtocol);
        }

        [Fact]
        public void AddJsonProtocolAddsProtocol()
        {
            var connectionBuilder = new HubConnectionBuilder();
            connectionBuilder.AddJsonProtocol();

            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var resolvedHubProtocol = serviceProvider.GetService<IHubProtocol>();

            Assert.IsType<JsonHubProtocol>(resolvedHubProtocol);
        }

        [Fact]
        public void AddMessagePackProtocolAddsProtocol()
        {
            var connectionBuilder = new HubConnectionBuilder();
            connectionBuilder.AddMessagePackProtocol();

            var serviceProvider = connectionBuilder.Services.BuildServiceProvider();

            var resolvedHubProtocol = serviceProvider.GetService<IHubProtocol>();

            Assert.IsType<MessagePackHubProtocol>(resolvedHubProtocol);
        }
    }
}