add some test_functions (#216)

starcoinorg · Nov 22, 2022 · 87859a9 · 87859a9
1 parent a66f0f3
commit 87859a9
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 0 deletions.
diff --git a/build/StarcoinFramework/source_maps/Account.mvsm b/build/StarcoinFramework/source_maps/Account.mvsm
diff --git a/build/StarcoinFramework/source_maps/GenesisSignerCapability.mvsm b/build/StarcoinFramework/source_maps/GenesisSignerCapability.mvsm
diff --git a/sources/Account.move b/sources/Account.move
@@ -201,6 +201,11 @@ module Account {
         signer_cap
     }
 
+    #[test_only]
+    public fun get_genesis_capability_for_test(): SignerCapability {
+        get_genesis_capability()
+    }
+
     public fun create_signer_with_cap(cap: &SignerCapability): signer {
         create_signer(cap.addr)
     }

diff --git a/sources/GenesisSignerCapability.move b/sources/GenesisSignerCapability.move
@@ -34,4 +34,14 @@ module StarcoinFramework::GenesisSignerCapability {
         let cap = borrow_global<GenesisSignerCapability>(CoreAddresses::GENESIS_ADDRESS());
         Account::create_signer_with_cap(&cap.cap)
     }
+
+    #[test_only]
+    public fun initialize_for_test(signer: &signer, cap: Account::SignerCapability) {
+        initialize(signer, cap);
+    }
+
+    #[test_only]
+    public fun get_genesis_signer_for_test(): signer acquires GenesisSignerCapability {
+        get_genesis_signer()
+    }
 }