From 8b799d2a56dea9a2be91769a02c8f00f9ceb5233 Mon Sep 17 00:00:00 2001
From: 0o-de-lally <1364012+0o-de-lally@users.noreply.github.com>
Date: Fri, 27 Sep 2024 16:01:44 -0400
Subject: [PATCH] patch test_epoch_drip

---
 .../sources/ol_sources/tests/slow_wallet.test.move        | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/framework/libra-framework/sources/ol_sources/tests/slow_wallet.test.move b/framework/libra-framework/sources/ol_sources/tests/slow_wallet.test.move
index 06ff3027f..078302378 100644
--- a/framework/libra-framework/sources/ol_sources/tests/slow_wallet.test.move
+++ b/framework/libra-framework/sources/ol_sources/tests/slow_wallet.test.move
@@ -5,7 +5,7 @@ module ol_framework::test_slow_wallet {
   use diem_framework::stake;
   use diem_framework::account;
   use ol_framework::slow_wallet;
-  use ol_framework::mock;
+  use ol_framework::mock::{Self, default_tx_fee_account_at_genesis};
   use ol_framework::ol_account;
   use ol_framework::libra_coin;
   use ol_framework::epoch_boundary;
@@ -112,6 +112,7 @@ module ol_framework::test_slow_wallet {
   fun test_epoch_drip(root: signer) {
     let set = mock::genesis_n_vals(&root, 4);
     mock::ol_initialize_coin_and_fund_vals(&root, 100, false);
+    mock::mock_tx_fees_in_account(&root, default_tx_fee_account_at_genesis());
 
     let a = *vector::borrow(&set, 0);
     assert!(slow_wallet::is_slow(a), 7357000);
@@ -122,12 +123,13 @@ module ol_framework::test_slow_wallet {
 
     let (u, b) = ol_account::balance(a);
     print(&b);
-    assert!(b==500_000_100, 735702);
+    assert!(b==(default_tx_fee_account_at_genesis() + 100), 735702);
     assert!(u==100, 735703);
 
     slow_wallet::slow_wallet_epoch_drip(&root, 233);
     let (u, b) = ol_account::balance(a);
-    assert!(b==500_000_100, 735704);
+    // no change total balances
+    assert!(b==(default_tx_fee_account_at_genesis() + 100), 735704);
     assert!(u==333, 735705);
   }