From 40d719d83c759b05707e3eb15f822766da751b54 Mon Sep 17 00:00:00 2001
From: ungaro <alp.guneysel@gmail.com>
Date: Thu, 26 Dec 2024 23:14:52 -0500
Subject: [PATCH] WIP AtomicQueue tests

---
 nest/src/vault/NestAtomicQueue.sol |   1 -
 nest/test/NestAtomicQueue.t.sol    | 136 +++++++++++++++++++++++++++++
 2 files changed, 136 insertions(+), 1 deletion(-)
 create mode 100644 nest/test/NestAtomicQueue.t.sol

diff --git a/nest/src/vault/NestAtomicQueue.sol b/nest/src/vault/NestAtomicQueue.sol
index a7250f2..9ec0f96 100644
--- a/nest/src/vault/NestAtomicQueue.sol
+++ b/nest/src/vault/NestAtomicQueue.sol
@@ -11,7 +11,6 @@ import { FixedPointMathLib } from "@solmate/utils/FixedPointMathLib.sol";
 import { AtomicQueue } from "@boringvault/src/atomic-queue/AtomicQueue.sol";
 
 import { IAtomicQueue } from "../interfaces/IAtomicQueue.sol";
-import { IComponentToken } from "../interfaces/IComponentToken.sol";
 import { NestBoringVaultModule } from "./NestBoringVaultModule.sol";
 
 /**
diff --git a/nest/test/NestAtomicQueue.t.sol b/nest/test/NestAtomicQueue.t.sol
new file mode 100644
index 0000000..b11e953
--- /dev/null
+++ b/nest/test/NestAtomicQueue.t.sol
@@ -0,0 +1,136 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.25;
+
+import { NestAtomicQueue } from "../src/vault/NestAtomicQueue.sol";
+import { NestBoringVaultModule } from "../src/vault/NestBoringVaultModule.sol";
+import { NestBoringVaultModuleTest } from "./NestBoringVaultModuleTest.t.sol";
+import { AtomicQueue } from "@boringvault/src/atomic-queue/AtomicQueue.sol";
+import { ERC20 } from "@solmate/tokens/ERC20.sol";
+
+import { SafeCast } from "@openzeppelin/contracts/utils/math/SafeCast.sol";
+
+import { FixedPointMathLib } from "@solmate/utils/FixedPointMathLib.sol";
+
+// Add event definitions at the top of the contract
+event AtomicRequestUpdated(
+    address user,
+    address offerToken,
+    address wantToken,
+    uint256 amount,
+    uint256 deadline,
+    uint256 minPrice,
+    uint256 timestamp
+);
+
+contract NestAtomicQueueTest is NestBoringVaultModuleTest {
+
+    NestAtomicQueue public queue;
+
+    uint256 public constant DEADLINE_PERIOD = 1 days;
+    uint256 public constant PRICE_PERCENTAGE = 9900; // 99.00%
+
+    using SafeCast for uint256;
+    using FixedPointMathLib for uint256;
+
+    function setUp() public override {
+        super.setUp();
+
+        // Deploy NestAtomicQueue
+        queue = new NestAtomicQueue(
+            address(this), address(vault), address(accountant), asset, DEADLINE_PERIOD, PRICE_PERCENTAGE
+        );
+    }
+
+    function testInitialization() public override {
+        assertEq(queue.owner(), address(this));
+        assertEq(address(queue.vaultContract()), address(vault));
+        assertEq(address(queue.accountantContract()), address(accountant));
+        assertEq(address(queue.assetToken()), address(asset));
+        assertEq(queue.deadlinePeriod(), DEADLINE_PERIOD);
+        assertEq(queue.pricePercentage(), PRICE_PERCENTAGE);
+    }
+
+    function testRequestRedeem(
+        uint256 shares
+    ) public {
+        // Bound shares to reasonable values
+        shares = bound(shares, 1e6, 1_000_000e6);
+
+        // Give user some shares in the vault
+        deal(address(vault), user, shares);
+
+        // Start as user
+        vm.startPrank(user);
+
+        // Approve queue to spend shares
+        vault.approve(address(queue), shares);
+
+        // Calculate expected deadline and atomic price
+        uint256 expectedDeadline = block.timestamp + DEADLINE_PERIOD;
+        uint256 expectedAtomicPrice =
+            accountant.getRateInQuote(ERC20(address(asset))).mulDivDown(PRICE_PERCENTAGE, 10_000);
+
+        // Expect AtomicRequestUpdated event
+        vm.expectEmit(true, true, true, true);
+        emit AtomicRequestUpdated(
+            user, // user
+            address(vault), // offerToken
+            address(asset), // wantToken
+            shares, // amount
+            expectedDeadline, // deadline
+            expectedAtomicPrice, // minPrice
+            block.timestamp // timestamp
+        );
+
+        // Make redeem request
+        uint256 requestId = queue.requestRedeem(shares, user, user);
+
+        // Get request details using getUserAtomicRequest
+        AtomicQueue.AtomicRequest memory request =
+            queue.getUserAtomicRequest(user, ERC20(address(vault)), ERC20(address(asset)));
+
+        // Verify request details
+        assertEq(request.offerAmount, shares);
+        assertEq(request.deadline, expectedDeadline);
+        assertEq(request.atomicPrice, expectedAtomicPrice);
+        assertEq(request.inSolve, false);
+
+        vm.stopPrank();
+    }
+
+    function testRequestRedeemZeroShares() public {
+        vm.expectRevert(NestBoringVaultModule.InvalidAmount.selector);
+        queue.requestRedeem(0, user, user);
+    }
+
+    function testRequestRedeemInsufficientBalance(
+        uint256 shares
+    ) public {
+        shares = bound(shares, 1e6, 1_000_000e6);
+
+        vm.startPrank(user);
+        vm.expectRevert("TRANSFER_FROM_FAILED");
+        queue.requestRedeem(shares, user, user);
+        vm.stopPrank();
+    }
+
+    function testRequestRedeemInsufficientAllowance(
+        uint256 shares
+    ) public {
+        shares = bound(shares, 1e6, 1_000_000e6);
+
+        // Give user shares but don't approve queue
+        deal(address(vault), user, shares);
+
+        vm.startPrank(user);
+        vm.expectRevert("TRANSFER_FROM_FAILED");
+        queue.requestRedeem(shares, user, user);
+        vm.stopPrank();
+    }
+
+    function testDepositReverts() public {
+        vm.expectRevert(NestBoringVaultModule.Unimplemented.selector);
+        queue.deposit(1e6, address(this), address(this));
+    }
+
+}