// 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.UrlActions;
using Microsoft.Net.Http.Headers;
 
namespace Microsoft.AspNetCore.Rewrite.Tests.UrlActions;
 
public class ChangeCookieActionTests
{
    [Fact]
    public void SetsCookie()
    {
        var now = DateTimeOffset.UtcNow;
        var context = new RewriteContext { HttpContext = new DefaultHttpContext() };
        var action = new ChangeCookieAction("Cookie", () => now)
        {
            Value = "Chocolate Chip",
            Domain = "contoso.com",
            Lifetime = TimeSpan.FromMinutes(1440),
            Path = "/recipes",
            Secure = true,
            HttpOnly = true
        };
 
        action.ApplyAction(context, null, null);
 
        var cookieHeaders = context.HttpContext.Response.Headers.SetCookie;
        var header = Assert.Single(cookieHeaders);
        Assert.Equal($"Cookie=Chocolate%20Chip; expires={HeaderUtilities.FormatDate(now.AddMinutes(1440))}; domain=contoso.com; path=/recipes; secure; httponly", header);
    }
 
    [Fact]
    public void ZeroLifetime()
    {
        var context = new RewriteContext { HttpContext = new DefaultHttpContext() };
        var action = new ChangeCookieAction("Cookie")
        {
            Value = "Chocolate Chip",
        };
 
        action.ApplyAction(context, null, null);
 
        var cookieHeaders = context.HttpContext.Response.Headers.SetCookie;
        var header = Assert.Single(cookieHeaders);
        Assert.Equal($"Cookie=Chocolate%20Chip", header);
    }
 
    [Fact]
    public void UnsetCookie()
    {
        var context = new RewriteContext { HttpContext = new DefaultHttpContext() };
        var action = new ChangeCookieAction("Cookie");
 
        action.ApplyAction(context, null, null);
 
        var cookieHeaders = context.HttpContext.Response.Headers.SetCookie;
        var header = Assert.Single(cookieHeaders);
        Assert.Equal($"Cookie=", header);
    }
}