// 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.Collections.Generic;
using System.Globalization;
using System.Threading.Tasks;
using Microsoft.AspNet.Builder;
using Microsoft.AspNet.Hosting;
using Microsoft.AspNet.Http.Features;
using Microsoft.AspNet.Localization;
using Microsoft.AspNet.TestHost;
using Microsoft.Net.Http.Headers;
using Xunit;

namespace Microsoft.Extensions.Localization.Tests
{
    public class CookieRequestCultureProviderTest
    {
        [Fact]
        public async void GetCultureInfoFromPersistentCookie()
        {
            var builder = new WebApplicationBuilder()
                .Configure(app =>
                {
                    app.UseRequestLocalization(options =>
                    {
                        options.DefaultRequestCulture = new RequestCulture("en-US");
                        options.SupportedCultures = new List<CultureInfo>
                        {
                            new CultureInfo("ar-SA")
                        };
                        options.SupportedUICultures = new List<CultureInfo>
                        {
                            new CultureInfo("ar-SA")
                        };
                        var provider = new CookieRequestCultureProvider();
                        provider.CookieName = "Preferences";
                        options.RequestCultureProviders.Insert(0, provider);
                    });
                    app.Run(context =>
                    {
                        var requestCultureFeature = context.Features.Get<IRequestCultureFeature>();
                        var requestCulture = requestCultureFeature.RequestCulture;
                        Assert.Equal("ar-SA", requestCulture.Culture.Name);
                        return Task.FromResult(0);
                    });
                });

            using (var server = new TestServer(builder))
            {
                var client = server.CreateClient();
                var culture = new CultureInfo("ar-SA");
                var requestCulture = new RequestCulture(culture);
                var value = CookieRequestCultureProvider.MakeCookieValue(requestCulture);
                client.DefaultRequestHeaders.Add("Cookie", new CookieHeaderValue("Preferences", value).ToString());
                var response = await client.GetAsync(string.Empty);
                Assert.Equal("c=ar-SA|uic=ar-SA", value);
            }
        }

        [Fact]
        public async void GetDefaultCultureInfoIfCultureKeysAreMissingOrInvalid()
        {
            var builder = new WebApplicationBuilder()
                .Configure(app =>
                {
                    app.UseRequestLocalization(options =>
                    {
                        options.DefaultRequestCulture = new RequestCulture("en-US");
                        options.SupportedCultures = new List<CultureInfo>
                        {
                            new CultureInfo("ar-SA")
                        };
                        options.SupportedUICultures = new List<CultureInfo>
                        {
                            new CultureInfo("ar-SA")
                        };
                        var provider = new CookieRequestCultureProvider();
                        provider.CookieName = "Preferences";
                        options.RequestCultureProviders.Insert(0, provider);
                    });
                    app.Run(context =>
                    {
                        var requestCultureFeature = context.Features.Get<IRequestCultureFeature>();
                        var requestCulture = requestCultureFeature.RequestCulture;
                        Assert.Equal("en-US", requestCulture.Culture.Name);
                        return Task.FromResult(0);
                    });
                });

            using (var server = new TestServer(builder))
            {
                var client = server.CreateClient();
                client.DefaultRequestHeaders.Add("Cookie", new CookieHeaderValue("Preferences", "uic=ar-SA").ToString());
                var response = await client.GetAsync(string.Empty);
            }
        }

        [Fact]
        public async void GetDefaultCultureInfoIfCookieDoesNotExist()
        {
            var builder = new WebApplicationBuilder()
                .Configure(app =>
                {
                    app.UseRequestLocalization(options =>
                    {
                        options.DefaultRequestCulture = new RequestCulture("en-US");
                        options.SupportedCultures = new List<CultureInfo>
                        {
                            new CultureInfo("ar-SA")
                        };
                        options.SupportedUICultures = new List<CultureInfo>
                        {
                            new CultureInfo("ar-SA")
                        };
                        var provider = new CookieRequestCultureProvider();
                        provider.CookieName = "Preferences";
                        options.RequestCultureProviders.Insert(0, provider);
                    });
                    app.Run(context =>
                    {
                        var requestCultureFeature = context.Features.Get<IRequestCultureFeature>();
                        var requestCulture = requestCultureFeature.RequestCulture;
                        Assert.Equal("en-US", requestCulture.Culture.Name);
                        return Task.FromResult(0);
                    });
                });

            using (var server = new TestServer(builder))
            {
                var client = server.CreateClient();
                var response = await client.GetAsync(string.Empty);
            }
        }
    }
}