test(user-agent): add test helpers and update actor integration tests

server/crates/arbiter-server/src/actors/user_agent/session.rs

@@ -1,6 +1,7 @@
 use std::{borrow::Cow, convert::Infallible};
 
 use arbiter_proto::transport::Sender;
+use async_trait::async_trait;
 use ed25519_dalek::VerifyingKey;
 use kameo::{Actor, messages, prelude::Context};
 use thiserror::Error;
@@ -42,8 +43,8 @@ mod connection;
 pub(crate) use connection::{
     BootstrapError, HandleBootstrapEncryptedKey, HandleEvmWalletCreate, HandleEvmWalletList,
     HandleGrantCreate, HandleGrantDelete, HandleGrantList, HandleQueryVaultState,
-    HandleUnsealEncryptedKey, HandleUnsealRequest, UnsealError,
 };
+pub use connection::{HandleUnsealEncryptedKey, HandleUnsealRequest, UnsealError};
 
 impl UserAgentSession {
     pub(crate) fn new(props: UserAgentConnection, sender: Box<dyn Sender<OutOfBand>>) -> Self {
@@ -54,6 +55,22 @@ impl UserAgentSession {
         }
     }
 
+    pub fn new_test(db: crate::db::DatabasePool, actors: crate::actors::GlobalActors) -> Self {
+        struct DummySender;
+
+        #[async_trait]
+        impl Sender<OutOfBand> for DummySender {
+            async fn send(
+                &mut self,
+                _item: OutOfBand,
+            ) -> Result<(), arbiter_proto::transport::Error> {
+                Ok(())
+            }
+        }
+
+        Self::new(UserAgentConnection::new(db, actors), Box::new(DummySender))
+    }
+
     fn transition(&mut self, event: UserAgentEvents) -> Result<(), Error> {
         self.state.process_event(event).map_err(|e| {
             error!(?e, "State transition failed");

server/crates/arbiter-server/src/actors/user_agent/session/connection.rs

@@ -109,7 +109,7 @@ pub enum BootstrapError {
 #[messages]
 impl UserAgentSession {
     #[message]
-    pub(crate) async fn handle_unseal_request(
+    pub async fn handle_unseal_request(
         &mut self,
         client_pubkey: x25519_dalek::PublicKey,
     ) -> Result<UnsealStartResponse, Error> {
@@ -127,7 +127,7 @@ impl UserAgentSession {
     }
 
     #[message]
-    pub(crate) async fn handle_unseal_encrypted_key(
+    pub async fn handle_unseal_encrypted_key(
         &mut self,
         nonce: Vec<u8>,
         ciphertext: Vec<u8>,