calculate balance differences

Signed-off-by: Jim Zhang <[email protected]>

solidity/test/zeto_anon.ts

@@ -61,10 +61,11 @@ describe("Zeto based fungible token with anonymity without encryption or nullifi
   });
 
   it("mint ERC20 tokens to Alice to deposit to Zeto should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     const tx = await erc20.connect(deployer).mint(Alice.ethAddress, 100);
     await tx.wait();
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(100);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(100);
 
     const tx1 = await erc20.connect(Alice.signer).approve(zeto.target, 100);
     await tx1.wait();
@@ -117,6 +118,8 @@ describe("Zeto based fungible token with anonymity without encryption or nullifi
   });
 
   it("Alice withdraws her UTXOs to ERC20 tokens should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
+
     // Alice proposes the output ERC20 tokens
     const outputCommitment = newUTXO(20, Alice);
 
@@ -127,8 +130,8 @@ describe("Zeto based fungible token with anonymity without encryption or nullifi
     await tx.wait();
 
     // Alice checks her ERC20 balance
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(80);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(80);
   });
 
   describe('failure cases', function () {

solidity/test/zeto_anon_enc.ts

@@ -58,10 +58,11 @@ describe("Zeto based fungible token with anonymity and encryption", function ()
   });
 
   it("mint ERC20 tokens to Alice to deposit to Zeto should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     const tx = await erc20.connect(deployer).mint(Alice.ethAddress, 100);
     await tx.wait();
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(100);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(100);
 
     const tx1 = await erc20.connect(Alice.signer).approve(zeto.target, 100);
     await tx1.wait();
@@ -114,6 +115,7 @@ describe("Zeto based fungible token with anonymity and encryption", function ()
   });
 
   it("Alice withdraws her UTXOs to ERC20 tokens should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     // Alice proposes the output ERC20 tokens
     const outputCommitment = newUTXO(20, Alice);
 
@@ -124,8 +126,8 @@ describe("Zeto based fungible token with anonymity and encryption", function ()
     await tx.wait();
 
     // Alice checks her ERC20 balance
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(80);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(80);
   });
 
   describe("failure cases", function () {

solidity/test/zeto_anon_enc_nullifier.ts

@@ -73,10 +73,11 @@ describe("Zeto based fungible token with anonymity using nullifiers and encrypti
   });
 
   it("mint ERC20 tokens to Alice to deposit to Zeto should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     const tx = await erc20.connect(deployer).mint(Alice.ethAddress, 100);
     await tx.wait();
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(100);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(100);
 
     const tx1 = await erc20.connect(Alice.signer).approve(zeto.target, 100);
     await tx1.wait();
@@ -179,6 +180,8 @@ describe("Zeto based fungible token with anonymity using nullifiers and encrypti
   }).timeout(600000);
 
   it("Alice withdraws her UTXOs to ERC20 tokens should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
+
     // Alice generates the nullifiers for the UTXOs to be spent
     const nullifier1 = newNullifier(utxo100, Alice);
 
@@ -198,8 +201,8 @@ describe("Zeto based fungible token with anonymity using nullifiers and encrypti
     await tx.wait();
 
     // Alice checks her ERC20 balance
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(80);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(80);
   });
 
   describe("failure cases", function () {

solidity/test/zeto_anon_enc_nullifier_non_repudiation.ts

@@ -78,10 +78,11 @@ describe("Zeto based fungible token with anonymity using nullifiers and encrypti
   });
 
   it("mint ERC20 tokens to Alice to deposit to Zeto should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     const tx = await erc20.connect(deployer).mint(Alice.ethAddress, 100);
     await tx.wait();
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(100);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(100);
 
     const tx1 = await erc20.connect(Alice.signer).approve(zeto.target, 100);
     await tx1.wait();
@@ -216,6 +217,7 @@ describe("Zeto based fungible token with anonymity using nullifiers and encrypti
   }).timeout(600000);
 
   it("Alice withdraws her UTXOs to ERC20 tokens should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     // Alice generates the nullifiers for the UTXOs to be spent
     const nullifier1 = newNullifier(utxo100, Alice);
 
@@ -235,8 +237,8 @@ describe("Zeto based fungible token with anonymity using nullifiers and encrypti
     await tx.wait();
 
     // Alice checks her ERC20 balance
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(80);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(80);
   });
 
   describe("failure cases", function () {

solidity/test/zeto_anon_nullifier.ts

@@ -73,10 +73,11 @@ describe("Zeto based fungible token with anonymity using nullifiers without encr
   });
 
   it("mint ERC20 tokens to Alice to deposit to Zeto should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     const tx = await erc20.connect(deployer).mint(Alice.ethAddress, 100);
     await tx.wait();
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(100);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(100);
 
     const tx1 = await erc20.connect(Alice.signer).approve(zeto.target, 100);
     await tx1.wait();
@@ -181,6 +182,8 @@ describe("Zeto based fungible token with anonymity using nullifiers without encr
   }).timeout(600000);
 
   it("Alice withdraws her UTXOs to ERC20 tokens should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
+
     // Alice generates the nullifiers for the UTXOs to be spent
     const nullifier1 = newNullifier(utxo100, Alice);
 
@@ -200,8 +203,8 @@ describe("Zeto based fungible token with anonymity using nullifiers without encr
     await tx.wait();
 
     // Alice checks her ERC20 balance
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(80);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(80);
   });
 
   describe("failure cases", function () {

solidity/test/zeto_anon_nullifier_kyc.ts

@@ -95,10 +95,11 @@ describe("Zeto based fungible token with anonymity, KYC, using nullifiers withou
   });
 
   it("mint ERC20 tokens to Alice to deposit to Zeto should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     const tx = await erc20.connect(deployer).mint(Alice.ethAddress, 100);
     await tx.wait();
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(100);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(100);
 
     const tx1 = await erc20.connect(Alice.signer).approve(zeto.target, 100);
     await tx1.wait();
@@ -226,6 +227,7 @@ describe("Zeto based fungible token with anonymity, KYC, using nullifiers withou
   }).timeout(600000);
 
   it("Alice withdraws her UTXOs to ERC20 tokens should succeed", async function () {
+    const startingBalance = await erc20.balanceOf(Alice.ethAddress);
     // Alice generates the nullifiers for the UTXOs to be spent
     const nullifier1 = newNullifier(utxo100, Alice);
 
@@ -244,8 +246,8 @@ describe("Zeto based fungible token with anonymity, KYC, using nullifiers withou
     await tx.wait();
 
     // Alice checks her ERC20 balance
-    const balance = await erc20.balanceOf(Alice.ethAddress);
-    expect(balance).to.be.gte(80);
+    const endingBalance = await erc20.balanceOf(Alice.ethAddress);
+    expect(endingBalance - startingBalance).to.be.equal(80);
   });
 
   describe("unregistered user flows", function () {
@@ -303,6 +305,7 @@ describe("Zeto based fungible token with anonymity, KYC, using nullifiers withou
     });
 
     it("the unregistered user can still withdraw their UTXOs to ERC20 tokens", async function () {
+      const startingBalance = await erc20.balanceOf(unregistered.ethAddress);
       // unregistered user generates the nullifiers for the UTXOs to be spent
       const nullifier1 = newNullifier(unregisteredUtxo100, unregistered);
 
@@ -322,8 +325,8 @@ describe("Zeto based fungible token with anonymity, KYC, using nullifiers withou
       await tx.wait();
 
       // unregistered user checks her ERC20 balance
-      const balance = await erc20.balanceOf(unregistered.ethAddress);
-      expect(balance).to.be.gte(100);
+      const endingBalance = await erc20.balanceOf(unregistered.ethAddress);
+      expect(endingBalance - startingBalance).to.be.equal(100);
     });
   });