From f37f59c5fb8adf97ce05067a2307b6b8ce700330 Mon Sep 17 00:00:00 2001
From: liyaxing <liyaxing@buckyos.com>
Date: Thu, 20 Jun 2024 14:47:52 +0800
Subject: [PATCH] Update the test case for dividend contract for external
 deposit

---
 test/test_dividend2.ts | 56 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/test/test_dividend2.ts b/test/test_dividend2.ts
index 7f66a8f..c23cb28 100644
--- a/test/test_dividend2.ts
+++ b/test/test_dividend2.ts
@@ -642,4 +642,60 @@ describe("Devidend", function () {
             ).to.equal(true);
         }
     });
+
+    it("test transfer external", async () => {
+        const current_index = await dividend.getCurrentCycleIndex();
+        const current_balance = await dividend.getDepositTokenBalance(await gwt.getAddress());
+
+        mine(2, { interval: 1000 });
+        expect(await dividend.getCurrentCycleIndex()).to.equal(current_index);
+
+        {
+            // Try update the token balance, balance not change so will not step to next cycle
+            await (await dividend.updateTokenBalance(await gwt.getAddress())).wait();
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index);
+            expect(await dividend.getDepositTokenBalance(gwt.getAddress())).to.equal(
+                current_balance
+            );
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index);
+
+
+            // Deposit 90 GWT, will step to next cycle
+            await (await dividend.deposit(90, await gwt.getAddress())).wait();
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 1n);
+
+            // Update token balance, balance not changed so will not step to next cycle
+            await (await dividend.updateTokenBalance(await gwt.getAddress())).wait();
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 1n);
+            expect(await dividend.getDepositTokenBalance(gwt.getAddress())).to.equal(
+                current_balance + 90n
+            );
+
+            mine(2, { interval: 1000 });
+
+            // direct transfer to dividend contract and update token balance, will step to next cycle
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 1n);
+            await (await gwt.transfer(await dividend.getAddress(), 10)).wait();
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 1n);
+
+            // Update token balance, balance changed so will step to next cycle
+            await (await dividend.updateTokenBalance(await gwt.getAddress())).wait();
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 2n);
+            expect(await dividend.getDepositTokenBalance(gwt.getAddress())).to.equal(
+                current_balance + 100n
+            );
+
+            mine(2, { interval: 1000 });
+
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 2n);
+
+            // Update token balance, balance changed so will step to next cycle
+            await (await dividend.updateTokenBalance(await gwt.getAddress())).wait();
+            expect(await dividend.getCurrentCycleIndex()).to.equal(current_index + 2n);
+
+            expect(await dividend.getDepositTokenBalance(gwt.getAddress())).to.equal(
+                current_balance + 100n
+            );
+        }
+    });
 });