chore: add getter function for

ohmzeus · Sep 27, 2023 · 76262a5 · 76262a5
1 parent 0da529d
commit 76262a5
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/CoolerFactory.sol b/src/CoolerFactory.sol
@@ -42,8 +42,7 @@ contract CoolerFactory {
     mapping(address => bool) public created;
 
     /// @notice Mapping to prevent duplicate coolers.
-    mapping(address => mapping(ERC20 => mapping(ERC20 => address)))
-        private coolerFor;
+    mapping(address => mapping(ERC20 => mapping(ERC20 => address))) private coolerFor;
 
     /// @notice Mapping to query Coolers for Collateral-Debt pair.
     mapping(ERC20 => mapping(ERC20 => address[])) public coolersFor;
@@ -120,4 +119,11 @@ function generateCooler(ERC20 collateral_, ERC20 debt_) external returns (addres
     function logDefaultLoan(uint256 loanID_, uint256 collateral_) external onlyFromFactory {
         emit DefaultLoan(msg.sender, loanID_, collateral_);
     }
+
+    // --- AUX FUNCTIONS ---------------------------------------------
+
+    /// @notice Getter function to get an existing cooler for a given user <> collateral <> debt combination.
+    function getCoolerFor(address user_, address collateral_, address debt_) public view returns (address) {
+        return coolerFor[user_][ERC20(collateral_)][ERC20(debt_)];
+    }
 }
diff --git a/src/test/CoolerFactory.t.sol b/src/test/CoolerFactory.t.sol
@@ -55,6 +55,7 @@ contract CoolerFactoryTest is Test {
     }
 
     // -- CoolerFactory Functions -------------------------------------------------
+
     function test_generateCooler() public {
 
         vm.startPrank(alice);
@@ -141,4 +142,14 @@ contract CoolerFactoryTest is Test {
         vm.expectRevert(CoolerFactory.NotFromFactory.selector);
         coolerFactory.logRequestLoan(id);
     }
+
+
+    function test_getCoolerFor() public {
+        // Unexistent loans return address(0).
+        assertEq(address(0), coolerFactory.getCoolerFor(alice, address(collateral), address(debt)));
+
+        vm.startPrank(alice);
+        address coolerAlice = coolerFactory.generateCooler(collateral, debt);
+        assertEq(coolerAlice, coolerFactory.getCoolerFor(alice, address(collateral), address(debt)));
+    }
 }