// Copyright (c) 2021-present The Bitcoin Core developers

// Distributed under the MIT software license, see the accompanying

// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#include <wallet/transaction.h>

#include <primitives/transaction.h>

#include <serialize.h>

#include <streams.h>

#include <test/util/common.h>

#include <wallet/test/wallet_test_fixture.h>

#include <boost/test/unit_test.hpp>

namespace wallet {

BOOST_FIXTURE_TEST_SUITE(wallet_transaction_tests, WalletTestingSetup)

BOOST_AUTO_TEST_CASE(roundtrip)

{

    for (uint8_t hash = 0; hash < 5; ++hash) {

        for (int index = -2; index < 3; ++index) {

            TxState state = TxStateInterpretSerialized(TxStateUnrecognized{uint256{hash}, index});

            BOOST_CHECK_EQUAL(TxStateSerializedBlockHash(state), uint256{hash});

            BOOST_CHECK_EQUAL(TxStateSerializedIndex(state), index);

        }

    }

}

BOOST_AUTO_TEST_CASE(deserialize_rejects_mismatched_variant_txid)

{

    // Build tx_a and serialise it as a CWalletTx.

    // Needs at least one input: a zero-input tx serialises vin_count as 0x00,

    // which the witness-aware deserialiser misreads as the segwit marker byte.

    CMutableTransaction mtx_a;

    mtx_a.vin.emplace_back(COutPoint{Txid::FromUint256(uint256::ONE), 0});

    mtx_a.vout.emplace_back(COIN, CScript() << OP_TRUE);

    CTransactionRef tx_a = MakeTransactionRef(std::move(mtx_a));

    CWalletTx wtx_a{tx_a, TxStateInactive{}};

    DataStream ss;

    ss << wtx_a;

    // Build tx_b with a different txid to use as a bogus variant.

    CMutableTransaction mtx_b;

    mtx_b.vout.emplace_back(2 * COIN, CScript() << OP_TRUE);

    CTransactionRef tx_b = MakeTransactionRef(std::move(mtx_b));

    BOOST_REQUIRE(tx_b->GetHash() != tx_a->GetHash());

    // A variant whose txid doesn't match the canonical txid must be rejected.

    std::map<Wtxid, CTransactionRef> bad_variants{{tx_b->GetWitnessHash(), tx_b}};

    try {

        CWalletTx(deserialize, ss, bad_variants);

        BOOST_FAIL("expected std::runtime_error was not thrown");

    } catch (const std::runtime_error& e) {

        BOOST_CHECK_EQUAL(std::string(e.what()), "variant txid does not match wallet txid");

    }

}

BOOST_AUTO_TEST_SUITE_END()

} // namespace wallet