Update prover & add test

prover/prover_test.go

@@ -0,0 +1,83 @@
+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)
+}