Implement RPC for reading & writing to app cache file

- Implement RPC for writing to a file in a test runner / guest VM.
- Implement RPC for getting app cache directory
- Implement RPC for restarting the app in a test runner / guest vm
- Implement RPC for starting the app in a test runner / guest vm
- Implement RPC for stopping the app in a test runner / guest vm
- Implement `find_cache_traces` on Window & macOS

1 files changed, 46 insertions, 0 deletions

diff --git a/test/test-rpc/src/client.rs b/test/test-rpc/src/client.rs
index 6be77afb40..acc462dd60 100644
--- a/test/test-rpc/src/client.rs
+++ b/test/test-rpc/src/client.rs
@@ -151,6 +151,13 @@ impl ServiceClient {
             .await?
     }
 
+    /// Returns path of Mullvad app cache directorie on the test runner.
+    pub async fn find_mullvad_app_cache_dir(&self) -> Result<PathBuf, Error> {
+        self.client
+            .get_mullvad_app_cache_dir(tarpc::context::current())
+            .await?
+    }
+
     /// Send TCP packet
     pub async fn send_tcp(
         &self,
@@ -213,6 +220,34 @@ impl ServiceClient {
             .await?
     }
 
+    pub async fn restart_app(&self) -> Result<(), Error> {
+        let _ = self.client.restart_app(tarpc::context::current()).await?;
+        Ok(())
+    }
+
+    /// Stop the app.
+    ///
+    /// Shuts down a running app, making it disconnect from any current tunnel
+    /// connection and making it write to caches.
+    ///
+    /// # Note
+    /// This function will return *after* the app has been stopped, thus
+    /// blocking execution until then.
+    pub async fn stop_app(&self) -> Result<(), Error> {
+        let _ = self.client.stop_app(tarpc::context::current()).await?;
+        Ok(())
+    }
+
+    /// Start the app.
+    ///
+    /// # Note
+    /// This function will return *after* the app has been start, thus
+    /// blocking execution until then.
+    pub async fn start_app(&self) -> Result<(), Error> {
+        let _ = self.client.start_app(tarpc::context::current()).await?;
+        Ok(())
+    }
+
     pub async fn set_daemon_log_level(
         &self,
         verbosity_level: mullvad_daemon::Verbosity,
@@ -247,6 +282,17 @@ impl ServiceClient {
             .await?
     }
 
+    pub async fn write_file(&self, dest: String, bytes: Vec<u8>) -> Result<(), Error> {
+        log::debug!(
+            "Writing {bytes} bytes to \"{file}\"",
+            bytes = bytes.len(),
+            file = dest
+        );
+        self.client
+            .write_file(tarpc::context::current(), dest, bytes)
+            .await?
+    }
+
     pub async fn reboot(&mut self) -> Result<(), Error> {
         log::debug!("Rebooting server");