You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

80 lines
2.3 KiB

package gocircomprover
import (
"encoding/json"
"fmt"
"io/ioutil"
"math/big"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestSmallCircuitGenerateProf(t *testing.T) {
provingKeyJson, err := ioutil.ReadFile("testdata/small/proving_key.json")
require.Nil(t, err)
pk, err := ParseProvingKey(provingKeyJson)
require.Nil(t, err)
fmt.Println("polsA", pk.PolsA)
fmt.Println("polsB", pk.PolsB)
fmt.Println("polsC", pk.PolsC)
witnessJson, err := ioutil.ReadFile("testdata/small/witness.json")
require.Nil(t, err)
w, err := ParseWitness(witnessJson)
require.Nil(t, err)
assert.Equal(t, Witness{big.NewInt(1), big.NewInt(33), big.NewInt(3), big.NewInt(11)}, w)
proof, pubSignals, err := GenerateProof(pk, w)
assert.Nil(t, err)
fmt.Println("proof", proof)
fmt.Println("pubSignals", pubSignals)
proofStr, err := ProofToJson(proof)
assert.Nil(t, err)
fmt.Println("prover\n", string(proofStr))
err = ioutil.WriteFile("testdata/small/proof.json", proofStr, 0644)
assert.Nil(t, err)
publicStr, err := json.Marshal(ArrayBigIntToString(pubSignals))
assert.Nil(t, err)
err = ioutil.WriteFile("testdata/small/public.json", publicStr, 0644)
assert.Nil(t, err)
// to verify the proof:
// snarkjs verify --vk testdata/small/verification_key.json -p testdata/small/proof.json --pub testdata/small/public.json
}
func TestBigCircuitGenerateProf(t *testing.T) {
provingKeyJson, err := ioutil.ReadFile("testdata/big/proving_key.json")
require.Nil(t, err)
pk, err := ParseProvingKey(provingKeyJson)
require.Nil(t, err)
witnessJson, err := ioutil.ReadFile("testdata/big/witness.json")
require.Nil(t, err)
w, err := ParseWitness(witnessJson)
require.Nil(t, err)
proof, pubSignals, err := GenerateProof(pk, w)
assert.Nil(t, err)
fmt.Println("proof", proof)
fmt.Println("pubSignals", pubSignals)
proofStr, err := ProofToJson(proof)
assert.Nil(t, err)
fmt.Println("prover\n", string(proofStr))
err = ioutil.WriteFile("testdata/big/proof.json", proofStr, 0644)
assert.Nil(t, err)
publicStr, err := json.Marshal(ArrayBigIntToString(pubSignals))
assert.Nil(t, err)
err = ioutil.WriteFile("testdata/big/public.json", publicStr, 0644)
assert.Nil(t, err)
// to verify the proof:
// snarkjs verify --vk testdata/big/verification_key.json -p testdata/big/proof.json --pub testdata/big/public.json
}