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