hermez-node/prover/prover_test.go

package prover

import (
	"context"
	"math/big"
	"os"
	"testing"
	"time"

	"github.com/hermeznetwork/hermez-node/common"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

const apiURL = "http://localhost:3000/api"
const timeCons = 1 * time.Second

var proofServerClient *ProofServerClient

func TestMain(m *testing.M) {
	exitVal := 0
	if os.Getenv("INTEGRATION") != "" {
		proofServerClient = NewProofServerClient(apiURL, timeCons)
		err := proofServerClient.WaitReady(context.Background())
		if err != nil {
			panic(err)
		}
		exitVal = m.Run()
	}
	os.Exit(exitVal)
}

func TestApiServer(t *testing.T) {
	t.Run("testAPIStatus", testAPIStatus)
	t.Run("testCalculateProof", testCalculateProof)
	time.Sleep(time.Second / 4)
	err := proofServerClient.WaitReady(context.Background())
	require.NoError(t, err)
	t.Run("testGetProof", testGetProof)
	t.Run("testCancel", testCancel)
}

func testAPIStatus(t *testing.T) {
	status, err := proofServerClient.apiStatus(context.Background())
	require.NoError(t, err)
	assert.Equal(t, true, status.Status.IsReady())
}

func testCalculateProof(t *testing.T) {
	var zkInputs *common.ZKInputs
	zkInputs = common.NewZKInputs(100, 16, 512, 24, 32, big.NewInt(1))
	err := proofServerClient.CalculateProof(context.Background(), zkInputs)
	require.NoError(t, err)
}

func testGetProof(t *testing.T) {
	proof, err := proofServerClient.GetProof(context.Background())
	require.NoError(t, err)
	require.NotNil(t, proof)
	require.NotNil(t, proof.PiA)
	require.NotNil(t, proof.PiB)
	require.NotNil(t, proof.PiC)
	require.NotNil(t, proof.Protocol)
}

func testCancel(t *testing.T) {
	var zkInputs *common.ZKInputs
	zkInputs = common.NewZKInputs(100, 16, 512, 24, 32, big.NewInt(1))
	err := proofServerClient.CalculateProof(context.Background(), zkInputs)
	require.NoError(t, err)
	// TODO: remove sleep when the server has been reviewed
	time.Sleep(time.Second / 4)
	err = proofServerClient.Cancel(context.Background())
	require.NoError(t, err)
	status, err := proofServerClient.apiStatus(context.Background())
	require.NoError(t, err)
	for status.Status == StatusCodeBusy {
		time.Sleep(proofServerClient.timeCons)
		status, err = proofServerClient.apiStatus(context.Background())
		require.NoError(t, err)
	}
	assert.Equal(t, StatusCodeAborted, status.Status)
}