1package gocircomprover
 2
 3import (
 4	"encoding/json"
 5	"io/ioutil"
 6	"math/big"
 7	"testing"
 8
 9	"github.com/stretchr/testify/assert"
10	"github.com/stretchr/testify/require"
11)
12
13func TestSmallCircuitGenerateProf(t *testing.T) {
14	provingKeyJson, err := ioutil.ReadFile("testdata/small/proving_key.json")
15	require.Nil(t, err)
16	pk, err := ParsePk(provingKeyJson)
17	require.Nil(t, err)
18
19	witnessJson, err := ioutil.ReadFile("testdata/small/witness.json")
20	require.Nil(t, err)
21	w, err := ParseWitness(witnessJson)
22	require.Nil(t, err)
23
24	assert.Equal(t, Witness{big.NewInt(1), big.NewInt(33), big.NewInt(3), big.NewInt(11)}, w)
25
26	proof, pubSignals, err := GenerateProof(pk, w)
27	assert.Nil(t, err)
28
29	proofStr, err := ProofToJson(proof)
30	assert.Nil(t, err)
31
32	err = ioutil.WriteFile("testdata/small/proof.json", proofStr, 0644)
33	assert.Nil(t, err)
34	publicStr, err := json.Marshal(ArrayBigIntToString(pubSignals))
35	assert.Nil(t, err)
36	err = ioutil.WriteFile("testdata/small/public.json", publicStr, 0644)
37	assert.Nil(t, err)
38
39	// to verify the proof:
40	// snarkjs verify --vk testdata/small/verification_key.json -p testdata/small/proof.json --pub testdata/small/public.json
41}
42
43func TestBigCircuitGenerateProf(t *testing.T) {
44	provingKeyJson, err := ioutil.ReadFile("testdata/big/proving_key.json")
45	require.Nil(t, err)
46	pk, err := ParsePk(provingKeyJson)
47	require.Nil(t, err)
48
49	witnessJson, err := ioutil.ReadFile("testdata/big/witness.json")
50	require.Nil(t, err)
51	w, err := ParseWitness(witnessJson)
52	require.Nil(t, err)
53
54	proof, pubSignals, err := GenerateProof(pk, w)
55	assert.Nil(t, err)
56
57	proofStr, err := ProofToJson(proof)
58	assert.Nil(t, err)
59
60	err = ioutil.WriteFile("testdata/big/proof.json", proofStr, 0644)
61	assert.Nil(t, err)
62	publicStr, err := json.Marshal(ArrayBigIntToString(pubSignals))
63	assert.Nil(t, err)
64	err = ioutil.WriteFile("testdata/big/public.json", publicStr, 0644)
65	assert.Nil(t, err)
66
67	// to verify the proof:
68	// snarkjs verify --vk testdata/big/verification_key.json -p testdata/big/proof.json --pub testdata/big/public.json
69}