go-blockchain/consensus/fee_test.go
Claude 7ac618d339
feat(consensus): fee extraction with overflow checks
TxFee calculates pre-HF4 fees as sum(inputs) - sum(outputs) with
overflow detection. Coinbase transactions return zero fee.

Co-Authored-By: Charon <charon@lethean.io>
2026-02-21 00:45:01 +00:00

73 lines
1.6 KiB
Go

// Copyright (c) 2017-2026 Lethean (https://lt.hn)
//
// Licensed under the European Union Public Licence (EUPL) version 1.2.
// SPDX-License-Identifier: EUPL-1.2
//go:build !integration
package consensus
import (
"testing"
"forge.lthn.ai/core/go-blockchain/types"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestTxFee_Good(t *testing.T) {
// Coinbase tx: fee is 0.
coinbase := &types.Transaction{
Version: types.VersionInitial,
Vin: []types.TxInput{types.TxInputGenesis{Height: 1}},
}
fee, err := TxFee(coinbase)
require.NoError(t, err)
assert.Equal(t, uint64(0), fee)
// Normal v1 tx: fee = inputs - outputs.
tx := &types.Transaction{
Version: types.VersionPreHF4,
Vin: []types.TxInput{
types.TxInputToKey{Amount: 100},
types.TxInputToKey{Amount: 50},
},
Vout: []types.TxOutput{
types.TxOutputBare{Amount: 120},
},
}
fee, err = TxFee(tx)
require.NoError(t, err)
assert.Equal(t, uint64(30), fee)
}
func TestTxFee_Bad(t *testing.T) {
// Outputs exceed inputs.
tx := &types.Transaction{
Version: types.VersionPreHF4,
Vin: []types.TxInput{
types.TxInputToKey{Amount: 50},
},
Vout: []types.TxOutput{
types.TxOutputBare{Amount: 100},
},
}
_, err := TxFee(tx)
assert.ErrorIs(t, err, ErrNegativeFee)
}
func TestTxFee_Ugly(t *testing.T) {
// Input amounts that would overflow uint64.
tx := &types.Transaction{
Version: types.VersionPreHF4,
Vin: []types.TxInput{
types.TxInputToKey{Amount: ^uint64(0)},
types.TxInputToKey{Amount: 1},
},
Vout: []types.TxOutput{
types.TxOutputBare{Amount: 1},
},
}
_, err := TxFee(tx)
assert.ErrorIs(t, err, ErrInputOverflow)
}