test for invalid proof ckzg (#1189)

* test for invalid proof ckzg Signed-off-by: nidhi-singh02 <trippin@berachain.com> * updated lib Signed-off-by: nidhi-singh02 <trippin@berachain.com> --------- Signed-off-by: nidhi-singh02 <trippin@berachain.com>
berachain · May 27, 2024 · 1a91f6b · 1a91f6b
1 parent 560444e
commit 1a91f6b
Show file tree

Hide file tree

Showing 5 changed files with 83 additions and 2 deletions.
diff --git a/contracts/lib/forge-std b/contracts/lib/forge-std
diff --git a/contracts/lib/solady b/contracts/lib/solady
diff --git a/mod/da/pkg/kzg/ckzg/ckzg_cgo_test.go b/mod/da/pkg/kzg/ckzg/ckzg_cgo_test.go
@@ -124,3 +124,40 @@ func TestVerifyBlobProofBatch(t *testing.T) {
 	err = globalVerifier.VerifyBlobProofBatch(args)
 	require.NoError(t, err)
 }
+
+// TestVerifyBlobKZGInvalidProof tests the VerifyBlobProof function for an
+// invalid proof
+func TestVerifyBlobKZGInvalidProof(t *testing.T) {
+	validBlob, invalidProof, validCommitment := setupTestData(
+		t, "test_data_incorrect_proof.json")
+	testCases := []struct {
+		name        string
+		blob        *eip4844.Blob
+		proof       eip4844.KZGProof
+		commitment  eip4844.KZGCommitment
+		expectError bool
+	}{
+		{
+			name:        "Invalid Proof",
+			blob:        validBlob,
+			proof:       invalidProof,
+			commitment:  validCommitment,
+			expectError: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			err := globalVerifier.VerifyBlobProof(
+				tc.blob,
+				tc.proof,
+				tc.commitment,
+			)
+			if tc.expectError {
+				require.Error(t, err, "invalid proof")
+			} else {
+				require.NoError(t, err)
+			}
+		})
+	}
+}
diff --git a/mod/da/pkg/kzg/ckzg/ckzg_no_cgo_test.go b/mod/da/pkg/kzg/ckzg/ckzg_no_cgo_test.go
@@ -121,3 +121,40 @@ func TestVerifyBlobProofBatch(t *testing.T) {
 	err = globalVerifier.VerifyBlobProofBatch(args)
 	require.Error(t, err, "cgo is not enabled")
 }
+
+// TestVerifyBlobKZGInvalidProof tests the VerifyBlobProof function for invalid
+// proofs.
+func TestVerifyBlobKZGInvalidProof(t *testing.T) {
+	validBlob, invalidProof, validCommitment := setupTestData(
+		t, "test_data_incorrect_proof.json")
+	testCases := []struct {
+		name        string
+		blob        *eip4844.Blob
+		proof       eip4844.KZGProof
+		commitment  eip4844.KZGCommitment
+		expectError bool
+	}{
+		{
+			name:        "Invalid Proof",
+			blob:        validBlob,
+			proof:       invalidProof,
+			commitment:  validCommitment,
+			expectError: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			err := globalVerifier.VerifyBlobProof(
+				tc.blob,
+				tc.proof,
+				tc.commitment,
+			)
+			if tc.expectError {
+				require.Error(t, err, "invalid proof")
+			} else {
+				require.NoError(t, err)
+			}
+		})
+	}
+}
diff --git a/testing/files/test_data_incorrect_proof.json b/testing/files/test_data_incorrect_proof.json