EPROD-1156-live mode on pocket ic client

Author: veeso

ic-canister-client/src/pocket_ic.rs

@@ -13,6 +13,7 @@ pub struct PocketIcClient {
     client: Option<Arc<PocketIc>>,
     pub canister: Principal,
     pub caller: Principal,
+    live: bool,
 }
 
 impl PocketIcClient {
@@ -22,6 +23,15 @@ impl PocketIcClient {
         Self::from_client(PocketIc::new().await, canister, caller)
     }
 
+    /// Creates a new instance of a PocketIcClient in live mode
+    /// The new instance is independent and have no access to canisters of other instances.
+    pub async fn new_live(canister: Principal, caller: Principal) -> Self {
+        let mut client = PocketIc::new().await;
+        client.make_live(None).await;
+
+        Self::from_client(client, canister, caller)
+    }
+
     /// Crates new instance of PocketIcClient from an existing client instance.
     pub fn from_client<P: Into<Arc<PocketIc>>>(
         client: P,
@@ -32,6 +42,21 @@ impl PocketIcClient {
             client: Some(client.into()),
             canister,
             caller,
+            live: false,
+        }
+    }
+
+    /// Crates new instance of PocketIcClient from an existing client instance.
+    pub fn from_client_live<P: Into<Arc<PocketIc>>>(
+        client: P,
+        canister: Principal,
+        caller: Principal,
+    ) -> Self {
+        Self {
+            client: Some(client.into()),
+            canister,
+            caller,
+            live: true,
         }
     }
 
@@ -50,10 +75,17 @@ impl PocketIcClient {
     {
         let args = candid::encode_args(args)?;
 
-        let reply = self
-            .client()
-            .update_call(self.canister, self.caller, method, args)
-            .await?;
+        let reply = if self.live {
+            let id = self
+                .client()
+                .submit_call(self.canister, self.caller, method, args)
+                .await?;
+            self.client().await_call_no_ticks(id).await
+        } else {
+            self.client()
+                .update_call(self.canister, self.caller, method, args)
+                .await
+        }?;
 
         let decoded = Decode!(&reply, R)?;
         Ok(decoded)