123 lines
3.9 KiB
C++
123 lines
3.9 KiB
C++
// Copyright (c) 2019-2023 The Zcash developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or https://www.opensource.org/licenses/mit-license.php .
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <iostream>
|
|
|
|
#include "coins.h"
|
|
#include "init.h"
|
|
#include "key.h"
|
|
#include "transaction_builder.h"
|
|
#include "gtest/utils.h"
|
|
#include "util/test.h"
|
|
|
|
#include <optional>
|
|
|
|
TEST(RecursiveDynamicUsageTests, TestTransactionTransparent)
|
|
{
|
|
auto consensusParams = RegtestActivateSapling();
|
|
|
|
CBasicKeyStore keystore;
|
|
CKey tsk = AddTestCKeyToKeyStore(keystore);
|
|
auto scriptPubKey = GetScriptForDestination(tsk.GetPubKey().GetID());
|
|
CTxDestination taddr = tsk.GetPubKey().GetID();
|
|
|
|
auto builder = TransactionBuilder(consensusParams, 1, std::nullopt, &keystore);
|
|
builder.SetFee(10000);
|
|
builder.AddTransparentInput(COutPoint(), scriptPubKey, 50000);
|
|
builder.AddTransparentOutput(taddr, 40000);
|
|
|
|
auto tx = builder.Build().GetTxOrThrow();
|
|
// 1 vin + 1 vout
|
|
// (96 + 128) + 64
|
|
EXPECT_EQ(288, RecursiveDynamicUsage(tx));
|
|
|
|
RegtestDeactivateSapling();
|
|
}
|
|
|
|
TEST(RecursiveDynamicUsageTests, TestTransactionJoinSplit)
|
|
{
|
|
auto consensusParams = RegtestActivateSapling();
|
|
|
|
auto sproutSk = libzcash::SproutSpendingKey::random();
|
|
|
|
auto wtx = GetValidSproutReceive(sproutSk, 25000, true);
|
|
// 2 vin + 1 vJoinSplit + 1 vShieldedOutput
|
|
// 160 + 1856 + 976
|
|
EXPECT_EQ(2992, RecursiveDynamicUsage(wtx));
|
|
|
|
RegtestDeactivateSapling();
|
|
}
|
|
|
|
TEST(RecursiveDynamicUsageTests, TestTransactionSaplingToSapling)
|
|
{
|
|
LoadProofParameters();
|
|
|
|
auto consensusParams = RegtestActivateSapling();
|
|
|
|
auto sk = libzcash::SaplingSpendingKey::random();
|
|
auto testNote = GetTestSaplingNote(sk.default_address(), 50000);
|
|
|
|
auto builder = TransactionBuilder(consensusParams, 1, std::nullopt);
|
|
builder.SetFee(10000);
|
|
builder.AddSaplingSpend(sk.expanded_spending_key(), testNote.note, testNote.tree.root(), testNote.tree.witness());
|
|
builder.AddSaplingOutput(sk.full_viewing_key().ovk, sk.default_address(), 5000, {});
|
|
|
|
auto tx = builder.Build().GetTxOrThrow();
|
|
// 1 vShieldedSpend + 2 vShieldedOutput
|
|
// 400 + 1920
|
|
EXPECT_EQ(2320, RecursiveDynamicUsage(tx));
|
|
|
|
RegtestDeactivateSapling();
|
|
}
|
|
|
|
TEST(RecursiveDynamicUsageTests, TestTransactionTransparentToSapling)
|
|
{
|
|
LoadProofParameters();
|
|
|
|
auto consensusParams = RegtestActivateSapling();
|
|
|
|
CBasicKeyStore keystore;
|
|
CKey tsk = AddTestCKeyToKeyStore(keystore);
|
|
auto scriptPubKey = GetScriptForDestination(tsk.GetPubKey().GetID());
|
|
auto sk = libzcash::SaplingSpendingKey::random();
|
|
|
|
auto builder = TransactionBuilder(consensusParams, 1, std::nullopt, &keystore);
|
|
builder.SetFee(10000);
|
|
builder.AddTransparentInput(COutPoint(), scriptPubKey, 50000);
|
|
builder.AddSaplingOutput(sk.full_viewing_key().ovk, sk.default_address(), 40000, {});
|
|
|
|
auto tx = builder.Build().GetTxOrThrow();
|
|
// 1 vin + 1 vShieldedOutput
|
|
// (96 + 128) + 976
|
|
EXPECT_EQ(1200, RecursiveDynamicUsage(tx));
|
|
|
|
RegtestDeactivateSapling();
|
|
}
|
|
|
|
TEST(RecursiveDynamicUsageTests, TestTransactionSaplingToTransparent)
|
|
{
|
|
LoadProofParameters();
|
|
|
|
auto consensusParams = RegtestActivateSapling();
|
|
|
|
CBasicKeyStore keystore;
|
|
CKey tsk = AddTestCKeyToKeyStore(keystore);
|
|
CTxDestination taddr = tsk.GetPubKey().GetID();
|
|
auto sk = libzcash::SaplingSpendingKey::random();
|
|
auto testNote = GetTestSaplingNote(sk.default_address(), 50000);
|
|
|
|
auto builder = TransactionBuilder(consensusParams, 1, std::nullopt, &keystore);
|
|
builder.SetFee(10000);
|
|
builder.AddSaplingSpend(sk.expanded_spending_key(), testNote.note, testNote.tree.root(), testNote.tree.witness());
|
|
builder.AddTransparentOutput(taddr, 40000);
|
|
|
|
auto tx = builder.Build().GetTxOrThrow();
|
|
// 1 vShieldedSpend + 1 vout
|
|
// 400 + 64
|
|
EXPECT_EQ(464, RecursiveDynamicUsage(tx));
|
|
|
|
RegtestDeactivateSapling();
|
|
}
|