// 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.Collections.Generic;
using System.Configuration.Assemblies;
using System.Linq;
using System.Numerics;
using BasicTestApp;
using BasicTestApp.HierarchicalImportsTest.Subdir;
using Microsoft.AspNetCore.Blazor.E2ETest.Infrastructure;
using Microsoft.AspNetCore.Blazor.E2ETest.Infrastructure.ServerFixtures;
using OpenQA.Selenium;
using OpenQA.Selenium.Support.UI;
using Xunit;
using Xunit.Abstractions;

namespace Microsoft.AspNetCore.Blazor.E2ETest.Tests
{
    public class ComponentRenderingTest : BasicTestAppTestBase
    {
        public ComponentRenderingTest(
            BrowserFixture browserFixture,
            ToggleExecutionModeServerFixture<Program> serverFixture,
            ITestOutputHelper output)
            : base(browserFixture, serverFixture, output)
        {
            Navigate(ServerPathBase, noReload: !serverFixture.UsingAspNetHost);
        }

        [Fact]
        public void BasicTestAppCanBeServed()
        {
            Assert.Equal("Basic test app", Browser.Title);
        }

        [Fact]
        public void CanRenderTextOnlyComponent()
        {
            var appElement = MountTestComponent<TextOnlyComponent>();
            Assert.Equal("Hello from TextOnlyComponent", appElement.Text);
        }

        // This verifies that we've correctly configured the Razor language version via MSBuild.
        // See #974
        [Fact]
        public void CanRenderComponentWithDataDash()
        {
            var appElement = MountTestComponent<DataDashComponent>();
            var element = appElement.FindElement(By.Id("cool_beans"));
            Assert.Equal("17", element.GetAttribute("data-tab"));
            Assert.Equal("17", element.Text);
        }

        [Fact]
        public void CanRenderComponentWithAttributes()
        {
            var appElement = MountTestComponent<RedTextComponent>();
            var styledElement = appElement.FindElement(By.TagName("h1"));
            Assert.Equal("Hello, world!", styledElement.Text);
            Assert.Equal("color: red;", styledElement.GetAttribute("style"));
            Assert.Equal("somevalue", styledElement.GetAttribute("customattribute"));
        }

        [Fact]
        public void CanTriggerEvents()
        {
            // Initial count is zero
            var appElement = MountTestComponent<CounterComponent>();
            var countDisplayElement = appElement.FindElement(By.TagName("p"));
            Assert.Equal("Current count: 0", countDisplayElement.Text);

            // Clicking button increments count
            appElement.FindElement(By.TagName("button")).Click();
            WaitAssert.Equal("Current count: 1", () => countDisplayElement.Text);
        }

        [Fact]
        public void CanTriggerAsyncEventHandlers()
        {
            // Initial state is stopped
            var appElement = MountTestComponent<AsyncEventHandlerComponent>();
            var stateElement = appElement.FindElement(By.Id("state"));
            Assert.Equal("Stopped", stateElement.Text);

            // Clicking 'tick' changes the state, and starts a task
            appElement.FindElement(By.Id("tick")).Click();
            WaitAssert.Equal("Started", () => stateElement.Text);

            // Clicking 'tock' completes the task, which updates the state
            appElement.FindElement(By.Id("tock")).Click();
            WaitAssert.Equal("Stopped", () => stateElement.Text);
        }

        [Fact]
        public void CanTriggerKeyPressEvents()
        {
            // List is initially empty
            var appElement = MountTestComponent<KeyPressEventComponent>();
            var inputElement = appElement.FindElement(By.TagName("input"));
            Func<IEnumerable<IWebElement>> liElements =
                () => appElement.FindElements(By.TagName("li"));
            Assert.Empty(liElements());

            // Typing adds element
            inputElement.SendKeys("a");
            WaitAssert.Collection(liElements,
                li => Assert.Equal("a", li.Text));

            // Typing again adds another element
            inputElement.SendKeys("b");
            WaitAssert.Collection(liElements,
                li => Assert.Equal("a", li.Text),
                li => Assert.Equal("b", li.Text));

            // Textbox contains typed text
            Assert.Equal("ab", inputElement.GetAttribute("value"));
        }

        [Fact]
        public void CanAddAndRemoveEventHandlersDynamically()
        {
            var appElement = MountTestComponent<CounterComponent>();
            var countDisplayElement = appElement.FindElement(By.TagName("p"));
            var incrementButton = appElement.FindElement(By.TagName("button"));
            var toggleClickHandlerCheckbox = appElement.FindElement(By.CssSelector("[type=checkbox]"));

            // Initial count is zero; clicking button increments count
            Assert.Equal("Current count: 0", countDisplayElement.Text);
            incrementButton.Click();
            WaitAssert.Equal("Current count: 1", () => countDisplayElement.Text);

            // We can remove an event handler
            toggleClickHandlerCheckbox.Click();
            WaitAssert.Empty(() => appElement.FindElements(By.Id("listening-message")));
            incrementButton.Click();
            WaitAssert.Equal("Current count: 1", () => countDisplayElement.Text);

            // We can add an event handler
            toggleClickHandlerCheckbox.Click();
            appElement.FindElement(By.Id("listening-message"));
            incrementButton.Click();
            WaitAssert.Equal("Current count: 2", () => countDisplayElement.Text);
        }

        [Fact]
        public void CanRenderChildComponents()
        {
            var appElement = MountTestComponent<ParentChildComponent>();
            Assert.Equal("Parent component",
                appElement.FindElement(By.CssSelector("fieldset > legend")).Text);

            var styledElement = appElement.FindElement(By.CssSelector("fieldset > h1"));
            Assert.Equal("Hello, world!", styledElement.Text);
            Assert.Equal("color: red;", styledElement.GetAttribute("style"));
            Assert.Equal("somevalue", styledElement.GetAttribute("customattribute"));
        }

        // Verifies we can render HTML content as a single block
        [Fact]
        public void CanRenderChildContent_StaticHtmlBlock()
        {
            var appElement = MountTestComponent<HtmlBlockChildContent>();
            Assert.Equal("<p>Some-Static-Text</p>",
                appElement.FindElement(By.Id("foo")).GetAttribute("innerHTML"));
        }

        // Verifies we can rewite more complex HTML content into blocks
        [Fact]
        public void CanRenderChildContent_MixedHtmlBlock()
        {
            var appElement = MountTestComponent<HtmlMixedChildContent>();

            var one = appElement.FindElement(By.Id("one"));
            Assert.Equal("<p>Some-Static-Text</p>", one.GetAttribute("innerHTML"));

            var two = appElement.FindElement(By.Id("two"));
            Assert.Equal("<span>More-Static-Text</span>", two.GetAttribute("innerHTML"));

            var three = appElement.FindElement(By.Id("three"));
            Assert.Equal("Some-Dynamic-Text", three.GetAttribute("innerHTML"));

            var four = appElement.FindElement(By.Id("four"));
            Assert.Equal("But this is static", four.GetAttribute("innerHTML"));
        }

        // Verifies we can rewrite HTML blocks with encoded HTML
        [Fact]
        public void CanRenderChildContent_EncodedHtmlInBlock()
        {
            var appElement = MountTestComponent<HtmlEncodedChildContent>();

            var one = appElement.FindElement(By.Id("one"));
            Assert.Equal("<p>Some-Static-Text</p>", one.GetAttribute("innerHTML"));

            var two = appElement.FindElement(By.Id("two"));
            Assert.Equal("&lt;span&gt;More-Static-Text&lt;/span&gt;", two.GetAttribute("innerHTML"));

            var three = appElement.FindElement(By.Id("three"));
            Assert.Equal("Some-Dynamic-Text", three.GetAttribute("innerHTML"));

            var four = appElement.FindElement(By.Id("four"));
            Assert.Equal("But this is static", four.GetAttribute("innerHTML"));
        }

        [Fact]
        public void CanTriggerEventsOnChildComponents()
        {
            // Counter is displayed as child component. Initial count is zero.
            var appElement = MountTestComponent<CounterComponentWrapper>();
            var counterDisplay = appElement
                .FindElements(By.TagName("p"))
                .Single(element => element.Text == "Current count: 0");

            // Clicking increments count in child component
            appElement.FindElement(By.TagName("button")).Click();
            WaitAssert.Equal("Current count: 1", () => counterDisplay.Text);
        }

        [Fact]
        public void ChildComponentsRerenderWhenPropertiesChanged()
        {
            // Count value is displayed in child component with initial value zero
            var appElement = MountTestComponent<CounterComponentUsingChild>();
            var wholeCounterElement = appElement.FindElement(By.TagName("p"));
            var messageElementInChild = wholeCounterElement.FindElement(By.ClassName("message"));
            Assert.Equal("Current count: 0", wholeCounterElement.Text);
            Assert.Equal("0", messageElementInChild.Text);

            // Clicking increments count in child element
            appElement.FindElement(By.TagName("button")).Click();
            WaitAssert.Equal("1", () => messageElementInChild.Text);
        }

        [Fact]
        public void CanAddAndRemoveChildComponentsDynamically()
        {
            // Initially there are zero child components
            var appElement = MountTestComponent<AddRemoveChildComponents>();
            var addButton = appElement.FindElement(By.ClassName("addChild"));
            var removeButton = appElement.FindElement(By.ClassName("removeChild"));
            Func<IEnumerable<IWebElement>> childComponentWrappers = () => appElement.FindElements(By.TagName("p"));
            Assert.Empty(childComponentWrappers());

            // Click to add/remove some child components
            addButton.Click();
            WaitAssert.Collection(childComponentWrappers,
                elem => Assert.Equal("Child 1", elem.FindElement(By.ClassName("message")).Text));

            addButton.Click();
            WaitAssert.Collection(childComponentWrappers,
                elem => Assert.Equal("Child 1", elem.FindElement(By.ClassName("message")).Text),
                elem => Assert.Equal("Child 2", elem.FindElement(By.ClassName("message")).Text));

            removeButton.Click();
            WaitAssert.Collection(childComponentWrappers,
                elem => Assert.Equal("Child 1", elem.FindElement(By.ClassName("message")).Text));

            addButton.Click();
            WaitAssert.Collection(childComponentWrappers,
                elem => Assert.Equal("Child 1", elem.FindElement(By.ClassName("message")).Text),
                elem => Assert.Equal("Child 3", elem.FindElement(By.ClassName("message")).Text));
        }

        [Fact]
        public void ChildComponentsNotifiedWhenPropertiesChanged()
        {
            // Child component receives notification that lets it compute a property before first render
            var appElement = MountTestComponent<PropertiesChangedHandlerParent>();
            var suppliedValueElement = appElement.FindElement(By.ClassName("supplied"));
            var computedValueElement = appElement.FindElement(By.ClassName("computed"));
            var incrementButton = appElement.FindElement(By.TagName("button"));
            Assert.Equal("You supplied: 100", suppliedValueElement.Text);
            Assert.Equal("I computed: 200", computedValueElement.Text);

            // When property changes, child is renotified before rerender
            incrementButton.Click();
            WaitAssert.Equal("You supplied: 101", () => suppliedValueElement.Text);
            Assert.Equal("I computed: 202", computedValueElement.Text);
        }

        [Fact]
        public void CanRenderFragmentsWhilePreservingSurroundingElements()
        {
            // Initially, the region isn't shown
            var appElement = MountTestComponent<RenderFragmentToggler>();
            var originalButton = appElement.FindElement(By.TagName("button"));
            Func<IEnumerable<IWebElement>> fragmentElements = () => appElement.FindElements(By.CssSelector("p[name=fragment-element]"));
            Assert.Empty(fragmentElements());

            // The JS-side DOM builder handles regions correctly, placing elements
            // after the region after the corresponding elements
            Assert.Equal("The end", appElement.FindElements(By.CssSelector("div > *:last-child")).Single().Text);

            // When we click the button, the region is shown
            originalButton.Click();
            WaitAssert.Single(fragmentElements);

            // The button itself was preserved, so we can click it again and see the effect
            originalButton.Click();
            WaitAssert.Empty(fragmentElements);
        }

        [Fact]
        public void CanUseViewImportsHierarchically()
        {
            // The component is able to compile and output these type names only because
            // of the _ViewImports.cshtml files at the same and ancestor levels
            var appElement = MountTestComponent<ComponentUsingImports>();
            Assert.Collection(appElement.FindElements(By.TagName("p")),
                elem => Assert.Equal(typeof(Complex).FullName, elem.Text),
                elem => Assert.Equal(typeof(AssemblyHashAlgorithm).FullName, elem.Text));
        }

        [Fact]
        public void CanUseComponentAndStaticContentFromExternalNuGetPackage()
        {
            var appElement = MountTestComponent<ExternalContentPackage>();

            // NuGet packages can use Blazor's JS interop features to provide
            // .NET code access to browser APIs
            var showPromptButton = appElement.FindElements(By.TagName("button")).First();
            showPromptButton.Click();
            
            var modal = new WebDriverWait(Browser, TimeSpan.FromSeconds(3))
                .Until(SwitchToAlert);
            modal.SendKeys("Some value from test");
            modal.Accept();
            var promptResult = appElement.FindElement(By.TagName("strong"));
            WaitAssert.Equal("Some value from test", () => promptResult.Text);

            // NuGet packages can also embed entire Blazor components (themselves
            // authored as Razor files), including static content. The CSS value
            // here is in a .css file, so if it's correct we know that static content
            // file was loaded.
            var specialStyleDiv = appElement.FindElement(By.ClassName("special-style"));
            Assert.Equal("50px", specialStyleDiv.GetCssValue("padding"));

            // The external Blazor components are fully functional, not just static HTML
            var externalComponentButton = specialStyleDiv.FindElement(By.TagName("button"));
            Assert.Equal("Click me", externalComponentButton.Text);
            externalComponentButton.Click();
            WaitAssert.Equal("It works", () => externalComponentButton.Text);
        }

        [Fact]
        public void CanRenderSvgWithCorrectNamespace()
        {
            var appElement = MountTestComponent<SvgComponent>();

            var svgElement = appElement.FindElement(By.XPath("//*[local-name()='svg' and namespace-uri()='http://www.w3.org/2000/svg']"));
            Assert.NotNull(svgElement);

            var svgCircleElement = appElement.FindElement(By.XPath("//*[local-name()='circle' and namespace-uri()='http://www.w3.org/2000/svg']"));
            Assert.NotNull(svgCircleElement);
            Assert.Equal("10", svgCircleElement.GetAttribute("r"));

            appElement.FindElement(By.TagName("button")).Click();
            WaitAssert.Equal("20", () => svgCircleElement.GetAttribute("r"));
        }

        [Fact]
        public void CanRenderSvgChildComponentWithCorrectNamespace()
        {
            var appElement = MountTestComponent<SvgWithChildComponent>();

            var svgElement = appElement.FindElement(By.XPath("//*[local-name()='svg' and namespace-uri()='http://www.w3.org/2000/svg']"));
            Assert.NotNull(svgElement);

            var svgCircleElement = appElement.FindElement(By.XPath("//*[local-name()='circle' and namespace-uri()='http://www.w3.org/2000/svg']"));
            Assert.NotNull(svgCircleElement);
        }

        [Fact]
        public void LogicalElementInsertionWorksHierarchically()
        {
            var appElement = MountTestComponent<LogicalElementInsertionCases>();
            WaitAssert.Equal("First Second Third", () => appElement.Text);
        }

        [Fact]
        public void CanUseJsInteropToReferenceElements()
        {
            var appElement = MountTestComponent<ElementRefComponent>();
            var inputElement = appElement.FindElement(By.Id("capturedElement"));
            var buttonElement = appElement.FindElement(By.TagName("button"));

            Assert.Equal(string.Empty, inputElement.GetAttribute("value"));

            buttonElement.Click();
            WaitAssert.Equal("Clicks: 1", () => inputElement.GetAttribute("value"));
            buttonElement.Click();
            WaitAssert.Equal("Clicks: 2", () => inputElement.GetAttribute("value"));
        }

        [Fact]
        public void CanCaptureReferencesToDynamicallyAddedElements()
        {
            var appElement = MountTestComponent<ElementRefComponent>();
            var buttonElement = appElement.FindElement(By.TagName("button"));
            var checkbox = appElement.FindElement(By.CssSelector("input[type=checkbox]"));

            // We're going to remove the input. But first, put in some contents
            // so we can observe it's not the same instance later
            appElement.FindElement(By.Id("capturedElement")).SendKeys("some text");

            // Remove the captured element
            checkbox.Click();
            WaitAssert.Empty(() => appElement.FindElements(By.Id("capturedElement")));

            // Re-add it; observe it starts empty again
            checkbox.Click();
            var inputElement = appElement.FindElement(By.Id("capturedElement"));
            Assert.Equal(string.Empty, inputElement.GetAttribute("value"));

            // See that the capture variable was automatically updated to reference the new instance
            buttonElement.Click();
            WaitAssert.Equal("Clicks: 1", () => inputElement.GetAttribute("value"));
        }

        [Fact]
        public void CanCaptureReferencesToDynamicallyAddedComponents()
        {
            var appElement = MountTestComponent<ComponentRefComponent>();
            var incrementButtonSelector = By.CssSelector("#child-component button");
            var currentCountTextSelector = By.CssSelector("#child-component p:first-of-type");
            var resetButton = appElement.FindElement(By.Id("reset-child"));
            var toggleChildCheckbox = appElement.FindElement(By.Id("toggle-child"));
            Func<string> currentCountText = () => appElement.FindElement(currentCountTextSelector).Text;

            // Verify the reference was captured initially
            appElement.FindElement(incrementButtonSelector).Click();
            WaitAssert.Equal("Current count: 1", currentCountText);
            resetButton.Click();
            WaitAssert.Equal("Current count: 0", currentCountText);
            appElement.FindElement(incrementButtonSelector).Click();
            WaitAssert.Equal("Current count: 1", currentCountText);

            // Remove and re-add a new instance of the child, checking the text was reset
            toggleChildCheckbox.Click();
            WaitAssert.Empty(() => appElement.FindElements(incrementButtonSelector));
            toggleChildCheckbox.Click();
            WaitAssert.Equal("Current count: 0", currentCountText);

            // Verify we have a new working reference
            appElement.FindElement(incrementButtonSelector).Click();
            WaitAssert.Equal("Current count: 1", currentCountText);
            resetButton.Click();
            WaitAssert.Equal("Current count: 0", currentCountText);
        }

        [Fact]
        public void CanUseJsInteropForRefElementsDuringOnAfterRender()
        {
            var appElement = MountTestComponent<AfterRenderInteropComponent>();
            var inputElement = appElement.FindElement(By.TagName("input"));
            Assert.Equal("Value set after render", inputElement.GetAttribute("value"));
        }

        [Fact]
        public void CanRenderMarkupBlocks()
        {
            var appElement = MountTestComponent<MarkupBlockComponent>();

            // Static markup
            Assert.Equal(
                "attributes",
                appElement.FindElement(By.CssSelector("p span#attribute-example")).Text);

            // Dynamic markup (from a custom RenderFragment)
            Assert.Equal(
                "[Here is an example. We support multiple-top-level nodes.]",
                appElement.FindElement(By.Id("dynamic-markup-block")).Text);
            Assert.Equal(
                "example",
                appElement.FindElement(By.CssSelector("#dynamic-markup-block strong#dynamic-element em")).Text);

            // Dynamic markup (from a MarkupString)
            Assert.Equal(
                "This is a markup string.",
                appElement.FindElement(By.ClassName("markup-string-value")).Text);
            Assert.Equal(
                "markup string",
                appElement.FindElement(By.CssSelector(".markup-string-value em")).Text);

            // Updating markup blocks
            appElement.FindElement(By.TagName("button")).Click();
            WaitAssert.Equal(
                "[The output was changed completely.]",
                () => appElement.FindElement(By.Id("dynamic-markup-block")).Text);
            Assert.Equal(
                "changed",
                appElement.FindElement(By.CssSelector("#dynamic-markup-block span em")).Text);
        }

        [Fact]
        public void CanRenderRazorTemplates()
        {
            var appElement = MountTestComponent<RazorTemplates>();

            // code block template (component parameter)
            var element = appElement.FindElement(By.CssSelector("div#codeblocktemplate ol"));
            Assert.Collection(
                element.FindElements(By.TagName("li")),
                e => Assert.Equal("#1 - a", e.Text),
                e => Assert.Equal("#2 - b", e.Text),
                e => Assert.Equal("#3 - c", e.Text));
        }

        [Fact]
        public void CanRenderMultipleChildContent()
        {
            var appElement = MountTestComponent<MultipleChildContent>();

            var table = appElement.FindElement(By.TagName("table"));

            var thead = table.FindElement(By.TagName("thead"));
            Assert.Collection(
                thead.FindElements(By.TagName("th")),
                e => Assert.Equal("Col1", e.Text),
                e => Assert.Equal("Col2", e.Text),
                e => Assert.Equal("Col3", e.Text));

            var tfoot = table.FindElement(By.TagName("tfoot"));
            Assert.Empty(tfoot.FindElements(By.TagName("td")));

            var toggle = appElement.FindElement(By.Id("toggle"));
            toggle.Click();

            WaitAssert.Collection(
                () => tfoot.FindElements(By.TagName("td")),
                e => Assert.Equal("The", e.Text),
                e => Assert.Equal("", e.Text),
                e => Assert.Equal("End", e.Text));
        }

        static IAlert SwitchToAlert(IWebDriver driver)
        {
            try
            {
                return driver.SwitchTo().Alert();
            }
            catch (NoAlertPresentException)
            {
                return null;
            }
        }
    }
}