fix: sanity check that host functions don't increase action costs

near · Apr 2, 2021 · a3884c4 · a3884c4
1 parent 46f5ce8
commit a3884c4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/core/primitives-core/src/config.rs b/core/primitives-core/src/config.rs
@@ -537,7 +537,7 @@ pub enum ExtCosts {
     #[cfg(feature = "protocol_feature_alt_bn128")]
     alt_bn128_g1_sum_byte,
 
-    // NB: this should be the last element of the enum.
+    // NOTE: this should be the last element of the enum.
     __count,
 }
 
@@ -556,7 +556,7 @@ pub enum ActionCosts {
     value_return,
     new_receipt,
 
-    // NB: this should be the last element of the enum.
+    // NOTE: this should be the last element of the enum.
     __count,
 }
 

diff --git a/runtime/near-vm-runner/src/tests/rs_contract.rs b/runtime/near-vm-runner/src/tests/rs_contract.rs
@@ -134,6 +134,7 @@ fn run_test_ext(
     let fees = RuntimeFeesConfig::default();
     let context = create_context(input.to_vec());
 
+    let profile = ProfileData::new_enabled();
     let (outcome, err) = run_vm(
         &code,
         &method,
@@ -145,9 +146,12 @@ fn run_test_ext(
         vm_kind,
         LATEST_PROTOCOL_VERSION,
         None,
-        ProfileData::new_disabled(),
+        profile.clone(),
     );
 
+    assert_eq!(profile.action_gas(), 0);
+    assert!(profile.host_gas() > 0);
+
     if let Some(_) = err {
         panic!("Failed execution: {:?}", err);
     }