feat: Update drawn rate in refresh premium (#716)

CheyenneAtapour · web-flow · commit 422aee7acbaf · 2025-09-12T10:27:21.000+02:00
diff --git a/snapshots/Hub.Operations.json b/snapshots/Hub.Operations.json
@@ -1,7 +1,7 @@
 {
   "add": "119571",
   "draw": "112344",
-  "refreshPremium": "84297",
+  "refreshPremium": "100848",
   "remove: full": "80336",
   "remove: partial": "86836",
   "restore: full": "114054",
diff --git a/snapshots/Spoke.Operations.json b/snapshots/Spoke.Operations.json
@@ -1,13 +1,13 @@
 {
-  "borrow: first": "250505",
-  "borrow: second action, same reserve": "213977",
-  "liquidationCall: full": "321346",
-  "liquidationCall: partial": "345017",
-  "permitReserve + repay (multicall)": "273361",
+  "borrow: first": "257756",
+  "borrow: second action, same reserve": "221228",
+  "liquidationCall: full": "328597",
+  "liquidationCall: partial": "352268",
+  "permitReserve + repay (multicall)": "280612",
   "permitReserve + supply (multicall)": "140631",
   "permitReserve + supply + enable collateral (multicall)": "174116",
   "repay: full": "175218",
-  "repay: partial": "232122",
+  "repay: partial": "239373",
   "setUserPositionManagerWithSig: disable": "38977",
   "setUserPositionManagerWithSig: enable": "63000",
   "supply + enable collateral (multicall)": "152550",
@@ -17,12 +17,12 @@
   "supply: second action, same reserve": "102491",
   "updateUserDynamicConfig: 1 collateral": "53152",
   "updateUserDynamicConfig: 2 collaterals": "58433",
-  "updateUserRiskPremium: 1 borrow": "123817",
-  "updateUserRiskPremium: 2 borrows": "162132",
+  "updateUserRiskPremium: 1 borrow": "143568",
+  "updateUserRiskPremium: 2 borrows": "195134",
   "usingAsCollateral: 0 borrows, enable": "54026",
-  "usingAsCollateral: 1 borrow, disable": "134043",
+  "usingAsCollateral: 1 borrow, disable": "153794",
   "usingAsCollateral: 1 borrow, enable": "24786",
   "withdraw: 0 borrows, full": "135168",
   "withdraw: 0 borrows, partial": "137271",
-  "withdraw: 1 borrow, partial": "240391"
+  "withdraw: 1 borrow, partial": "254442"
 }
diff --git a/src/contracts/Hub.sol b/src/contracts/Hub.sol
@@ -337,6 +337,8 @@ contract Hub is IHub, AccessManaged {
     // no premium change allowed
     _applyPremiumDelta(assetId, asset, spoke, premiumDelta, 0);
 
+    asset.updateDrawnRate(assetId);
+
     emit RefreshPremium(assetId, msg.sender, premiumDelta);
   }
 
diff --git a/tests/unit/Hub/Hub.RefreshPremium.t.sol b/tests/unit/Hub/Hub.RefreshPremium.t.sol
@@ -43,6 +43,7 @@ contract HubRefreshPremiumTest is HubBase {
       _applyPremiumDelta(premiumDataBefore, premiumDelta)
     );
     assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');
+    assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');
   }
 
   /// @dev offsetDelta can't be more than sharesDelta or else underflow
@@ -84,6 +85,7 @@ contract HubRefreshPremiumTest is HubBase {
         _applyPremiumDelta(premiumDataBefore, premiumDelta)
       );
       assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');
+      assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');
     }
   }
 
@@ -118,6 +120,7 @@ contract HubRefreshPremiumTest is HubBase {
       _applyPremiumDelta(premiumDataBefore, premiumDelta)
     );
     assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');
+    assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');
   }
 
   function test_refreshPremium_negativeDeltas_withAccrual(
@@ -180,6 +183,7 @@ contract HubRefreshPremiumTest is HubBase {
         _applyPremiumDelta(premiumDataBefore, premiumDelta)
       );
       assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');
+      assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');
     }
   }
 
@@ -252,6 +256,7 @@ contract HubRefreshPremiumTest is HubBase {
         _applyPremiumDelta(premiumDataBefore, premiumDelta)
       );
       assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');
+      assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"add": "119571",`
`3`	`3`	`"draw": "112344",`
`4`		`- "refreshPremium": "84297",`
	`4`	`+ "refreshPremium": "100848",`
`5`	`5`	`"remove: full": "80336",`
`6`	`6`	`"remove: partial": "86836",`
`7`	`7`	`"restore: full": "114054",`
Original file line number	Diff line number	Diff line change
`@@ -337,6 +337,8 @@ contract Hub is IHub, AccessManaged {`
`337`	`337`	`// no premium change allowed`
`338`	`338`	`_applyPremiumDelta(assetId, asset, spoke, premiumDelta, 0);`
`339`	`339`
	`340`	`+ asset.updateDrawnRate(assetId);`
	`341`	`+`
`340`	`342`	`emit RefreshPremium(assetId, msg.sender, premiumDelta);`
`341`	`343`	`}`
`342`	`344`
Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,7 @@ contract HubRefreshPremiumTest is HubBase {`
`43`	`43`	`_applyPremiumDelta(premiumDataBefore, premiumDelta)`
`44`	`44`	`);`
`45`	`45`	`assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');`
	`46`	`+ assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');`
`46`	`47`	`}`
`47`	`48`
`48`	`49`	`/// @dev offsetDelta can't be more than sharesDelta or else underflow`
`@@ -84,6 +85,7 @@ contract HubRefreshPremiumTest is HubBase {`
`84`	`85`	`_applyPremiumDelta(premiumDataBefore, premiumDelta)`
`85`	`86`	`);`
`86`	`87`	`assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');`
	`88`	`+ assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');`
`87`	`89`	`}`
`88`	`90`	`}`
`89`	`91`
`@@ -118,6 +120,7 @@ contract HubRefreshPremiumTest is HubBase {`
`118`	`120`	`_applyPremiumDelta(premiumDataBefore, premiumDelta)`
`119`	`121`	`);`
`120`	`122`	`assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');`
	`123`	`+ assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');`
`121`	`124`	`}`
`122`	`125`
`123`	`126`	`function test_refreshPremium_negativeDeltas_withAccrual(`
`@@ -180,6 +183,7 @@ contract HubRefreshPremiumTest is HubBase {`
`180`	`183`	`_applyPremiumDelta(premiumDataBefore, premiumDelta)`
`181`	`184`	`);`
`182`	`185`	`assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');`
	`186`	`+ assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');`
`183`	`187`	`}`
`184`	`188`	`}`
`185`	`189`
`@@ -252,6 +256,7 @@ contract HubRefreshPremiumTest is HubBase {`
`252`	`256`	`_applyPremiumDelta(premiumDataBefore, premiumDelta)`
`253`	`257`	`);`
`254`	`258`	`assertLe(premiumAfter - premiumBefore, 2, 'premium should not increase by more than 2');`
	`259`	`+ assertBorrowRateSynced(hub1, daiAssetId, 'after refreshPremium');`
`255`	`260`	`}`
`256`	`261`	`}`
`257`	`262`