mirror of
https://github.com/QuilibriumNetwork/ceremonyclient.git
synced 2026-02-21 18:37:26 +08:00
dbd95bd9e9
* v2.1.0 [omit consensus and adjacent] - this commit will be amended with the full release after the file copy is complete * 2.1.0 main node rollup
84 lines
2.5 KiB
Go
84 lines
2.5 KiB
Go
package hypergraph_test
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
nodehg "source.quilibrium.com/quilibrium/monorepo/hypergraph"
|
|
"source.quilibrium.com/quilibrium/monorepo/node/execution/intrinsics/hypergraph"
|
|
hgstate "source.quilibrium.com/quilibrium/monorepo/node/execution/state/hypergraph"
|
|
"source.quilibrium.com/quilibrium/monorepo/node/keys"
|
|
"source.quilibrium.com/quilibrium/monorepo/types/mocks"
|
|
crypto "source.quilibrium.com/quilibrium/monorepo/types/tries"
|
|
)
|
|
|
|
func TestVertexRemove_GetCost(t *testing.T) {
|
|
// Setup
|
|
domain := [32]byte{1, 2, 3}
|
|
dataAddress := [32]byte{4, 5, 6}
|
|
|
|
// Create signer
|
|
signer, err := keys.NewEd448Key()
|
|
require.NoError(t, err)
|
|
|
|
vertexRemove := hypergraph.NewVertexRemove(domain, dataAddress, signer)
|
|
cost, err := vertexRemove.GetCost()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, big.NewInt(64), cost) // Fixed cost for vertex removal
|
|
}
|
|
|
|
func TestVertexRemove_Prove(t *testing.T) {
|
|
// Setup
|
|
domain := [32]byte{1, 2, 3}
|
|
dataAddress := [32]byte{4, 5, 6}
|
|
|
|
// Create signer
|
|
signer, err := keys.NewEd448Key()
|
|
require.NoError(t, err)
|
|
|
|
// Test prove
|
|
vertexRemove := hypergraph.NewVertexRemove(domain, dataAddress, signer)
|
|
err = vertexRemove.Prove(1)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestVertexRemove_Verify(t *testing.T) {
|
|
// Setup
|
|
domain := [32]byte{1, 2, 3}
|
|
dataAddress := [32]byte{4, 5, 6}
|
|
|
|
// Create signer
|
|
signer, err := keys.NewEd448Key()
|
|
require.NoError(t, err)
|
|
|
|
// Test verify
|
|
vertexRemove := hypergraph.NewVertexRemove(domain, dataAddress, signer)
|
|
// Note: Verify requires keyManager and config to be set, which happens in InvokeStep
|
|
// For unit tests, we'll skip this verification as it gets tested in integration tests
|
|
_ = vertexRemove // Unused in unit test, but kept for clarity
|
|
}
|
|
|
|
func TestVertexRemove_Materialize(t *testing.T) {
|
|
// Setup
|
|
domain := [32]byte{1, 2, 3}
|
|
dataAddress := [32]byte{4, 5, 6}
|
|
mockHypergraph := &mocks.MockHypergraph{}
|
|
mockHypergraph.On("GetVertex", mock.Anything).Return(nodehg.NewVertex(domain, dataAddress, make([]byte, 74), big.NewInt(74)), nil)
|
|
mockHypergraph.On("GetVertexData", mock.Anything).Return(&crypto.VectorCommitmentTree{}, nil)
|
|
hgState := hgstate.NewHypergraphState(mockHypergraph)
|
|
|
|
// Create signer
|
|
signer, err := keys.NewEd448Key()
|
|
require.NoError(t, err)
|
|
|
|
vertexRemove := hypergraph.NewVertexRemove(domain, dataAddress, signer)
|
|
|
|
// Test materialization
|
|
resultState, err := vertexRemove.Materialize(1, hgState)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, hgState, resultState)
|
|
}
|