fix

near · Feb 11, 2025 · b4c4c2f · b4c4c2f
1 parent 6500dde
commit b4c4c2f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/integration-tests/src/test_loop/tests/contract_distribution_cross_shard.rs b/integration-tests/src/test_loop/tests/contract_distribution_cross_shard.rs
@@ -118,14 +118,21 @@ fn test_global_contract(
         &rpc_id,
         &accounts[0],
         contract.code().into(),
-        deploy_mode,
+        deploy_mode.clone(),
         nonce,
     );
     nonce += 1;
     env.test_loop.run_for(Duration::seconds(3));
     check_txs(&env.test_loop.data, &env.datas, &rpc_id, &[deploy_tx]);
-    let code_hash = CryptoHash::hash_bytes(contract.code());
-    let identifier = GlobalContractIdentifier::CodeHash(code_hash);
+    let identifier = match deploy_mode {
+        GlobalContractDeployMode::CodeHash => {
+            let code_hash = CryptoHash::hash_bytes(contract.code());
+            GlobalContractIdentifier::CodeHash(code_hash)
+        }
+        GlobalContractDeployMode::AccountId => {
+            GlobalContractIdentifier::AccountId(accounts[0].clone())
+        }
+    };
     // test on accounts from different shards
     for account in [&accounts[1], &accounts[6]] {
         let use_tx = use_global_contract(

diff --git a/runtime/runtime/src/lib.rs b/runtime/runtime/src/lib.rs
@@ -493,6 +493,7 @@ impl Runtime {
             Action::FunctionCall(function_call) => {
                 let account = account.as_mut().expect(EXPECT_ACCOUNT_EXISTS);
                 let account_contract = account.contract();
+
                 let code_hash = match account_contract.as_ref() {
                     AccountContract::None => CryptoHash::default(),
                     AccountContract::Local(code_hash) | AccountContract::Global(code_hash) => {