// 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.Http;
using Microsoft.AspNetCore.Rewrite.PatternSegments;
using Microsoft.Net.Http.Headers;
 
namespace Microsoft.AspNetCore.Rewrite.Tests.PatternSegments;
 
public class HeaderSegmentTests
{
    [Fact]
    public void HeaderSegment_AssertGettingWithHeaderReturnsCorrectValue()
    {
        // Arrange
        var context = new RewriteContext { HttpContext = new DefaultHttpContext() };
 
        context.HttpContext.Request.Headers.Location = "foo";
        var segment = new HeaderSegment(HeaderNames.Location);
 
        // Act
        var results = segment.Evaluate(context, null, null);
 
        // Assert
        Assert.Equal("foo", results);
    }
 
    [Fact]
    public void HeaderSegment_AssertGettingANonExistantHeaderReturnsNull()
    {
        // Arrange
        var context = new RewriteContext { HttpContext = new DefaultHttpContext() };
        var segment = new HeaderSegment(HeaderNames.Location);
 
        // Act
        var results = segment.Evaluate(context, null, null);
 
        // Assert
        Assert.Null(results);
    }
}