add circuit for delete identities feature

go.mod

@@ -6,6 +6,7 @@ require (
  github.com/consensys/gnark v0.8.0
  github.com/iden3/go-iden3-crypto v0.0.13
  github.com/prometheus/client_golang v1.14.0
+ github.com/reilabs/gnark-lean-extractor v1.1.0
  github.com/urfave/cli/v2 v2.10.2
 )
 
@@ -25,7 +26,6 @@ require (
  github.com/prometheus/client_model v0.3.0 // indirect
  github.com/prometheus/common v0.37.0 // indirect
  github.com/prometheus/procfs v0.8.0 // indirect
- github.com/reilabs/gnark-lean-extractor v1.1.0 // indirect
  github.com/rogpeppe/go-internal v1.10.0 // indirect
  github.com/russross/blackfriday/v2 v2.1.0 // indirect
  github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673 // indirect

integration_test.go

@@ -1,6 +1,7 @@
-package main
+package main_test
 
 import (
+ gnarkLogger "github.com/consensys/gnark/logger"
  "io"
  "net/http"
  "strings"
@@ -13,24 +14,39 @@ import (
 const ProverAddress = "localhost:8080"
 const MetricsAddress = "localhost:9999"
 
+var mode string
+
 func TestMain(m *testing.M) {
+ gnarkLogger.Set(*logging.Logger())
  logging.Logger().Info().Msg("Setting up the prover")
- ps, err := prover.Setup(3, 2)
+ ps, err := prover.SetupInsertion(3, 2)
  if err != nil {
  panic(err)
  }
  cfg := server.Config{
  ProverAddress: ProverAddress,
  MetricsAddress: MetricsAddress,
+ Mode: server.InsertionMode,
  }
- logging.Logger().Info().Msg("Starting the server")
+ logging.Logger().Info().Msg("Starting the insertion server")
  instance := server.Run(&cfg, ps)
- logging.Logger().Info().Msg("Running the tests")
- defer func() {
- instance.RequestStop()
- instance.AwaitStop()
- }()
+ logging.Logger().Info().Msg("Running the insertion tests")
+ mode = server.InsertionMode
+ m.Run()
+ instance.RequestStop()
+ instance.AwaitStop()
+ cfg.Mode = server.DeletionMode
+ ps, err = prover.SetupDeletion(3, 2)
+ if err != nil {
+ panic(err)
+ }
+ logging.Logger().Info().Msg("Starting the deletion server")
+ instance = server.Run(&cfg, ps)
+ logging.Logger().Info().Msg("Running the deletion tests")
+ mode = server.DeletionMode
  m.Run()
+ instance.RequestStop()
+ instance.AwaitStop()
 }
 
 func TestWrongMethod(t *testing.T) {
@@ -43,7 +59,10 @@ func TestWrongMethod(t *testing.T) {
  }
 }
 
-func TestHappyPath(t *testing.T) {
+func TestInsertionHappyPath(t *testing.T) {
+ if mode != server.InsertionMode {
+ return
+ }
  body := `{
  "inputHash":"0x5057a31740d54d42ac70c05e0768fb770c682cb2c559bdd03fe4099f7e584e4f",
  "startIndex":0,
@@ -63,7 +82,33 @@ func TestHappyPath(t *testing.T) {
  }
 }
 
-func TestWrongInput(t *testing.T) {
+func TestDeletionHappyPath(t *testing.T) {
+ if mode != server.DeletionMode {
+ return
+ }
+ body := `{
+ "inputHash":"0xdcd389a94b549222fadc9e335c358a3fe4d534155182f46927f82ea8491c7480",
+ "deletionIndices":[0,2],
+ "preRoot":"0xd11eefe87b985333c0d327b0cdd39a9641b5ac32c35c2bda84301ef3231a8ac",
+ "postRoot":"0x1912415186579e1d9ff6282b76d081f0acd527d8549ea803385b1382d9498f35",
+ "identityCommitments":["0x1","0x3"],
+ "merkleProofs":[
+ ["0x2","0x20a3af0435914ccd84b806164531b0cd36e37d4efb93efab76913a93e1f30996","0x1069673dcdb12263df301a6ff584a7ec261a44cb9dc68df067a4774460b1f1e1"],
+ ["0x4","0x65e2c6cc08a36c4a943286bc91c216054a1981eb4f7570f67394ef8937a21b8","0x1069673dcdb12263df301a6ff584a7ec261a44cb9dc68df067a4774460b1f1e1"]
+ ]}`
+ response, err := http.Post("http://localhost:8080/prove", "application/json", strings.NewReader(body))
+ if err != nil {
+ t.Fatal(err)
+ }
+ if response.StatusCode != http.StatusOK {
+ t.Fatalf("Expected status code %d, got %d", http.StatusOK, response.StatusCode)
+ }
+}
+
+func TestInsertionWrongInput(t *testing.T) {
+ if mode != server.InsertionMode {
+ return
+ }
  body := `{
  "inputHash":"0x5057a31740d54d42ac70c05e0768fb770c682cb2c559bdd03fe4099f7e584e4f",
  "startIndex":0,
@@ -88,4 +133,35 @@ func TestWrongInput(t *testing.T) {
  if !strings.Contains(string(responseBody), "proving_error") {
  t.Fatalf("Expected error message to be tagged with 'proving_error', got %s", string(responseBody))
  }
+
+}
+
+func TestDeletionWrongInput(t *testing.T) {
+ if mode != server.DeletionMode {
+ return
+ }
+ body := `{
+ "inputHash":"0xdcd389a94b549222fadc9e335c358a3fe4d534155182f46927f82ea8491c7480",
+ "deletionIndices":[0,2],
+ "preRoot":"0xd11eefe87b985333c0d327b0cdd39a9641b5ac32c35c2bda84301ef3231a8ac",
+ "postRoot":"0x1912415186579e1d9ff6282b76d081f0acd527d8549ea803385b1382d9498f35",
+ "identityCommitments":["0x1","0x3"],
+ "merkleProofs":[
+ ["0x2","0xD","0xD"],
+ ["0x4","0xD","0xD"]
+ ]}`
+ response, err := http.Post("http://localhost:8080/prove", "application/json", strings.NewReader(body))
+ if err != nil {
+ t.Fatal(err)
+ }
+ if response.StatusCode != http.StatusBadRequest {
+ t.Fatalf("Expected status code %d, got %d", http.StatusBadRequest, response.StatusCode)
+ }
+ responseBody, err := io.ReadAll(response.Body)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if !strings.Contains(string(responseBody), "proving_error") {
+ t.Fatalf("Expected error message to be tagged with 'proving_error', got %s", string(responseBody))
+ }
 }