From 46844c076447d479dde45bbe4e1ae36f83ac805b Mon Sep 17 00:00:00 2001
From: JSHan94 <saw151515@gmail.com>
Date: Wed, 27 Mar 2024 10:14:11 +0900
Subject: [PATCH] fix reward per stage

---
 precompile/binaries/stdlib/vip_vault.mv       | Bin 1565 -> 1566 bytes
 .../initia_stdlib/sources/vip/vault.move      |  38 +-----------------
 .../initia_stdlib/sources/vip/vip.move        |   1 +
 3 files changed, 3 insertions(+), 36 deletions(-)

diff --git a/precompile/binaries/stdlib/vip_vault.mv b/precompile/binaries/stdlib/vip_vault.mv
index 686e3993324c32c6071bce1994d57b0dd3498d88..c9909d4ab0f05d7d8b5fcbfde41f37af48958d7f 100644
GIT binary patch
delta 32
ocmbQsGmmG2F~>tD-fP?({P#9m&tv6aWoOf75MUIU{D9RK0GzW3lK=n!

delta 31
ncmbQoGnZ$AF~<WY-mBak{C786&tqk05o8tQv|yP0kku9dmFftZ

diff --git a/precompile/modules/initia_stdlib/sources/vip/vault.move b/precompile/modules/initia_stdlib/sources/vip/vault.move
index ff3416ca..0c772311 100644
--- a/precompile/modules/initia_stdlib/sources/vip/vault.move
+++ b/precompile/modules/initia_stdlib/sources/vip/vault.move
@@ -122,50 +122,16 @@ module initia_std::vip_vault {
 
     #[view]
     public fun reward_per_stage(): u64 acquires ModuleStore {
-        let vault_store_addr = get_vault_store_address();
-        primary_fungible_store::balance(vault_store_addr, vip_reward::reward_metadata())
+        let vault_store = borrow_global<ModuleStore>(@initia_std);
+        vault_store.reward_per_stage
     }
     
     //
     // Tests
     //
 
-    #[test_only]
-    use std::string;
-
-    #[test_only]
-    use std::option;
-
-    #[test_only]
-    use initia_std::coin;
-
-    #[test_only]
-    use initia_std::fungible_asset::Metadata;
-
-    #[test_only]
-    use initia_std::object::Object;
-
     #[test_only]
     public fun init_module_for_test(chain: &signer){
         init_module(chain);
     }
-
-    #[test_only]
-    fun initialize_coin(
-        account: &signer,
-        symbol: string::String,
-    ): (coin::BurnCapability, coin::FreezeCapability, coin::MintCapability, Object<Metadata>) {
-        let (mint_cap, burn_cap, freeze_cap) = coin::initialize(
-            account,
-            option::none(),
-            string::utf8(b""),
-            symbol,
-            6,
-            string::utf8(b""),
-            string::utf8(b""),
-        );
-        let metadata = coin::metadata(signer::address_of(account), symbol);
-
-        (burn_cap, freeze_cap, mint_cap, metadata)
-    }
 }
\ No newline at end of file
diff --git a/precompile/modules/initia_stdlib/sources/vip/vip.move b/precompile/modules/initia_stdlib/sources/vip/vip.move
index 95c196aa..e331fe49 100644
--- a/precompile/modules/initia_stdlib/sources/vip/vip.move
+++ b/precompile/modules/initia_stdlib/sources/vip/vip.move
@@ -1493,6 +1493,7 @@ module initia_std::vip {
         );
         
         let total_reward_per_stage = 100_000_000_000;
+        assert!(vip_vault::reward_per_stage() == total_reward_per_stage, 0);
         let portion = 10;
         let reward_per_stage = total_reward_per_stage/portion;
         let vesting_period = 10;