fix tests

gear-tech · Dec 5, 2024 · 1d0fb38 · 1d0fb38
1 parent d735d88
commit 1d0fb38
Show file tree

Hide file tree

Showing 6 changed files with 82 additions and 67 deletions.
diff --git a/gtest/src/program.rs b/gtest/src/program.rs
@@ -867,11 +867,8 @@ pub mod gbuild {
 #[cfg(test)]
 mod tests {
     use super::Program;
-
     use crate::{Log, ProgramIdWrapper, System, Value, DEFAULT_USER_ALICE, EXISTENTIAL_DEPOSIT};
     use demo_constructor::{Arg, Scheme};
-    use gear_common::Origin;
-
     use gear_core::ids::ActorId;
     use gear_core_errors::{ErrorReplyReason, ReplyCode, SimpleExecutionError};
 
@@ -1085,7 +1082,7 @@ mod tests {
 
     #[test]
     #[should_panic(
-        expected = "Insufficient balance: user (0x0500000000000000000000000000000000000000000000000000000000000000) \
+        expected = "Insufficient balance: user (0x0000000000000000000000000500000000000000000000000000000000000000) \
     tries to send (1000000000001) value, (4500000000000) gas and ED (1000000000000), while his balance (1000000000000)"
     )]
     fn fails_on_insufficient_balance() {
@@ -1446,7 +1443,12 @@ mod tests {
         let handle = Calls::builder()
             .reserve_gas(10_000_000_000, 5)
             .store("reservation")
-            .reservation_send_value("reservation", user_id.into_origin().0, payload.clone(), 0);
+            .reservation_send_value(
+                "reservation",
+                ActorId::from(user_id).into_bytes(),
+                payload.clone(),
+                0,
+            );
         let msg_id = prog.send(user_id, handle);
         let res = sys.run_next_block();
         assert!(res.succeed.contains(&msg_id));
@@ -1459,7 +1461,7 @@ mod tests {
         let new_prog_id = 4343;
         let new_program = Program::from_binary_with_id(&sys, new_prog_id, WASM_BINARY);
         let payload = b"sup!".to_vec();
-        let handle = Calls::builder().send(user_id.into_origin().0, payload.clone());
+        let handle = Calls::builder().send(ActorId::from(user_id).into_bytes(), payload.clone());
         let scheme = Scheme::predefined(
             Calls::builder().noop(),
             handle,
@@ -1474,7 +1476,12 @@ mod tests {
         let handle = Calls::builder()
             .reserve_gas(10_000_000_000, 5)
             .store("reservation")
-            .reservation_send_value("reservation", new_prog_id.into_origin().0, [], 0);
+            .reservation_send_value(
+                "reservation",
+                ActorId::from(new_prog_id).into_bytes(),
+                [],
+                0,
+            );
         let msg_id = prog.send(user_id, handle);
         let res = sys.run_next_block();
         assert!(res.succeed.contains(&msg_id));

diff --git a/pallets/gear-builtin/src/tests/basic.rs b/pallets/gear-builtin/src/tests/basic.rs
@@ -237,7 +237,7 @@ fn calculate_gas_info_works() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER) && message.details().is_some() && {
+                message.destination() == SIGNER.cast() && message.details().is_some() && {
                     let details = message.details().expect("Value checked above");
                     details.to_reply_code()
                         == ReplyCode::Error(ErrorReplyReason::Execution(

diff --git a/pallets/gear-builtin/src/tests/bls381.rs b/pallets/gear-builtin/src/tests/bls381.rs
@@ -95,7 +95,7 @@ fn decoding_error() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::UserspacePanic,
@@ -137,7 +137,7 @@ fn multi_miller_loop() {
 
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::UserspacePanic,
@@ -172,7 +172,7 @@ fn multi_miller_loop() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -197,7 +197,7 @@ fn multi_miller_loop() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -256,7 +256,7 @@ fn final_exponentiation() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -281,7 +281,7 @@ fn final_exponentiation() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -339,7 +339,7 @@ fn msm_g1() {
 
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::UserspacePanic,
@@ -374,7 +374,7 @@ fn msm_g1() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -399,7 +399,7 @@ fn msm_g1() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -457,7 +457,7 @@ fn msm_g2() {
 
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::UserspacePanic,
@@ -492,7 +492,7 @@ fn msm_g2() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -517,7 +517,7 @@ fn msm_g2() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -574,7 +574,7 @@ fn mul_projective_g1() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -599,7 +599,7 @@ fn mul_projective_g1() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -656,7 +656,7 @@ fn mul_projective_g2() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -681,7 +681,7 @@ fn mul_projective_g2() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -717,7 +717,7 @@ fn aggregate_g1() {
         let encoded_points = ark_points.encode();
 
         let payload = Request::AggregateG1 { points: encoded_points }.encode();
-        let builtin_actor_id: ProgramId = H256::from(ACTOR_ID).cast();
+        let builtin_actor_id = ACTOR_ID.into();
         let gas_info = get_gas_info(builtin_actor_id, payload.clone());
 
         // Check the case of insufficient gas
@@ -736,7 +736,7 @@ fn aggregate_g1() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -761,7 +761,7 @@ fn aggregate_g1() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())
@@ -816,7 +816,7 @@ fn map_to_g2affine() {
         // An error reply should have been sent.
         assert!(System::events().into_iter().any(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                message.destination() == ProgramId::from(SIGNER)
+                message.destination() == SIGNER.cast()
                     && matches!(message.details(), Some(details) if details.to_reply_code()
                     == ReplyCode::Error(ErrorReplyReason::Execution(
                         SimpleExecutionError::RanOutOfGas,
@@ -841,7 +841,7 @@ fn map_to_g2affine() {
 
         let response = match System::events().into_iter().find_map(|e| match e.event {
             RuntimeEvent::Gear(pallet_gear::Event::<Test>::UserMessageSent { message, .. }) => {
-                assert_eq!(message.destination(), ProgramId::from(SIGNER));
+                assert_eq!(message.destination(), SIGNER.cast());
                 assert!(matches!(message.details(), Some(details) if matches!(details.to_reply_code(), ReplyCode::Success(..))));
 
                 Some(message.payload_bytes().to_vec())

diff --git a/pallets/gear-debug/src/tests/mod.rs b/pallets/gear-debug/src/tests/mod.rs
@@ -18,7 +18,7 @@
 
 use super::*;
 use crate::mock::*;
-use common::{self, event::MessageEntry, CodeStorage, Origin as _};
+use common::{self, event::MessageEntry, CodeStorage, Origin};
 use frame_support::assert_ok;
 use gear_core::{
     ids::{prelude::*, CodeId, MessageId, ProgramId},
@@ -251,7 +251,7 @@ fn debug_mode_works() {
                         DispatchKind::Handle,
                         StoredMessage::new(
                             message_id_1,
-                            1.into(),
+                            1.cast(),
                             program_id_1,
                             Default::default(),
                             0,
@@ -263,7 +263,7 @@ fn debug_mode_works() {
                         DispatchKind::Handle,
                         StoredMessage::new(
                             message_id_2,
-                            1.into(),
+                            1.cast(),
                             program_id_2,
                             Default::default(),
                             0,
@@ -356,10 +356,12 @@ fn get_last_program_id() -> ProgramId {
 }
 
 #[track_caller]
-fn maybe_last_message(account: u64) -> Option<UserMessage> {
+fn maybe_last_message(account: impl Origin) -> Option<UserMessage> {
+    let account = account.cast();
+
     System::events().into_iter().rev().find_map(|e| {
         if let super::mock::RuntimeEvent::Gear(Event::UserMessageSent { message, .. }) = e.event {
-            if message.destination() == account.into() {
+            if message.destination() == account {
                 Some(message)
             } else {
                 None

diff --git a/pallets/gear-scheduler/src/tests.rs b/pallets/gear-scheduler/src/tests.rs
@@ -48,12 +48,12 @@ fn wl_cost_for(amount_of_blocks: u64) -> u128 {
     gas_price(<Pallet<Test> as Scheduler>::CostsPerBlock::waitlist() * amount_of_blocks)
 }
 
-fn dispatch_from(src: impl Into<ProgramId>) -> StoredDispatch {
+fn dispatch_from(src: impl Origin) -> StoredDispatch {
     StoredDispatch::new(
         DispatchKind::Handle,
         StoredMessage::new(
             H256::random().cast(),
-            src.into(),
+            src.cast(),
             H256::random().cast(),
             Default::default(),
             0,