chore: Fix deprecated test stuff.

laina-defi · Jan 15, 2025 · 66b8168 · 66b8168
1 parent a88bf51
commit 66b8168
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/contracts/loan_manager/src/contract.rs b/contracts/loan_manager/src/contract.rs
@@ -571,7 +571,8 @@ mod tests {
     #[test]
     fn initialize() {
         let e = Env::default();
-        e.budget().reset_default();
+        // e.budget().reset_default();
+        e.cost_estimate();
         let admin = Address::generate(&e);
 
         let contract_id = e.register(LoanManager, ());
@@ -583,7 +584,6 @@ mod tests {
     #[test]
     fn cannot_re_initialize() {
         let e = Env::default();
-        e.budget().reset_default();
         let admin = Address::generate(&e);
 
         let contract_id = e.register(LoanManager, ());
@@ -598,7 +598,6 @@ mod tests {
     fn deploy_pool() {
         // ARRANGE
         let e = Env::default();
-        e.budget().reset_default();
         e.mock_all_auths();
 
         let admin = Address::generate(&e);
@@ -631,7 +630,6 @@ mod tests {
     fn upgrade_manager_and_pool() {
         // ARRANGE
         let e = Env::default();
-        e.budget().reset_default();
         e.mock_all_auths();
 
         let admin = Address::generate(&e);
@@ -656,7 +654,6 @@ mod tests {
     fn create_loan() {
         // ARRANGE
         let e = Env::default();
-        e.budget().reset_default();
         e.mock_all_auths_allowing_non_root_auth();
 
         let admin = Address::generate(&e);
@@ -719,7 +716,6 @@ mod tests {
         // ARRANGE
         let e = Env::default();
         e.mock_all_auths_allowing_non_root_auth();
-        e.budget().reset_unlimited();
         e.ledger().with_mut(|li| {
             li.sequence_number = 100_000;
             li.timestamp = 1;
@@ -811,7 +807,6 @@ mod tests {
     fn repay() {
         // ARRANGE
         let e = Env::default();
-        e.budget().reset_unlimited();
         e.mock_all_auths_allowing_non_root_auth();
 
         let admin = Address::generate(&e);
@@ -886,7 +881,6 @@ mod tests {
     fn repay_more_than_borrowed() {
         // ARRANGE
         let e = Env::default();
-        e.budget().reset_unlimited();
         e.mock_all_auths_allowing_non_root_auth();
 
         let admin = Address::generate(&e);
@@ -945,8 +939,6 @@ mod tests {
     fn liquidate() {
         // ARRANGE
         let e = Env::default();
-        e.budget().reset_unlimited();
-        e.budget().print();
         e.mock_all_auths_allowing_non_root_auth();
         e.ledger().with_mut(|li| {
             li.sequence_number = 100_000;
@@ -1048,6 +1040,5 @@ mod tests {
         assert_eq!(user_loan.borrowed_amount, 7_998);
         assert_eq!(user_loan.health_factor, 8_440_860);
         assert_eq!(user_loan.collateral_amount, 6_751);
-        e.budget().print();
     }
 }