mirror of
https://github.com/luxfi/node.git
synced 2026-10-30 15:24:05 +00:00
105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package tree
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
consensustest "github.com/luxfi/consensus/test/helpers"
|
|
"github.com/luxfi/vm/chain/blocktest"
|
|
)
|
|
|
|
func TestAcceptSingleBlock(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
tr := New()
|
|
|
|
block := blocktest.BuildChild(blocktest.Genesis)
|
|
_, contains := tr.Get(block)
|
|
require.False(contains)
|
|
|
|
tr.Add(block)
|
|
|
|
_, contains = tr.Get(block)
|
|
require.True(contains)
|
|
|
|
require.NoError(tr.Accept(context.Background(), block))
|
|
require.Equal(consensustest.Accepted, block.Status())
|
|
|
|
_, contains = tr.Get(block)
|
|
require.False(contains)
|
|
}
|
|
|
|
func TestAcceptBlockConflict(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
tr := New()
|
|
|
|
blockToAccept := blocktest.BuildChild(blocktest.Genesis)
|
|
blockToReject := blocktest.BuildChild(blocktest.Genesis)
|
|
|
|
// add conflicting blocks
|
|
tr.Add(blockToAccept)
|
|
_, contains := tr.Get(blockToAccept)
|
|
require.True(contains)
|
|
|
|
tr.Add(blockToReject)
|
|
_, contains = tr.Get(blockToReject)
|
|
require.True(contains)
|
|
|
|
// accept one of them
|
|
require.NoError(tr.Accept(context.Background(), blockToAccept))
|
|
|
|
// check their statuses and that they are removed from the tree
|
|
require.Equal(consensustest.Accepted, blockToAccept.Status())
|
|
_, contains = tr.Get(blockToAccept)
|
|
require.False(contains)
|
|
|
|
require.Equal(consensustest.Rejected, blockToReject.Status())
|
|
_, contains = tr.Get(blockToReject)
|
|
require.False(contains)
|
|
}
|
|
|
|
func TestAcceptChainConflict(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
tr := New()
|
|
|
|
blockToAccept := blocktest.BuildChild(blocktest.Genesis)
|
|
blockToReject := blocktest.BuildChild(blocktest.Genesis)
|
|
blockToRejectChild := blocktest.BuildChild(blockToReject)
|
|
|
|
// add conflicting blocks.
|
|
tr.Add(blockToAccept)
|
|
_, contains := tr.Get(blockToAccept)
|
|
require.True(contains)
|
|
|
|
tr.Add(blockToReject)
|
|
_, contains = tr.Get(blockToReject)
|
|
require.True(contains)
|
|
|
|
tr.Add(blockToRejectChild)
|
|
_, contains = tr.Get(blockToRejectChild)
|
|
require.True(contains)
|
|
|
|
// accept one of them
|
|
require.NoError(tr.Accept(context.Background(), blockToAccept))
|
|
|
|
// check their statuses and whether they are removed from tree
|
|
require.Equal(consensustest.Accepted, blockToAccept.Status())
|
|
_, contains = tr.Get(blockToAccept)
|
|
require.False(contains)
|
|
|
|
require.Equal(consensustest.Rejected, blockToReject.Status())
|
|
_, contains = tr.Get(blockToReject)
|
|
require.False(contains)
|
|
|
|
require.Equal(consensustest.Rejected, blockToRejectChild.Status())
|
|
_, contains = tr.Get(blockToRejectChild)
|
|
require.False(contains)
|
|
}
|