From ed2e79dd3c9dd74b613e2d4bf2658566a4aa27e8 Mon Sep 17 00:00:00 2001
From: KOSASIH <kosasihg88@gmail.com>
Date: Thu, 31 Oct 2024 08:27:54 +0700
Subject: [PATCH] Create test_InvestmentPortfolio.js

---
 .../tests/test_InvestmentPortfolio.js         | 80 +++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 blockchain_integration/pi_network/tests/test_InvestmentPortfolio.js

diff --git a/blockchain_integration/pi_network/tests/test_InvestmentPortfolio.js b/blockchain_integration/pi_network/tests/test_InvestmentPortfolio.js
new file mode 100644
index 000000000..a4a9bf506
--- /dev/null
+++ b/blockchain_integration/pi_network/tests/test_InvestmentPortfolio.js
@@ -0,0 +1,80 @@
+// tests/test_InvestmentPortfolio.js
+const { expect } = require("chai");
+const { ethers } = require("hardhat");
+
+describe("InvestmentPortfolio", function () {
+    let InvestmentPortfolio;
+    let investmentPortfolio;
+    let owner;
+    let addr1;
+    let token;
+
+    beforeEach(async function () {
+        // Deploy a mock ERC20 token for testing
+        const Token = await ethers.getContractFactory("MockERC20");
+        token = await Token.deploy("Test Token", "TTK", ethers.utils.parseEther("1000"));
+        await token.deployed();
+
+        InvestmentPortfolio = await ethers.getContractFactory("InvestmentPortfolio");
+        [owner, addr1] = await ethers.getSigners();
+        investmentPortfolio = await InvestmentPortfolio.deploy();
+        await investmentPortfolio.deployed();
+    });
+
+    it("Should allow investment in the portfolio", async function () {
+        const investmentAmount = ethers.utils.parseEther("100.0");
+
+        // Approve the investment amount
+        await token.approve(investmentPortfolio.address, investmentAmount);
+        
+        // Invest in the portfolio
+        await investmentPortfolio.invest(token.address, investmentAmount);
+        
+        // Check the investment amount
+        expect(await investmentPortfolio.getInvestmentAmount(token.address)).to.equal(investmentAmount);
+    });
+
+    it("Should allow withdrawal of investment", async function () {
+        const investmentAmount = ethers.utils.parseEther("100.0");
+
+        // Approve and invest in the portfolio
+        await token.approve(investmentPortfolio.address, investmentAmount);
+        await investmentPortfolio.invest(token.address, investmentAmount);
+        
+        // Withdraw the investment
+        await investmentPortfolio.withdrawInvestment(token.address, investmentAmount);
+        
+        // Check the investment amount
+        expect(await investmentPortfolio.getInvestmentAmount(token.address)).to.equal(0);
+    });
+
+    it("Should not allow withdrawal of more than invested amount", async function () {
+        const investmentAmount = ethers.utils.parseEther("100.0");
+
+        // Approve and invest in the portfolio
+        await token.approve(investmentPortfolio.address, investmentAmount);
+        await investmentPortfolio.invest(token.address, investmentAmount);
+        
+        // Attempt to withdraw more than invested
+        await expect(investmentPortfolio.withdrawInvestment(token.address, ethers.utils.parseEther("200.0"))).to.be.revertedWith("Insufficient investment amount");
+    });
+
+    it("Should track multiple investments", async function () {
+        const investmentAmount1 = ethers.utils.parseEther("100.0");
+        const investmentAmount2 = ethers.utils.parseEther("200.0");
+
+        // Approve and invest in the portfolio
+        await token.approve(investmentPortfolio.address, investmentAmount1);
+        await investmentPortfolio.invest(token.address, investmentAmount1);
+        
+        // Approve and invest in another asset
+        const token2 = await token.deploy("Another Token", "ATK", ethers.utils.parseEther("1000"));
+        await token2.deployed();
+        await token2.approve(investmentPortfolio.address, investmentAmount2);
+        await investmentPortfolio.invest(token2.address, investmentAmount2);
+        
+        // Check the investment amounts
+        expect(await investmentPortfolio.getInvestmentAmount(token.address)).to.equal(investmentAmount1);
+        expect(await investmentPortfolio.getInvestmentAmount(token2.address)).to.equal(investmentAmount2);
+    });
+});