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>
73 lines
1.6 KiB
Go
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)
|
|
}
|