c3680d0ce0
Upcoming changes constitute breaking public API changes to the blockchain/stake module, therefore, this follows the process for introducing major API breaks which consists of: - Bump the major version in the go.mod of the affected module if not already done since the last release tag - Add a replacement to the go.mod in the main module if not already done since the last release tag - Update all imports in the repo to use the new major version as necessary - Make necessary modifications to allow all other modules to use the new version in the same commit - Repeat the process for any other modules the require a new major as a result of consuming the new major(s)
118 lines
3.9 KiB
Go
118 lines
3.9 KiB
Go
// Copyright (c) 2016 The btcsuite developers
|
|
// Copyright (c) 2015-2022 The Decred developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package mining
|
|
|
|
import (
|
|
"container/heap"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/decred/dcrd/blockchain/stake/v5"
|
|
)
|
|
|
|
// TestStakeTxFeePrioHeap tests the priority heap including the stake types for
|
|
// both transaction fees per KB and transaction priority. It ensures that the
|
|
// primary sorting is first by stake type, and then by the priority type.
|
|
func TestStakeTxFeePrioHeap(t *testing.T) {
|
|
numTestItems := 1000
|
|
|
|
// Create some fake priority items that exercise the expected sort
|
|
// edge conditions.
|
|
testItems := []*txPrioItem{
|
|
{feePerKB: 5678, txType: stake.TxTypeRegular, priority: 3},
|
|
{feePerKB: 5678, txType: stake.TxTypeRegular, priority: 1},
|
|
{feePerKB: 5678, txType: stake.TxTypeRegular, priority: 1}, // Duplicate fee and prio
|
|
{feePerKB: 5678, txType: stake.TxTypeRegular, priority: 5},
|
|
{feePerKB: 5678, txType: stake.TxTypeRegular, priority: 2},
|
|
{feePerKB: 1234, txType: stake.TxTypeRegular, priority: 3},
|
|
{feePerKB: 1234, txType: stake.TxTypeRegular, priority: 1},
|
|
{feePerKB: 1234, txType: stake.TxTypeRegular, priority: 5},
|
|
{feePerKB: 1234, txType: stake.TxTypeRegular, priority: 5}, // Duplicate fee and prio
|
|
{feePerKB: 1234, txType: stake.TxTypeRegular, priority: 2},
|
|
{feePerKB: 10000, txType: stake.TxTypeRegular, priority: 0}, // Higher fee, lower prio
|
|
{feePerKB: 0, txType: stake.TxTypeRegular, priority: 10000}, // Higher prio, lower fee
|
|
{txType: stake.TxTypeSSRtx, autoRevocation: true},
|
|
}
|
|
|
|
// Add random data in addition to the edge conditions already manually
|
|
// specified.
|
|
for i := len(testItems); i < numTestItems; i++ {
|
|
randType := stake.TxType(rand.Intn(4))
|
|
randPrio := rand.Float64() * 100
|
|
randFeePerKB := rand.Float64() * 10
|
|
testItems = append(testItems, &txPrioItem{
|
|
txDesc: nil,
|
|
txType: randType,
|
|
feePerKB: randFeePerKB,
|
|
priority: randPrio,
|
|
})
|
|
}
|
|
|
|
// Test sorting by stake and fee per KB.
|
|
ph := newTxPriorityQueue(numTestItems, txPQByStakeAndFee)
|
|
for i := 0; i < numTestItems; i++ {
|
|
heap.Push(ph, testItems[i])
|
|
}
|
|
last := &txPrioItem{
|
|
txDesc: nil,
|
|
txType: stake.TxTypeSSGen,
|
|
priority: 10000.0,
|
|
feePerKB: 10000.0,
|
|
}
|
|
for i := 0; i < numTestItems; i++ {
|
|
prioItem := heap.Pop(ph)
|
|
txpi, ok := prioItem.(*txPrioItem)
|
|
if ok {
|
|
if txpi.feePerKB > last.feePerKB &&
|
|
compareStakePriority(txpi, last) >= 0 {
|
|
t.Errorf("bad pop: %v fee per KB was more than last of %v "+
|
|
"while the txtype was %v but last was %v",
|
|
txpi.feePerKB, last.feePerKB, txpi.txType, last.txType)
|
|
}
|
|
last = txpi
|
|
}
|
|
}
|
|
|
|
// Test sorting with fees per KB for high stake priority, then
|
|
// priority for low stake priority.
|
|
ph = newTxPriorityQueue(numTestItems, txPQByStakeAndFeeAndThenPriority)
|
|
for i := 0; i < numTestItems; i++ {
|
|
heap.Push(ph, testItems[i])
|
|
}
|
|
last = &txPrioItem{
|
|
txDesc: nil,
|
|
txType: stake.TxTypeSSGen,
|
|
priority: 10000.0,
|
|
feePerKB: 10000.0,
|
|
}
|
|
for i := 0; i < numTestItems; i++ {
|
|
prioItem := heap.Pop(ph)
|
|
txpi, ok := prioItem.(*txPrioItem)
|
|
if ok {
|
|
bothAreLowStakePriority :=
|
|
txStakePriority(txpi.txType, txpi.autoRevocation) == regOrRevocPriority &&
|
|
txStakePriority(last.txType, last.autoRevocation) == regOrRevocPriority
|
|
if !bothAreLowStakePriority {
|
|
if txpi.feePerKB > last.feePerKB &&
|
|
compareStakePriority(txpi, last) >= 0 {
|
|
t.Errorf("bad pop: %v fee per KB was more than last of %v "+
|
|
"while the txtype was %v but last was %v",
|
|
txpi.feePerKB, last.feePerKB, txpi.txType, last.txType)
|
|
}
|
|
}
|
|
if bothAreLowStakePriority {
|
|
if txpi.priority > last.priority &&
|
|
compareStakePriority(txpi, last) >= 0 {
|
|
t.Errorf("bad pop: %v priority was more than last of %v "+
|
|
"while the txtype was %v but last was %v",
|
|
txpi.feePerKB, last.feePerKB, txpi.txType, last.txType)
|
|
}
|
|
}
|
|
last = txpi
|
|
}
|
|
}
|
|
}
|