// 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();
}