chore(benchmarks): add redact shield tests

Author: NicoSerranoP

…ests/railgun-smart-wallet-shield.test.ts …lgun/railgun-smart-wallet-shield.test.tssubgraph/tests/railgun-smart-wallet-shield.test.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-shield.test.ts subgraph/tests/railgun-smart-wallet-shield.test.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-shield.test.ts

@@ -1,7 +1,7 @@
 import { Address, BigInt, Bytes } from "@graphprotocol/graph-ts";
 import { assert, describe, test, clearStore, beforeAll, afterAll } from "matchstick-as/assembly/index";
 
-import { handleShield } from "../src/railgun/railgun-smart-wallet";
+import { handleShield } from "../../src/railgun/railgun-smart-wallet";
 
 import { createShieldEvent, createCommitmentTuple, createShieldCiphertextTuple } from "./railgun-smart-wallet-utils";
 

…ts/railgun-smart-wallet-transact.test.ts …un/railgun-smart-wallet-transact.test.tssubgraph/tests/railgun-smart-wallet-transact.test.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-transact.test.ts subgraph/tests/railgun-smart-wallet-transact.test.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-transact.test.ts

@@ -1,7 +1,7 @@
 import { BigInt, Bytes } from "@graphprotocol/graph-ts";
 import { assert, describe, test, clearStore, beforeAll, afterAll } from "matchstick-as/assembly/index";
 
-import { handleTransact } from "../src/railgun/railgun-smart-wallet";
+import { handleTransact } from "../../src/railgun/railgun-smart-wallet";
 
 import { createTransactEvent, createTransactCiphertextTuple } from "./railgun-smart-wallet-utils";
 

…ts/railgun-smart-wallet-unshield.test.ts …un/railgun-smart-wallet-unshield.test.tssubgraph/tests/railgun-smart-wallet-unshield.test.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-unshield.test.ts subgraph/tests/railgun-smart-wallet-unshield.test.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-unshield.test.ts

@@ -1,7 +1,7 @@
 import { Address, BigInt } from "@graphprotocol/graph-ts";
 import { assert, describe, test, clearStore, beforeAll, afterAll } from "matchstick-as/assembly/index";
 
-import { handleUnshield } from "../src/railgun/railgun-smart-wallet";
+import { handleUnshield } from "../../src/railgun/railgun-smart-wallet";
 
 import { createUnshieldEvent } from "./railgun-smart-wallet-utils";
 

…raph/tests/railgun-smart-wallet-utils.ts …ts/railgun/railgun-smart-wallet-utils.tssubgraph/tests/railgun-smart-wallet-utils.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-utils.ts subgraph/tests/railgun-smart-wallet-utils.ts renamed to subgraph/tests/railgun/railgun-smart-wallet-utils.ts

@@ -1,7 +1,7 @@
 import { ethereum, BigInt, Address, Bytes } from "@graphprotocol/graph-ts";
 import { newMockEvent } from "matchstick-as";
 
-import { Shield, Transact, Unshield } from "../generated/RailgunSmartWallet/RailgunSmartWallet";
+import { Shield, Transact, Unshield } from "../../generated/RailgunSmartWallet/RailgunSmartWallet";
 
 export function createShieldEvent(
   treeNumber: BigInt,

subgraph/tests/redact/redact-confidential-eth-shield.test.ts

@@ -0,0 +1,54 @@
+import { Address, BigInt } from "@graphprotocol/graph-ts";
+import { assert, describe, test, clearStore, beforeAll, afterAll } from "matchstick-as/assembly/index";
+
+import { handleShieldedNative } from "../../src/redact/confidential-eth";
+
+import { createShieldedNativeEvent } from "./redact-confidential-eth-utils";
+
+describe("ShieldedNative event tests", () => {
+  beforeAll(() => {
+    const event = createShieldedNativeEvent(
+      Address.fromString("0x0000000000000000000000000000000000000001"),
+      Address.fromString("0x0000000000000000000000000000000000000002"),
+      BigInt.fromI32(100),
+    );
+
+    handleShieldedNative(event);
+  });
+
+  afterAll(() => {
+    clearStore();
+  });
+
+  test("RedactShieldedNative created", () => {
+    assert.entityCount("RedactShieldedNative", 1);
+  });
+
+  test("RedactShieldedNative values stored correctly", () => {
+    const id = "0xa16081f360e3847006db660bae1c6d1b2e17ec2a-1";
+
+    assert.fieldEquals("RedactShieldedNative", id, "from", "0x0000000000000000000000000000000000000001");
+    assert.fieldEquals("RedactShieldedNative", id, "to", "0x0000000000000000000000000000000000000002");
+    assert.fieldEquals("RedactShieldedNative", id, "value", "100");
+  });
+
+  test("Protocol stats updated", () => {
+    assert.fieldEquals("RedactProtocolStats", "redact-protocol-stats", "totalTxCount", "1");
+    assert.fieldEquals("RedactProtocolStats", "redact-protocol-stats", "totalGasUsed", "1");
+  });
+
+  test("Operation stats updated (shield-native)", () => {
+    assert.fieldEquals("RedactOperationStats", "redact-protocol-stats-shield-native", "totalCount", "1");
+    assert.fieldEquals("RedactOperationStats", "redact-protocol-stats-shield-native", "totalGasUsed", "1");
+  });
+
+  test("Shielded native token stats updated", () => {
+    const id = "redact-protocol-stats-shield-native";
+
+    assert.entityCount("RedactShieldedNativeStats", 1);
+    assert.fieldEquals("RedactShieldedNativeStats", id, "tokenAddress", "0x0000000000000000000000000000000000000000");
+    assert.fieldEquals("RedactShieldedNativeStats", id, "totalValue", "100");
+    assert.fieldEquals("RedactShieldedNativeStats", id, "totalCount", "1");
+    assert.fieldEquals("RedactShieldedNativeStats", id, "totalGasUsed", "1");
+  });
+});

subgraph/tests/redact/redact-confidential-eth-utils.ts

@@ -0,0 +1,15 @@
+import { ethereum, Address, BigInt } from "@graphprotocol/graph-ts";
+import { newMockEvent } from "matchstick-as";
+
+import { ShieldedNative } from "../../generated/ConfidentialETH/ConfidentialETH";
+
+export function createShieldedNativeEvent(from: Address, to: Address, value: BigInt): ShieldedNative {
+  const event = changetype<ShieldedNative>(newMockEvent());
+  event.parameters = [];
+
+  event.parameters.push(new ethereum.EventParam("from", ethereum.Value.fromAddress(from)));
+  event.parameters.push(new ethereum.EventParam("to", ethereum.Value.fromAddress(to)));
+  event.parameters.push(new ethereum.EventParam("value", ethereum.Value.fromUnsignedBigInt(value)));
+
+  return event;
+}