feat: update gas meter

Cargo.toml

@@ -112,27 +112,28 @@ tiny-keccak = { version = "2.0.2", features = ["keccak", "sha3"] }
 # Note: the BEGIN and END comments below are required for external tooling. Do not remove.
 # BEGIN MOVE DEPENDENCIES
 
-move-binary-format = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-bytecode-verifier = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-bytecode-utils = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-cli = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-command-line-common = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-compiler = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-core-types = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-docgen = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-model = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-package = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-prover = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-prover-boogie-backend = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-prover-bytecode-pipeline = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-resource-viewer = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-stackless-bytecode = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-stdlib = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-symbol-pool = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-unit-test = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-vm-runtime = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-vm-test-utils = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
-move-vm-types = { git = "https://github.com/initia-labs/aptos-core.git", rev = "1cae58ce6121080564a238238dde2c8124bcb112" }
+move-binary-format = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-bytecode-verifier = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-bytecode-utils = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-cli = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-command-line-common = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-compiler = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-coverage = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-core-types = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-docgen = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-model = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-package = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-prover = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-prover-boogie-backend = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-prover-bytecode-pipeline = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-resource-viewer = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-stackless-bytecode = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-stdlib = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-symbol-pool = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-unit-test = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-vm-runtime = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-vm-test-utils = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
+move-vm-types = { git = "https://github.com/aptos-labs/aptos-core.git", rev = "85966af7ddcf971f3f04f4f409c3e67c9d9b11cb" }
 
 # END MOVE DEPENDENCIES
 
@@ -143,6 +144,7 @@ move-vm-types = { git = "https://github.com/initia-labs/aptos-core.git", rev = "
 # move-cli = { path = "../aptos-core/third_party/move/tools/move-cli" }
 # move-command-line-common = { path = "../aptos-core/third_party/move/move-command-line-common" }
 # move-compiler = { path = "../aptos-core/third_party/move/move-compiler" }
+# move-coverage ={ path = "../aptos-core/third_party/move/tools/move-coverage" }
 # move-core-types = { path = "../aptos-core/third_party/move/move-core/types" }
 # move-docgen = { path = "../aptos-core/third_party/move/move-prover/move-docgen" }
 # move-model = { path = "../aptos-core/third_party/move/move-model" }

crates/compiler/Cargo.toml

@@ -34,6 +34,7 @@ initia-move-storage = { workspace = true }
 move-command-line-common = { workspace = true }
 move-cli = { workspace = true }
 move-core-types = { workspace = true }
+move-coverage ={ workspace = true }
 move-package = { workspace = true }
 move-unit-test = { workspace = true }
 move-vm-runtime = { workspace = true, features = ["testing"] }

crates/compiler/src/test_package.rs

@@ -11,6 +11,7 @@ use move_cli::base::{
     test::{run_move_unit_tests_with_factory, Test, UnitTestResult},
 };
 use move_core_types::effects::ChangeSet;
+use move_coverage::source_coverage::{ColorChoice, TextIndicator};
 use move_model::metadata::{CompilerVersion, LanguageVersion};
 use move_package::BuildConfig;
 use move_unit_test::UnitTestingConfig;
@@ -76,6 +77,8 @@ impl TestPackage {
                         functions: false,
                         output_csv: false,
                     },
+                    color: ColorChoice::Default,
+                    tag: TextIndicator::Explicit,
                 },
                 Some(self.package_path.clone()),
                 self.build_config.clone(),

crates/compiler/src/unit_test_factory.rs

@@ -4,7 +4,7 @@ use initia_move_types::write_set::WriteSet;
 use move_binary_format::errors::{Location, PartialVMError, VMResult};
 use move_core_types::{effects::ChangeSet, vm_status::StatusCode};
 use move_unit_test::test_reporter::{TestRunInfo, UnitTestFactory};
-use move_vm_runtime::session::Session;
+use move_vm_runtime::native_extensions::NativeContextExtensions;
 
 pub struct InitiaUnitTestFactory {
     pub gas_params: InitiaGasParameters,
@@ -40,17 +40,19 @@ impl InitiaUnitTestFactory {
 }
 
 // gas meter required for testing gas metering
-impl UnitTestFactory<InitiaGasMeter> for InitiaUnitTestFactory {
-    fn new_gas_meter(&self) -> InitiaGasMeter {
+impl UnitTestFactory for InitiaUnitTestFactory {
+    type GasMeter = InitiaGasMeter;
+    fn new_gas_meter(&self) -> Self::GasMeter {
         InitiaGasMeter::new(self.gas_params.clone(), self.balance)
     }
 
-    fn finish_session(
+    fn finalize_test_run_info(
         &self,
-        session: Session,
-        mut gas_meter: InitiaGasMeter,
+        change_set: &ChangeSet,
+        extensions: &mut NativeContextExtensions,
+        mut gas_meter: Self::GasMeter,
         mut test_run_info: TestRunInfo,
-    ) -> (VMResult<ChangeSet>, TestRunInfo) {
+    ) -> TestRunInfo {
         let mut apply_gas_used = |gas_meter: InitiaGasMeter| {
             test_run_info.gas_used = gas_meter
                 .gas_limit()
@@ -59,23 +61,18 @@ impl UnitTestFactory<InitiaGasMeter> for InitiaUnitTestFactory {
                 .into();
         };
 
-        match session.finish_with_extensions() {
-            Ok((cs, mut exts)) => {
-                let table_context: NativeTableContext = exts.remove::<NativeTableContext>();
-                match Self::charge_write_set_gas(&mut gas_meter, &cs, table_context) {
-                    Ok(()) => {
-                        apply_gas_used(gas_meter);
-                        (Ok(cs), test_run_info)
-                    }
-                    Err(err) => {
-                        apply_gas_used(gas_meter);
-                        (Err(err), test_run_info)
-                    },
-                }
+        match Self::charge_write_set_gas(
+            &mut gas_meter,
+            change_set,
+            extensions.remove::<NativeTableContext>(),
+        ) {
+            Ok(()) => {
+                apply_gas_used(gas_meter);
+                test_run_info
             }
-            Err(err) => {
+            Err(_) => {
                 apply_gas_used(gas_meter);
-                (Err(err), test_run_info)
+                test_run_info
             }
         }
     }

crates/types/Cargo.toml

@@ -13,6 +13,7 @@ move-cli = { workspace = true }
 move-docgen = { workspace = true }
 move-package = { workspace = true }
 move-model = { workspace = true }
+move-coverage = { workspace = true }
 
 anyhow = { workspace = true }
 bcs = { workspace = true }

crates/types/src/compiler.rs

@@ -8,6 +8,7 @@ use move_cli::{
     Move,
 };
 use move_core_types::account_address::AccountAddress;
+use move_coverage::source_coverage::{ColorChoice, TextIndicator};
 use move_docgen::DocgenOptions;
 use move_model::metadata::{CompilerVersion, LanguageVersion};
 use move_package::{Architecture, BuildConfig, CompilerConfig};
@@ -150,6 +151,8 @@ impl From<CompilerCoverageSummaryOptions> for Coverage {
                 functions: val.functions,
                 output_csv: val.output_csv,
             },
+            color: ColorChoice::Default,
+            tag: TextIndicator::Explicit,
         }
     }
 }
@@ -166,6 +169,8 @@ impl From<CompilerCoverageSourceOptions> for Coverage {
             options: CoverageSummaryOptions::Source {
                 module_name: module_name.unwrap(),
             },
+            color: ColorChoice::Default,
+            tag: TextIndicator::Explicit,
         }
     }
 }
@@ -182,6 +187,8 @@ impl From<CompilerCoverageBytecodeOptions> for Coverage {
             options: CoverageSummaryOptions::Bytecode {
                 module_name: module_name.unwrap(),
             },
+            color: ColorChoice::Default,
+            tag: TextIndicator::Explicit,
         }
     }
 }