poetcore_test.go

package main

import (
	"context"
	"crypto/rand"
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/spacemeshos/poet/hash"
	"github.com/spacemeshos/poet/prover"
	"github.com/spacemeshos/poet/shared"
	"github.com/spacemeshos/poet/verifier"
)

func BenchmarkProverAndVerifierBig(b *testing.B) {
	r := require.New(b)
	challenge := make([]byte, 32)

	_, err := rand.Read(challenge)
	r.NoError(err)

	securityParam := shared.T

	b.Log("Computing dag...")
	t1 := time.Now()
	numLeaves, merkleProof, err := prover.GenerateProofWithoutPersistency(
		context.Background(),
		prover.TreeConfig{Datadir: b.TempDir()},
		hash.GenLabelHashFunc(challenge),
		hash.GenMerkleHashFunc(challenge),
		time.Now(),
		securityParam,
	)
	r.NoError(err, "Failed to generate proof")

	e := time.Since(t1)
	b.Logf("Proof generated in %s (%f) \n", e, e.Seconds())
	b.Logf("Dag root label: %x\n", merkleProof.Root)

	err = verifier.Validate(
		*merkleProof,
		hash.GenLabelHashFunc(challenge),
		hash.GenMerkleHashFunc(challenge),
		numLeaves,
		securityParam,
	)
	r.NoError(err, "Failed to verify NIP")

	e1 := time.Since(t1)
	b.Logf("Proof verified in %s (%f)\n", e1-e, (e1 - e).Seconds())
}

func TestNip(t *testing.T) {
	challenge := []byte("Spacemesh launched its mainnet")

	securityParam := shared.T

	numLeaves, merkleProof, err := prover.GenerateProofWithoutPersistency(
		context.Background(),
		prover.TreeConfig{Datadir: t.TempDir()},
		hash.GenLabelHashFunc(challenge),
		hash.GenMerkleHashFunc(challenge),
		time.Now(),
		securityParam,
	)
	require.NoError(t, err)
	fmt.Printf("Dag root label: %x\n", merkleProof.Root)

	err = verifier.Validate(
		*merkleProof,
		hash.GenLabelHashFunc(challenge),
		hash.GenMerkleHashFunc(challenge),
		numLeaves,
		securityParam,
	)
	require.NoError(t, err, "failed to verify proof")
}

func BenchmarkProofEx(t *testing.B) {
	for j := 0; j < 10; j++ {
		// generate random commitment
		challenge := make([]byte, 32)
		_, err := rand.Read(challenge)
		require.NoError(t, err)

		securityParam := shared.T

		numLeaves, merkleProof, err := prover.GenerateProofWithoutPersistency(
			context.Background(),
			prover.TreeConfig{Datadir: t.TempDir()},
			hash.GenLabelHashFunc(challenge),
			hash.GenMerkleHashFunc(challenge),
			time.Now(),
			securityParam,
		)
		require.NoError(t, err)
		fmt.Printf("Dag root label: %x\n", merkleProof.Root)

		err = verifier.Validate(
			*merkleProof,
			hash.GenLabelHashFunc(challenge),
			hash.GenMerkleHashFunc(challenge),
			numLeaves,
			securityParam,
		)
		require.NoError(t, err, "failed to verify proof")
	}
}