fix: check collateral exists on chainlink set (#917)

ubiquity · Mar 20, 2024 · f1144a8 · f1144a8
1 parent 898d45c
commit f1144a8
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/packages/contracts/src/dollar/libraries/LibUbiquityPool.sol b/packages/contracts/src/dollar/libraries/LibUbiquityPool.sol
@@ -750,6 +750,11 @@ library LibUbiquityPool {
         address chainLinkPriceFeedAddress,
         uint256 stalenessThreshold
     ) internal {
+        require(
+            collateralExists(collateralAddress),
+            "Collateral does not exist"
+        );
+
         UbiquityPoolStorage storage poolStorage = ubiquityPoolStorage();
 
         uint256 collateralIndex = poolStorage.collateralIndex[

diff --git a/packages/contracts/test/diamond/facets/UbiquityPoolFacet.t.sol b/packages/contracts/test/diamond/facets/UbiquityPoolFacet.t.sol
@@ -831,6 +831,21 @@ contract UbiquityPoolFacetTest is DiamondTestSetup {
         vm.stopPrank();
     }
 
+    function testSetCollateralChainLinkPriceFeed_ShouldRevertIfCollateralDoesNotExist()
+        public
+    {
+        vm.prank(admin);
+        vm.expectRevert("Collateral does not exist");
+        address invalidCollateralAddress = address(0);
+        address newPriceFeedAddress = address(1);
+        uint256 newStalenessThreshold = 1 days;
+        ubiquityPoolFacet.setCollateralChainLinkPriceFeed(
+            invalidCollateralAddress,
+            newPriceFeedAddress,
+            newStalenessThreshold
+        );
+    }
+
     function testSetCollateralChainLinkPriceFeed_ShouldSetPriceFeed() public {
         vm.startPrank(admin);