HttpClientHints/tests/HttpClientHints.AspNetCore.UnitTests/HttpClientHintsRegistrationTests.cs at db923681c000cbf046d8dc106efd44e2ee1386bd · mycsharp/HttpClientHints · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
// Copyright © https://myCSharp.de - all rights reserved

using Microsoft.AspNetCore.Builder;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options;
using NSubstitute;
using Xunit;

namespace MyCSharp.HttpClientHints.AspNetCore.UnitTests;

public class HttpClientHintsRegistrationTests
{
    [Fact]
    public void AddHttpClientHints_DefaultOptions_RegistersDefaultHeaders()
    {
        // Arrange
        ServiceCollection services = new();

        // Act
        services.AddHttpClientHints();
        ServiceProvider serviceProvider = services.BuildServiceProvider();
        HttpClientHintsMiddlewareConfig options = serviceProvider.GetRequiredService<IOptions<HttpClientHintsMiddlewareConfig>>().Value;

        // Assert
        Assert.Contains("User-Agent", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA-Platform", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA-Platform-Version", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA-Arch", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA-Bitness", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA-Model", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Contains("Sec-CH-UA-Mobile", options.ResponseHeader, StringComparison.Ordinal);
        Assert.Null(options.LifeTime);
    }

    [Fact]
    public void AddHttpClientHints_CustomLifetime_RegistersLifetime()
    {
        // Arrange
        ServiceCollection services = new();
        TimeSpan lifetime = TimeSpan.FromMinutes(30);

        // Act
        services.AddHttpClientHints(options => options.Lifetime = lifetime);
        ServiceProvider serviceProvider = services.BuildServiceProvider();
        HttpClientHintsMiddlewareConfig options = serviceProvider.GetRequiredService<IOptions<HttpClientHintsMiddlewareConfig>>().Value;

        // Assert
        Assert.Equal("1800", options.LifeTime); // 1800 seconds
    }

    [Fact]
    public void AddHttpClientHints_DisablePlatformHeader_DoesNotRegisterPlatformHeaders()
    {
        // Arrange
        ServiceCollection services = new();

        // Act
        services.AddHttpClientHints(options => options.Platform = false);
        ServiceProvider serviceProvider = services.BuildServiceProvider();
        HttpClientHintsMiddlewareConfig options = serviceProvider.GetRequiredService<IOptions<HttpClientHintsMiddlewareConfig>>().Value;

        // Assert
        Assert.DoesNotContain("Sec-CH-UA-Platform", options.ResponseHeader, StringComparison.Ordinal);
        Assert.DoesNotContain("Sec-CH-UA-Platform-Version", options.ResponseHeader, StringComparison.Ordinal);
    }

    [Fact]
    public void AddHttpClientHints_CustomAdditionalHeader_RegistersAdditionalHeader()
    {
        // Arrange
        ServiceCollection services = new();
        string[] customHeaders = ["X-Custom-Header"];

        // Act
        services.AddHttpClientHints(options => options.Additional = customHeaders);
        ServiceProvider serviceProvider = services.BuildServiceProvider();
        HttpClientHintsMiddlewareConfig options = serviceProvider.GetRequiredService<IOptions<HttpClientHintsMiddlewareConfig>>().Value;

        // Assert
        Assert.Contains("X-Custom-Header", options.ResponseHeader, StringComparison.Ordinal);
    }

    [Fact]
    public void AddHttpClientHints_AllHeadersDisabled_RegistersNoHeaders()
    {
        // Arrange
        ServiceCollection services = new();

        // Act
        services.AddHttpClientHints(options =>
        {
            options.UserAgent = false;
            options.Platform = false;
            options.Architecture = false;
            options.Device = false;
            options.Mobile = false;
            options.Additional = null;
        });
        ServiceProvider serviceProvider = services.BuildServiceProvider();
        HttpClientHintsMiddlewareConfig options = serviceProvider.GetRequiredService<IOptions<HttpClientHintsMiddlewareConfig>>().Value;

        // Assert
        Assert.Empty(options.ResponseHeader);
    }

    [Fact]
    public void UseHttpClientHints_ReturnsApplicationBuilder()
    {
        // Arrange
        IApplicationBuilder appBuilder = Substitute.For<IApplicationBuilder>();

        // Act
        IApplicationBuilder result = appBuilder.UseHttpClientHints();

        // Assert
        Assert.NotNull(result);
        Assert.Same(appBuilder, result); // Ensure the same instance is returned
    }
}