mirror of
https://github.com/btcpayserver/btcpayserver.git
synced 2024-11-19 09:54:30 +01:00
526 lines
21 KiB
C#
526 lines
21 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Net.Http;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using BTCPayServer.Client;
|
|
using BTCPayServer.Client.Models;
|
|
using BTCPayServer.Controllers;
|
|
using BTCPayServer.Data;
|
|
using BTCPayServer.Events;
|
|
using BTCPayServer.Lightning;
|
|
using BTCPayServer.Lightning.CLightning;
|
|
using BTCPayServer.Models.AccountViewModels;
|
|
using BTCPayServer.Models.StoreViewModels;
|
|
using BTCPayServer.Services;
|
|
using BTCPayServer.Services.Stores;
|
|
using BTCPayServer.Services.Wallets;
|
|
using BTCPayServer.Tests.Logging;
|
|
using Microsoft.AspNetCore.Identity;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.CodeAnalysis.Operations;
|
|
using NBitcoin;
|
|
using NBitcoin.Payment;
|
|
using NBitpayClient;
|
|
using NBXplorer.DerivationStrategy;
|
|
using NBXplorer.Models;
|
|
using Newtonsoft.Json;
|
|
using Newtonsoft.Json.Linq;
|
|
using Xunit;
|
|
using Xunit.Sdk;
|
|
|
|
namespace BTCPayServer.Tests
|
|
{
|
|
public class TestAccount
|
|
{
|
|
readonly ServerTester parent;
|
|
|
|
public TestAccount(ServerTester parent)
|
|
{
|
|
this.parent = parent;
|
|
BitPay = new Bitpay(new Key(), parent.PayTester.ServerUri);
|
|
}
|
|
|
|
public void GrantAccess(bool isAdmin = false)
|
|
{
|
|
GrantAccessAsync(isAdmin).GetAwaiter().GetResult();
|
|
}
|
|
|
|
public async Task MakeAdmin(bool isAdmin = true)
|
|
{
|
|
var userManager = parent.PayTester.GetService<UserManager<ApplicationUser>>();
|
|
var u = await userManager.FindByIdAsync(UserId);
|
|
if (isAdmin)
|
|
await userManager.AddToRoleAsync(u, Roles.ServerAdmin);
|
|
else
|
|
await userManager.RemoveFromRoleAsync(u, Roles.ServerAdmin);
|
|
IsAdmin = true;
|
|
}
|
|
|
|
public Task<BTCPayServerClient> CreateClient()
|
|
{
|
|
return Task.FromResult(new BTCPayServerClient(parent.PayTester.ServerUri, RegisterDetails.Email,
|
|
RegisterDetails.Password));
|
|
}
|
|
|
|
public async Task<BTCPayServerClient> CreateClient(params string[] permissions)
|
|
{
|
|
var manageController = parent.PayTester.GetController<ManageController>(UserId, StoreId, IsAdmin);
|
|
var x = Assert.IsType<RedirectToActionResult>(await manageController.AddApiKey(
|
|
new ManageController.AddApiKeyViewModel()
|
|
{
|
|
PermissionValues = permissions.Select(s =>
|
|
{
|
|
Permission.TryParse(s, out var p);
|
|
return p;
|
|
}).GroupBy(permission => permission.Policy).Select(p =>
|
|
{
|
|
var stores = p.Where(permission => !string.IsNullOrEmpty(permission.Scope))
|
|
.Select(permission => permission.Scope).ToList();
|
|
return new ManageController.AddApiKeyViewModel.PermissionValueItem()
|
|
{
|
|
Permission = p.Key,
|
|
Forbidden = false,
|
|
StoreMode = stores.Any() ? ManageController.AddApiKeyViewModel.ApiKeyStoreMode.Specific : ManageController.AddApiKeyViewModel.ApiKeyStoreMode.AllStores,
|
|
SpecificStores = stores,
|
|
Value = true
|
|
};
|
|
}).ToList()
|
|
}));
|
|
var statusMessage = manageController.TempData.GetStatusMessageModel();
|
|
Assert.NotNull(statusMessage);
|
|
var str = "<code class='alert-link'>";
|
|
var apiKey = statusMessage.Html.Substring(statusMessage.Html.IndexOf(str) + str.Length);
|
|
apiKey = apiKey.Substring(0, apiKey.IndexOf("</code>"));
|
|
return new BTCPayServerClient(parent.PayTester.ServerUri, apiKey);
|
|
}
|
|
|
|
public void Register(bool isAdmin = false)
|
|
{
|
|
RegisterAsync(isAdmin).GetAwaiter().GetResult();
|
|
}
|
|
|
|
public async Task GrantAccessAsync(bool isAdmin = false)
|
|
{
|
|
await RegisterAsync(isAdmin);
|
|
await CreateStoreAsync();
|
|
var store = this.GetController<StoresController>();
|
|
var pairingCode = BitPay.RequestClientAuthorization("test", Facade.Merchant);
|
|
Assert.IsType<ViewResult>(await store.RequestPairing(pairingCode.ToString()));
|
|
await store.Pair(pairingCode.ToString(), StoreId);
|
|
}
|
|
|
|
public BTCPayServerClient CreateClientFromAPIKey(string apiKey)
|
|
{
|
|
return new BTCPayServerClient(parent.PayTester.ServerUri, apiKey);
|
|
}
|
|
|
|
public void CreateStore()
|
|
{
|
|
CreateStoreAsync().GetAwaiter().GetResult();
|
|
}
|
|
|
|
public void SetNetworkFeeMode(NetworkFeeMode mode)
|
|
{
|
|
ModifyStore((store) =>
|
|
{
|
|
store.NetworkFeeMode = mode;
|
|
});
|
|
}
|
|
|
|
public void ModifyStore(Action<StoreViewModel> modify)
|
|
{
|
|
var storeController = GetController<StoresController>();
|
|
StoreViewModel store = (StoreViewModel)((ViewResult)storeController.UpdateStore()).Model;
|
|
modify(store);
|
|
storeController.UpdateStore(store).GetAwaiter().GetResult();
|
|
}
|
|
public Task ModifyStoreAsync(Action<StoreViewModel> modify)
|
|
{
|
|
var storeController = GetController<StoresController>();
|
|
StoreViewModel store = (StoreViewModel)((ViewResult)storeController.UpdateStore()).Model;
|
|
modify(store);
|
|
return storeController.UpdateStore(store);
|
|
}
|
|
|
|
public T GetController<T>(bool setImplicitStore = true) where T : Controller
|
|
{
|
|
var controller = parent.PayTester.GetController<T>(UserId, setImplicitStore ? StoreId : null, IsAdmin);
|
|
return controller;
|
|
}
|
|
|
|
public async Task CreateStoreAsync()
|
|
{
|
|
if (UserId is null)
|
|
{
|
|
await RegisterAsync();
|
|
}
|
|
var store = this.GetController<UserStoresController>();
|
|
await store.CreateStore(new CreateStoreViewModel() { Name = "Test Store" });
|
|
StoreId = store.CreatedStoreId;
|
|
parent.Stores.Add(StoreId);
|
|
}
|
|
|
|
public BTCPayNetwork SupportedNetwork { get; set; }
|
|
|
|
public WalletId RegisterDerivationScheme(string crytoCode, ScriptPubKeyType segwit = ScriptPubKeyType.Legacy, bool importKeysToNBX = false)
|
|
{
|
|
return RegisterDerivationSchemeAsync(crytoCode, segwit, importKeysToNBX).GetAwaiter().GetResult();
|
|
}
|
|
|
|
public async Task<WalletId> RegisterDerivationSchemeAsync(string cryptoCode, ScriptPubKeyType segwit = ScriptPubKeyType.Legacy,
|
|
bool importKeysToNBX = false)
|
|
{
|
|
if (StoreId is null)
|
|
await CreateStoreAsync();
|
|
SupportedNetwork = parent.NetworkProvider.GetNetwork<BTCPayNetwork>(cryptoCode);
|
|
var store = parent.PayTester.GetController<StoresController>(UserId, StoreId);
|
|
GenerateWalletResponseV = await parent.ExplorerClient.GenerateWalletAsync(new GenerateWalletRequest()
|
|
{
|
|
ScriptPubKeyType = segwit,
|
|
SavePrivateKeys = importKeysToNBX,
|
|
});
|
|
await store.AddDerivationScheme(StoreId,
|
|
new DerivationSchemeViewModel()
|
|
{
|
|
Enabled = true,
|
|
CryptoCode = cryptoCode,
|
|
Network = SupportedNetwork,
|
|
RootFingerprint = GenerateWalletResponseV.AccountKeyPath.MasterFingerprint.ToString(),
|
|
RootKeyPath = SupportedNetwork.GetRootKeyPath(),
|
|
Source = "NBXplorer",
|
|
AccountKey = GenerateWalletResponseV.AccountHDKey.Neuter().ToWif(),
|
|
DerivationSchemeFormat = "BTCPay",
|
|
KeyPath = GenerateWalletResponseV.AccountKeyPath.KeyPath.ToString(),
|
|
DerivationScheme = DerivationScheme.ToString(),
|
|
Confirmation = true
|
|
}, cryptoCode);
|
|
return new WalletId(StoreId, cryptoCode);
|
|
}
|
|
|
|
public Task EnablePayJoin()
|
|
{
|
|
return ModifyStoreAsync(s => s.PayJoinEnabled = true);
|
|
}
|
|
|
|
public GenerateWalletResponse GenerateWalletResponseV { get; set; }
|
|
|
|
public DerivationStrategyBase DerivationScheme
|
|
{
|
|
get
|
|
{
|
|
return GenerateWalletResponseV.DerivationScheme;
|
|
}
|
|
}
|
|
|
|
private async Task RegisterAsync(bool isAdmin = false)
|
|
{
|
|
var account = parent.PayTester.GetController<AccountController>();
|
|
RegisterDetails = new RegisterViewModel()
|
|
{
|
|
Email = Guid.NewGuid() + "@toto.com",
|
|
ConfirmPassword = "Kitten0@",
|
|
Password = "Kitten0@",
|
|
IsAdmin = isAdmin
|
|
};
|
|
await account.Register(RegisterDetails);
|
|
UserId = account.RegisteredUserId;
|
|
IsAdmin = account.RegisteredAdmin;
|
|
}
|
|
|
|
public RegisterViewModel RegisterDetails { get; set; }
|
|
|
|
public Bitpay BitPay
|
|
{
|
|
get;
|
|
set;
|
|
}
|
|
|
|
public string UserId
|
|
{
|
|
get;
|
|
set;
|
|
}
|
|
|
|
public string StoreId
|
|
{
|
|
get;
|
|
set;
|
|
}
|
|
|
|
public bool IsAdmin { get; internal set; }
|
|
|
|
public void RegisterLightningNode(string cryptoCode, LightningConnectionType connectionType, bool isMerchant = true)
|
|
{
|
|
RegisterLightningNodeAsync(cryptoCode, connectionType, isMerchant).GetAwaiter().GetResult();
|
|
}
|
|
|
|
public async Task RegisterLightningNodeAsync(string cryptoCode, LightningConnectionType connectionType, bool isMerchant = true)
|
|
{
|
|
var storeController = this.GetController<StoresController>();
|
|
|
|
string connectionString = null;
|
|
if (connectionType == LightningConnectionType.Charge)
|
|
{
|
|
if (isMerchant)
|
|
connectionString = $"type=charge;server={parent.MerchantCharge.Client.Uri.AbsoluteUri};allowinsecure=true";
|
|
else
|
|
throw new NotSupportedException();
|
|
}
|
|
else if (connectionType == LightningConnectionType.CLightning)
|
|
{
|
|
if (isMerchant)
|
|
connectionString = "type=clightning;server=" +
|
|
((CLightningClient)parent.MerchantLightningD).Address.AbsoluteUri;
|
|
else
|
|
connectionString = "type=clightning;server=" +
|
|
((CLightningClient)parent.CustomerLightningD).Address.AbsoluteUri;
|
|
}
|
|
else if (connectionType == LightningConnectionType.LndREST)
|
|
{
|
|
if (isMerchant)
|
|
connectionString = $"type=lnd-rest;server={parent.MerchantLnd.Swagger.BaseUrl};allowinsecure=true";
|
|
else
|
|
throw new NotSupportedException();
|
|
}
|
|
else
|
|
throw new NotSupportedException(connectionType.ToString());
|
|
|
|
await storeController.AddLightningNode(StoreId,
|
|
new LightningNodeViewModel() { ConnectionString = connectionString, SkipPortTest = true }, "save", "BTC");
|
|
if (storeController.ModelState.ErrorCount != 0)
|
|
Assert.False(true, storeController.ModelState.FirstOrDefault().Value.Errors[0].ErrorMessage);
|
|
}
|
|
|
|
public async Task<Coin> ReceiveUTXO(Money value, BTCPayNetwork network)
|
|
{
|
|
var cashCow = parent.ExplorerNode;
|
|
var btcPayWallet = parent.PayTester.GetService<BTCPayWalletProvider>().GetWallet(network);
|
|
var address = (await btcPayWallet.ReserveAddressAsync(this.DerivationScheme)).Address;
|
|
await parent.WaitForEvent<NewOnChainTransactionEvent>(async () =>
|
|
{
|
|
await cashCow.SendToAddressAsync(address, value);
|
|
});
|
|
int i = 0;
|
|
while (i < 30)
|
|
{
|
|
var result = (await btcPayWallet.GetUnspentCoins(DerivationScheme))
|
|
.FirstOrDefault(c => c.ScriptPubKey == address.ScriptPubKey)?.Coin;
|
|
if (result != null)
|
|
{
|
|
return result;
|
|
}
|
|
|
|
await Task.Delay(1000);
|
|
i++;
|
|
}
|
|
Assert.False(true);
|
|
return null;
|
|
}
|
|
|
|
public async Task<BitcoinAddress> GetNewAddress(BTCPayNetwork network)
|
|
{
|
|
var cashCow = parent.ExplorerNode;
|
|
var btcPayWallet = parent.PayTester.GetService<BTCPayWalletProvider>().GetWallet(network);
|
|
var address = (await btcPayWallet.ReserveAddressAsync(this.DerivationScheme)).Address;
|
|
return address;
|
|
}
|
|
|
|
public async Task<PSBT> Sign(PSBT psbt)
|
|
{
|
|
var btcPayWallet = parent.PayTester.GetService<BTCPayWalletProvider>()
|
|
.GetWallet(psbt.Network.NetworkSet.CryptoCode);
|
|
var explorerClient = parent.PayTester.GetService<ExplorerClientProvider>()
|
|
.GetExplorerClient(psbt.Network.NetworkSet.CryptoCode);
|
|
psbt = (await explorerClient.UpdatePSBTAsync(new UpdatePSBTRequest()
|
|
{
|
|
DerivationScheme = DerivationScheme,
|
|
PSBT = psbt
|
|
})).PSBT;
|
|
return psbt.SignAll(this.DerivationScheme, GenerateWalletResponseV.AccountHDKey,
|
|
GenerateWalletResponseV.AccountKeyPath);
|
|
}
|
|
|
|
public async Task<PSBT> SubmitPayjoin(Invoice invoice, PSBT psbt, string expectedError = null, bool senderError = false)
|
|
{
|
|
var endpoint = GetPayjoinBitcoinUrl(invoice, psbt.Network);
|
|
if (endpoint == null)
|
|
{
|
|
throw new InvalidOperationException("No payjoin endpoint for the invoice");
|
|
}
|
|
var pjClient = parent.PayTester.GetService<PayjoinClient>();
|
|
var storeRepository = parent.PayTester.GetService<StoreRepository>();
|
|
var store = await storeRepository.FindStore(StoreId);
|
|
var settings = store.GetSupportedPaymentMethods(parent.NetworkProvider).OfType<DerivationSchemeSettings>()
|
|
.First();
|
|
Logs.Tester.LogInformation($"Proposing {psbt.GetGlobalTransaction().GetHash()}");
|
|
if (expectedError is null && !senderError)
|
|
{
|
|
var proposed = await pjClient.RequestPayjoin(endpoint, settings, psbt, default);
|
|
Logs.Tester.LogInformation($"Proposed payjoin is {proposed.GetGlobalTransaction().GetHash()}");
|
|
Assert.NotNull(proposed);
|
|
return proposed;
|
|
}
|
|
else
|
|
{
|
|
if (senderError)
|
|
{
|
|
await Assert.ThrowsAsync<PayjoinSenderException>(async () => await pjClient.RequestPayjoin(endpoint, settings, psbt, default));
|
|
}
|
|
else
|
|
{
|
|
var ex = await Assert.ThrowsAsync<PayjoinReceiverException>(async () => await pjClient.RequestPayjoin(endpoint, settings, psbt, default));
|
|
var split = expectedError.Split('|');
|
|
Assert.Equal(split[0], ex.ErrorCode);
|
|
if (split.Length > 1)
|
|
Assert.Contains(split[1], ex.ReceiverMessage);
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public async Task<Transaction> SubmitPayjoin(Invoice invoice, Transaction transaction, BTCPayNetwork network,
|
|
string expectedError = null)
|
|
{
|
|
var response =
|
|
await SubmitPayjoinCore(transaction.ToHex(), invoice, network.NBitcoinNetwork, expectedError);
|
|
if (response == null)
|
|
return null;
|
|
var signed = Transaction.Parse(await response.Content.ReadAsStringAsync(), network.NBitcoinNetwork);
|
|
return signed;
|
|
}
|
|
|
|
async Task<HttpResponseMessage> SubmitPayjoinCore(string content, Invoice invoice, Network network,
|
|
string expectedError)
|
|
{
|
|
var bip21 = GetPayjoinBitcoinUrl(invoice, network);
|
|
bip21.TryGetPayjoinEndpoint(out var endpoint);
|
|
var response = await parent.PayTester.HttpClient.PostAsync(endpoint,
|
|
new StringContent(content, Encoding.UTF8, "text/plain"));
|
|
if (expectedError != null)
|
|
{
|
|
var split = expectedError.Split('|');
|
|
Assert.False(response.IsSuccessStatusCode);
|
|
var error = JObject.Parse(await response.Content.ReadAsStringAsync());
|
|
if (split.Length > 0)
|
|
Assert.Equal(split[0], error["errorCode"].Value<string>());
|
|
if (split.Length > 1)
|
|
Assert.Contains(split[1], error["message"].Value<string>());
|
|
return null;
|
|
}
|
|
else
|
|
{
|
|
if (!response.IsSuccessStatusCode)
|
|
{
|
|
var error = JObject.Parse(await response.Content.ReadAsStringAsync());
|
|
Assert.True(false,
|
|
$"Error: {error["errorCode"].Value<string>()}: {error["message"].Value<string>()}");
|
|
}
|
|
}
|
|
|
|
return response;
|
|
}
|
|
|
|
public static BitcoinUrlBuilder GetPayjoinBitcoinUrl(Invoice invoice, Network network)
|
|
{
|
|
var parsedBip21 = new BitcoinUrlBuilder(
|
|
invoice.CryptoInfo.First(c => c.CryptoCode == network.NetworkSet.CryptoCode).PaymentUrls.BIP21,
|
|
network);
|
|
if (!parsedBip21.TryGetPayjoinEndpoint(out var endpoint))
|
|
return null;
|
|
return parsedBip21;
|
|
}
|
|
|
|
class WebhookListener : IDisposable
|
|
{
|
|
private Client.Models.StoreWebhookData _wh;
|
|
private FakeServer _server;
|
|
private readonly List<WebhookInvoiceEvent> _webhookEvents;
|
|
private CancellationTokenSource _cts;
|
|
public WebhookListener(Client.Models.StoreWebhookData wh, FakeServer server, List<WebhookInvoiceEvent> webhookEvents)
|
|
{
|
|
_wh = wh;
|
|
_server = server;
|
|
_webhookEvents = webhookEvents;
|
|
_cts = new CancellationTokenSource();
|
|
_ = Listen(_cts.Token);
|
|
}
|
|
|
|
async Task Listen(CancellationToken cancellation)
|
|
{
|
|
while (!cancellation.IsCancellationRequested)
|
|
{
|
|
var req = await _server.GetNextRequest(cancellation);
|
|
var bytes = await req.Request.Body.ReadBytesAsync((int)req.Request.Headers.ContentLength);
|
|
var callback = Encoding.UTF8.GetString(bytes);
|
|
_webhookEvents.Add(JsonConvert.DeserializeObject<WebhookInvoiceEvent>(callback));
|
|
req.Response.StatusCode = 200;
|
|
_server.Done();
|
|
}
|
|
}
|
|
public void Dispose()
|
|
{
|
|
_cts.Cancel();
|
|
_server.Dispose();
|
|
}
|
|
}
|
|
|
|
public List<WebhookInvoiceEvent> WebhookEvents { get; set; } = new List<WebhookInvoiceEvent>();
|
|
public TEvent AssertHasWebhookEvent<TEvent>(WebhookEventType eventType, Action<TEvent> assert) where TEvent : class
|
|
{
|
|
int retry = 0;
|
|
retry:
|
|
foreach (var evt in WebhookEvents)
|
|
{
|
|
if (evt.Type == eventType)
|
|
{
|
|
var typedEvt = evt.ReadAs<TEvent>();
|
|
try
|
|
{
|
|
assert(typedEvt);
|
|
return typedEvt;
|
|
}
|
|
catch (XunitException)
|
|
{
|
|
}
|
|
}
|
|
}
|
|
if (retry < 3)
|
|
{
|
|
Thread.Sleep(1000);
|
|
retry++;
|
|
goto retry;
|
|
}
|
|
Assert.True(false, "No webhook event match the assertion");
|
|
return null;
|
|
}
|
|
public async Task SetupWebhook()
|
|
{
|
|
FakeServer server = new FakeServer();
|
|
await server.Start();
|
|
var client = await CreateClient(Policies.CanModifyStoreWebhooks);
|
|
var wh = await client.CreateWebhook(StoreId, new CreateStoreWebhookRequest()
|
|
{
|
|
AutomaticRedelivery = false,
|
|
Url = server.ServerUri.AbsoluteUri
|
|
});
|
|
|
|
parent.Resources.Add(new WebhookListener(wh, server, WebhookEvents));
|
|
}
|
|
|
|
public async Task PayInvoice(string invoiceId)
|
|
{
|
|
var inv = await BitPay.GetInvoiceAsync(invoiceId);
|
|
var net = parent.ExplorerNode.Network;
|
|
this.parent.ExplorerNode.SendToAddress(BitcoinAddress.Create(inv.BitcoinAddress, net), inv.BtcDue);
|
|
await TestUtils.EventuallyAsync(async () =>
|
|
{
|
|
var localInvoice = await BitPay.GetInvoiceAsync(invoiceId, Facade.Merchant);
|
|
Assert.Equal("paid", localInvoice.Status);
|
|
});
|
|
}
|
|
}
|
|
}
|