🤖 Add Airdrop claimReward() test

Author: KimlikDAO-bot

mina/examples/Airdrop.test.ts

@@ -1,10 +1,12 @@
 import {
   AccountUpdate,
+  Field,
   MerkleTree,
   Mina,
   PrivateKey
 } from "o1js";
 import { Airdrop } from "./Airdrop";
+import { HumanIDWitness } from "../src/humanIDv1";
 
 describe('Example Airdrop zkApp', () => {
   const deployerKey = PrivateKey.random();
@@ -13,35 +15,39 @@ describe('Example Airdrop zkApp', () => {
   const sender = senderKey.toPublicKey();
   const appKey = PrivateKey.random();
   const appAddr = appKey.toPublicKey();
+  let tree: MerkleTree;
   let app: Airdrop;
 
   beforeAll(() => Airdrop.compile());
 
   beforeEach(() => Mina.LocalBlockchain({ proofsEnabled: true })
     .then((local) => {
+      tree = new MerkleTree(32);
       Mina.setActiveInstance(local);
       app = new Airdrop(appAddr);
       local.addAccount(deployer, "1000000000");
       local.addAccount(sender, "1000000000");
     }));
 
-  it('should deploy the app', async () => {
-    const tree = new MerkleTree(32);
-
-    await Mina.transaction(deployer, async () => {
-      AccountUpdate.fundNewAccount(deployer);
-      return app.deploy()
-        .then(() => app.initRoot(tree.getRoot()));
-    }).then((txn) => txn.prove())
-      .then((txn) => txn.sign([deployerKey, appKey]).send())
-      .then((txn) => txn.wait());
+  const deploy = () => Mina.transaction(deployer, async () => {
+    AccountUpdate.fundNewAccount(deployer);
+    return app.deploy()
+      .then(() => app.initRoot(tree.getRoot()));
+  }).then((txn) => txn.prove())
+    .then((txn) => txn.sign([deployerKey, appKey]).send())
 
+  it('should deploy the app', async () => {
+    await deploy();
     console.log('Deployed HumanIDs contract at', app.address);
   });
 
   it('should let people claimReward()', async () => {
-    const tree = new MerkleTree(32);
+    await deploy();
 
-    // TODO
+    await Mina.transaction(sender, () => {
+      let zkApp = new Airdrop(appAddr);
+      return zkApp.claimReward(Field(100), new HumanIDWitness(tree.getWitness(100n)));
+    }).then((txn) => txn.prove())
+      .then((txn) => txn.sign([senderKey]).send());
   })
 });

mina/src/humanIDv1.ts

@@ -1,4 +1,4 @@
-import { State, Field, MerkleWitness } from "o1js";
+import { Field, MerkleWitness, State } from "o1js";
 
 class HumanIDWitness extends MerkleWitness(32) { }
 
@@ -8,4 +8,5 @@ const authenticate = (treeRoot: State<Field>, humanIDv1: Field, witness: HumanID
   treeRoot.set(witness.calculateRoot(Field(1)));
 }
 
-export { authenticate, HumanIDWitness };
+export { HumanIDWitness, authenticate };
+