btcpayserver/BTCPayServer.Tests/Checkoutv2Tests.cs

using System;
using System.Threading;
using System.Threading.Tasks;
using BTCPayServer.Client.Models;
using BTCPayServer.Payments;
using BTCPayServer.Tests.Logging;
using BTCPayServer.Views.Stores;
using NBitcoin;
using OpenQA.Selenium;
using OpenQA.Selenium.Support.Extensions;
using OpenQA.Selenium.Support.UI;
using Xunit;
using Xunit.Abstractions;

namespace BTCPayServer.Tests
{
    [Trait("Selenium", "Selenium")]
    [Collection(nameof(NonParallelizableCollectionDefinition))]
    public class CheckoutV2Tests : UnitTestBase
    {
        private const int TestTimeout = TestUtils.TestTimeout;

        public CheckoutV2Tests(ITestOutputHelper helper) : base(helper)
        {
        }

        [Fact(Timeout = TestTimeout)]
        [Trait("Lightning", "Lightning")]
        public async Task CanConfigureCheckout()
        {
            using var s = CreateSeleniumTester();
            s.Server.ActivateLightning();
            await s.StartAsync();
            s.GoToRegister();
            s.RegisterNewUser(true);
            s.CreateNewStore();
            s.AddLightningNode();
            // Use non-legacy derivation scheme
            s.AddDerivationScheme("BTC", "tpubDD79XF4pzhmPSJ9AyUay9YbXAeD1c6nkUqC32pnKARJH6Ja5hGUfGc76V82ahXpsKqN6UcSGXMkzR34aZq4W23C6DAdZFaVrzWqzj24F8BC");

            // Configure store url
            var storeUrl = "https://satoshisteaks.com/";
            s.GoToStore();
            s.Driver.FindElement(By.Id("StoreWebsite")).SendKeys(storeUrl);
            s.Driver.FindElement(By.Id("Save")).Click();
            Assert.Contains("Store successfully updated", s.FindAlertMessage().Text);

            s.GoToStore(StoreNavPages.CheckoutAppearance);
            s.Driver.WaitForAndClick(By.Id("Presets"));
            s.Driver.WaitForAndClick(By.Id("Presets_InStore"));
            Assert.True(s.Driver.SetCheckbox(By.Id("ShowPayInWalletButton"), true));
            s.Driver.FindElement(By.Id("Save")).SendKeys(Keys.Enter);
            Assert.Contains("Store successfully updated", s.FindAlertMessage().Text);

            // Top up/zero amount invoices
            var invoiceId = s.CreateInvoice(amount: null);
            s.GoToInvoiceCheckout(invoiceId);

            // Ensure we are seeing Checkout v2
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Equal(2, s.Driver.FindElements(By.CssSelector(".payment-method")).Count);
            Assert.Contains("Bitcoin", s.Driver.FindElement(By.CssSelector(".payment-method.active")).Text);
            Assert.Contains("LNURL", s.Driver.FindElement(By.CssSelector(".payment-method:nth-child(2)")).Text);
            var qrValue = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-qr-value");
            var address = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-clipboard");
            var payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
            var copyAddress = s.Driver.FindElement(By.Id("Address_BTC")).GetAttribute("value");
            Assert.Equal($"bitcoin:{address}", payUrl);
            Assert.StartsWith("bcrt", s.Driver.FindElement(By.Id("Address_BTC")).GetAttribute("value"));
            Assert.DoesNotContain("lightning=", payUrl);
            Assert.Equal(address, copyAddress);
            Assert.Equal($"bitcoin:{address.ToUpperInvariant()}", qrValue);
            s.Driver.ElementDoesNotExist(By.Id("Lightning_BTC"));

            // Details should show exchange rate
            s.Driver.ToggleCollapse("PaymentDetails");
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-TotalPrice"));
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-TotalFiat"));
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-AmountDue"));
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("sat/byte", s.Driver.FindElement(By.Id("PaymentDetails-RecommendedFee")).Text);

            // Switch to LNURL
            s.Driver.FindElement(By.CssSelector(".payment-method:nth-child(2)")).Click();
            TestUtils.Eventually(() =>
            {
                payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
                Assert.StartsWith("lightning:lnurl", payUrl);
                Assert.StartsWith("lnurl", s.Driver.WaitForElement(By.Id("Lightning_BTC")).GetAttribute("value"));
                s.Driver.ElementDoesNotExist(By.Id("Address_BTC"));
            });

            // Default payment method
            s.GoToHome();
            invoiceId = s.CreateInvoice(21000, "SATS", defaultPaymentMethod: "BTC_LightningLike");
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Equal(2, s.Driver.FindElements(By.CssSelector(".payment-method")).Count);
            Assert.Contains("Lightning", s.Driver.WaitForElement(By.CssSelector(".payment-method.active")).Text);
            Assert.Contains("Bitcoin", s.Driver.WaitForElement(By.CssSelector(".payment-method")).Text);
            qrValue = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-qr-value");
            address = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-clipboard");
            payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
            copyAddress = s.Driver.FindElement(By.Id("Lightning_BTC_LightningLike")).GetAttribute("value");
            Assert.Equal($"lightning:{address}", payUrl);
            Assert.Equal(address, copyAddress);
            Assert.Equal($"lightning:{address.ToUpperInvariant()}", qrValue);
            s.Driver.ElementDoesNotExist(By.Id("Address_BTC"));

            // Lightning amount in sats
            Assert.Contains("BTC", s.Driver.FindElement(By.Id("AmountDue")).Text);
            s.GoToHome();
            s.GoToLightningSettings();
            s.Driver.SetCheckbox(By.Id("LightningAmountInSatoshi"), true);
            s.Driver.FindElement(By.Id("save")).Click();
            Assert.Contains("BTC Lightning settings successfully updated", s.FindAlertMessage().Text);
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Contains("sats", s.Driver.FindElement(By.Id("AmountDue")).Text);

            // Details should not show exchange rate
            s.Driver.ToggleCollapse("PaymentDetails");
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-ExchangeRate"));
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-TotalFiat"));
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-RecommendedFee"));
            Assert.Contains("21 000 sats", s.Driver.FindElement(By.Id("PaymentDetails-AmountDue")).Text);
            Assert.Contains("21 000 sats", s.Driver.FindElement(By.Id("PaymentDetails-TotalPrice")).Text);

            // Expire
            var expirySeconds = s.Driver.FindElement(By.Id("ExpirySeconds"));
            expirySeconds.Clear();
            expirySeconds.SendKeys("3");
            s.Driver.FindElement(By.Id("Expire")).Click();

            var paymentInfo = s.Driver.WaitForElement(By.Id("PaymentInfo"));
            Assert.Contains("This invoice will expire in", paymentInfo.Text);
            Assert.DoesNotContain("Please send", paymentInfo.Text);
            TestUtils.Eventually(() =>
            {
                var expiredSection = s.Driver.FindElement(By.Id("unpaid"));
                Assert.True(expiredSection.Displayed);
                Assert.Contains("Invoice Expired", expiredSection.Text);
            });
            Assert.True(s.Driver.ElementDoesNotExist(By.Id("receipt-btn")));
            Assert.Equal(storeUrl, s.Driver.FindElement(By.Id("StoreLink")).GetAttribute("href"));

            // Test payment
            s.GoToHome();
            invoiceId = s.CreateInvoice();
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));

            // Details
            s.Driver.ToggleCollapse("PaymentDetails");
            var details = s.Driver.FindElement(By.CssSelector(".payment-details"));
            Assert.Contains("Total Price", details.Text);
            Assert.Contains("Total Fiat", details.Text);
            Assert.Contains("Exchange Rate", details.Text);
            Assert.Contains("Amount Due", details.Text);
            Assert.Contains("Recommended Fee", details.Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-TotalFiat")).Text);
            Assert.Contains("BTC", s.Driver.FindElement(By.Id("PaymentDetails-AmountDue")).Text);
            Assert.Contains("BTC", s.Driver.FindElement(By.Id("PaymentDetails-TotalPrice")).Text);

            // Pay partial amount
            await Task.Delay(200);
            address = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-clipboard");
            var amountFraction = "0.00001";
            await s.Server.ExplorerNode.SendToAddressAsync(BitcoinAddress.Create(address, Network.RegTest),
                Money.Parse(amountFraction));
            await s.Server.ExplorerNode.GenerateAsync(1);

            // Fake Pay
            TestUtils.Eventually(() =>
            {
                paymentInfo = s.Driver.WaitForElement(By.Id("PaymentInfo"));
                Assert.Contains("The invoice hasn't been paid in full", paymentInfo.Text);
                Assert.Contains("Please send", paymentInfo.Text);
            });

            s.Driver.Navigate().Refresh();

            // Pay full amount
            s.PayInvoice();

            // Processing
            TestUtils.Eventually(() =>
            {
                var processingSection = s.Driver.WaitForElement(By.Id("processing"));
                Assert.True(processingSection.Displayed);
                Assert.Contains("Payment Received", processingSection.Text);
                Assert.Contains("Your payment has been received and is now processing", processingSection.Text);
            });
            s.Driver.FindElement(By.Id("confetti"));

            // Mine
            s.MineBlockOnInvoiceCheckout();
            TestUtils.Eventually(() =>
            {
                Assert.Contains("Mined 1 block",
                    s.Driver.WaitForElement(By.Id("CheatSuccessMessage")).Text);
            });

            // Settled
            TestUtils.Eventually(() =>
            {
                var settledSection = s.Driver.WaitForElement(By.Id("settled"));
                Assert.True(settledSection.Displayed);
                Assert.Contains("Invoice Paid", settledSection.Text);
            });
            s.Driver.FindElement(By.Id("confetti"));
            s.Driver.FindElement(By.Id("receipt-btn"));
            Assert.Equal(storeUrl, s.Driver.FindElement(By.Id("StoreLink")).GetAttribute("href"));

            // BIP21
            s.GoToHome();
            s.GoToStore(StoreNavPages.CheckoutAppearance);
            s.Driver.SetCheckbox(By.Id("OnChainWithLnInvoiceFallback"), true);
            s.Driver.SetCheckbox(By.Id("LightningAmountInSatoshi"), false);
            s.Driver.FindElement(By.Id("Save")).Click();
            Assert.Contains("Store successfully updated", s.FindAlertMessage().Text);

            invoiceId = s.CreateInvoice();
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Empty(s.Driver.FindElements(By.CssSelector(".payment-method")));
            Assert.Contains("BTC", s.Driver.FindElement(By.Id("AmountDue")).Text);
            qrValue = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-qr-value");
            address = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-clipboard");
            payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
            var copyAddressOnchain = s.Driver.FindElement(By.Id("Address_BTC")).GetAttribute("value");
            var copyAddressLightning = s.Driver.FindElement(By.Id("Lightning_BTC")).GetAttribute("value");
            Assert.StartsWith($"bitcoin:{address}?amount=", payUrl);
            Assert.Contains("?amount=", payUrl);
            Assert.Contains("&lightning=", payUrl);
            Assert.StartsWith("bcrt", copyAddressOnchain);
            Assert.Equal(address, copyAddressOnchain);
            Assert.StartsWith("lnbcrt", copyAddressLightning);
            Assert.StartsWith($"bitcoin:{address.ToUpperInvariant()}?amount=", qrValue);
            Assert.Contains("&lightning=LNBCRT", qrValue);

            // Check details
            s.Driver.ToggleCollapse("PaymentDetails");
            Assert.Contains("1 BTC = ", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-TotalFiat")).Text);
            Assert.Contains("BTC", s.Driver.FindElement(By.Id("PaymentDetails-AmountDue")).Text);
            Assert.Contains("BTC", s.Driver.FindElement(By.Id("PaymentDetails-TotalPrice")).Text);

            // Switch to amount displayed in sats
            s.GoToHome();
            s.GoToStore(StoreNavPages.CheckoutAppearance);
            s.Driver.SetCheckbox(By.Id("LightningAmountInSatoshi"), true);
            s.Driver.FindElement(By.Id("Save")).Click();
            Assert.Contains("Store successfully updated", s.FindAlertMessage().Text);
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Contains("sats", s.Driver.FindElement(By.Id("AmountDue")).Text);

            // Check details
            s.Driver.ToggleCollapse("PaymentDetails");
            Assert.Contains("1 sat = ", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-TotalFiat")).Text);
            Assert.Contains("sats", s.Driver.FindElement(By.Id("PaymentDetails-AmountDue")).Text);
            Assert.Contains("sats", s.Driver.FindElement(By.Id("PaymentDetails-TotalPrice")).Text);

            // BIP21 with LN as default payment method
            s.GoToHome();
            invoiceId = s.CreateInvoice(defaultPaymentMethod: "BTC_LightningLike");
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Empty(s.Driver.FindElements(By.CssSelector(".payment-method")));
            payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
            Assert.StartsWith("bitcoin:", payUrl);
            Assert.Contains("&lightning=lnbcrt", payUrl);

            // Check details
            s.Driver.ToggleCollapse("PaymentDetails");
            Assert.Contains("1 sat = ", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-TotalFiat")).Text);
            Assert.Contains("sats", s.Driver.FindElement(By.Id("PaymentDetails-AmountDue")).Text);
            Assert.Contains("sats", s.Driver.FindElement(By.Id("PaymentDetails-TotalPrice")).Text);

            // Ensure LNURL is enabled
            s.GoToHome();
            s.GoToLightningSettings();
            Assert.True(s.Driver.FindElement(By.Id("LNURLEnabled")).Selected);

            // BIP21 with top-up invoice
            invoiceId = s.CreateInvoice(amount: null);
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Empty(s.Driver.FindElements(By.CssSelector(".payment-method")));
            qrValue = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-qr-value");
            address = s.Driver.FindElement(By.CssSelector(".qr-container")).GetAttribute("data-clipboard");
            payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
            copyAddressOnchain = s.Driver.FindElement(By.Id("Address_BTC")).GetAttribute("value");
            copyAddressLightning = s.Driver.FindElement(By.Id("Lightning_BTC")).GetAttribute("value");
            Assert.StartsWith($"bitcoin:{address}", payUrl);
            Assert.Contains("?lightning=lnurl", payUrl);
            Assert.DoesNotContain("amount=", payUrl);
            Assert.StartsWith("bcrt", copyAddressOnchain);
            Assert.Equal(address, copyAddressOnchain);
            Assert.StartsWith("lnurl", copyAddressLightning);
            Assert.StartsWith($"bitcoin:{address.ToUpperInvariant()}?lightning=LNURL", qrValue);

            // Check details
            s.Driver.ToggleCollapse("PaymentDetails");
            Assert.Contains("1 sat = ", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            Assert.Contains("$", s.Driver.FindElement(By.Id("PaymentDetails-ExchangeRate")).Text);
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-TotalFiat"));
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-AmountDue"));
            s.Driver.ElementDoesNotExist(By.Id("PaymentDetails-TotalPrice"));

            // Expiry message should not show amount for top-up invoice
            expirySeconds = s.Driver.FindElement(By.Id("ExpirySeconds"));
            expirySeconds.Clear();
            expirySeconds.SendKeys("5");
            s.Driver.FindElement(By.Id("Expire")).Click();

            paymentInfo = s.Driver.WaitForElement(By.Id("PaymentInfo"));
            Assert.Contains("This invoice will expire in", paymentInfo.Text);
            Assert.Contains("00:0", paymentInfo.Text);
            Assert.DoesNotContain("Please send", paymentInfo.Text);

            // Configure countdown timer
            s.GoToHome();
            invoiceId = s.CreateInvoice();
            s.GoToHome();
            s.GoToStore(StoreNavPages.CheckoutAppearance);
            var displayExpirationTimer = s.Driver.FindElement(By.Id("DisplayExpirationTimer"));
            Assert.Equal("5", displayExpirationTimer.GetAttribute("value"));
            displayExpirationTimer.Clear();
            displayExpirationTimer.SendKeys("10");
            s.Driver.FindElement(By.Id("Save")).Click();
            Assert.Contains("Store successfully updated", s.FindAlertMessage().Text);

            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            paymentInfo = s.Driver.FindElement(By.Id("PaymentInfo"));
            Assert.False(paymentInfo.Displayed);
            Assert.DoesNotContain("This invoice will expire in", paymentInfo.Text);

            expirySeconds = s.Driver.FindElement(By.Id("ExpirySeconds"));
            expirySeconds.Clear();
            expirySeconds.SendKeys("599");
            s.Driver.FindElement(By.Id("Expire")).Click();

            paymentInfo = s.Driver.WaitForElement(By.Id("PaymentInfo"));
            Assert.True(paymentInfo.Displayed);
            Assert.Contains("This invoice will expire in", paymentInfo.Text);
            Assert.Contains("09:5", paymentInfo.Text);

            // Disable LNURL again
            s.GoToHome();
            s.GoToLightningSettings();
            s.Driver.SetCheckbox(By.Id("LNURLEnabled"), false);
            s.Driver.FindElement(By.Id("save")).Click();
            Assert.Contains("BTC Lightning settings successfully updated", s.FindAlertMessage().Text);

            // Test:
            // - NFC/LNURL-W available with just Lightning
            // - BIP21 works correctly even though Lightning is default payment method
            s.GoToHome();
            invoiceId = s.CreateInvoice(defaultPaymentMethod: "BTC_LightningLike");
            s.GoToInvoiceCheckout(invoiceId);
            s.Driver.WaitUntilAvailable(By.Id("Checkout-v2"));
            Assert.Empty(s.Driver.FindElements(By.CssSelector(".payment-method")));
            payUrl = s.Driver.FindElement(By.Id("PayInWallet")).GetAttribute("href");
            Assert.StartsWith("bitcoin:", payUrl);
            Assert.Contains("&lightning=lnbcrt", payUrl);

            // Language Switch
            var languageSelect = new SelectElement(s.Driver.FindElement(By.Id("DefaultLang")));
            Assert.Equal("English", languageSelect.SelectedOption.Text);
            Assert.Equal("View Details", s.Driver.FindElement(By.Id("DetailsToggle")).Text);
            Assert.DoesNotContain("lang=", s.Driver.Url);
            languageSelect.SelectByText("Deutsch");
            Assert.Equal("Details anzeigen", s.Driver.FindElement(By.Id("DetailsToggle")).Text);
            Assert.Contains("lang=de", s.Driver.Url);

            s.Driver.Navigate().Refresh();
            languageSelect = new SelectElement(s.Driver.WaitForElement(By.Id("DefaultLang")));
            Assert.Equal("Deutsch", languageSelect.SelectedOption.Text);
            Assert.Equal("Details anzeigen", s.Driver.FindElement(By.Id("DetailsToggle")).Text);
            languageSelect.SelectByText("English");
            Assert.Equal("View Details", s.Driver.FindElement(By.Id("DetailsToggle")).Text);
            Assert.Contains("lang=en", s.Driver.Url);
        }

        [Fact(Timeout = TestTimeout)]
        public async Task CanUseCheckoutAsModal()
        {
            using var s = CreateSeleniumTester();
            await s.StartAsync();
            s.GoToRegister();
            s.RegisterNewUser();
            s.CreateNewStore();
            s.GoToStore();
            s.AddDerivationScheme();
            var invoiceId = s.CreateInvoice(0.001m, "BTC", "a@x.com");
            var invoice = await s.Server.PayTester.InvoiceRepository.GetInvoice(invoiceId);
            s.Driver.Navigate()
                .GoToUrl(new Uri(s.ServerUri, $"tests/index.html?invoice={invoiceId}"));
            s.Driver.WaitUntilAvailable(By.Name("btcpay"));

            var frameElement = s.Driver.FindElement(By.Name("btcpay"));
            Assert.True(frameElement.Displayed);
            var iframe = s.Driver.SwitchTo().Frame(frameElement);
            iframe.WaitUntilAvailable(By.Id("Checkout-v2"));

            await s.Server.ExplorerNode.SendToAddressAsync(BitcoinAddress.Create(invoice
                    .GetPaymentMethod(new PaymentMethodId("BTC", PaymentTypes.BTCLike))
                    .GetPaymentMethodDetails().GetPaymentDestination(), Network.RegTest),
                new Money(0.001m, MoneyUnit.BTC));

            IWebElement closebutton = null;
            TestUtils.Eventually(() =>
            {
                closebutton = iframe.FindElement(By.Id("close"));
                Assert.True(closebutton.Displayed);
            });
            closebutton.Click();
            s.Driver.AssertElementNotFound(By.Name("btcpay"));
            Assert.Equal(s.Driver.Url,
                new Uri(s.ServerUri, $"tests/index.html?invoice={invoiceId}").ToString());
        }
    }
}