Accomodate signer

MystenLabs · Feb 8, 2022 · 8a5a772 · 8a5a772
1 parent ef0c7ec
commit 8a5a772
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/fastpay_core/src/client.rs b/fastpay_core/src/client.rs
@@ -30,7 +30,7 @@ use std::{
 ///
 /// Typically instantiated with Box::pin(keypair) where keypair is a `KeyPair`
 ///
-type StableSyncSigner = Pin<Box<dyn signature::Signer<ed25519_dalek::Signature> + Send + Sync>>;
+pub type StableSyncSigner = Pin<Box<dyn signature::Signer<ed25519_dalek::Signature> + Send + Sync>>;
 
 pub mod client_store;
 use self::client_store::ClientStore;
@@ -237,8 +237,8 @@ impl<A> ClientState<A> {
     }
 
     #[cfg(test)]
-    pub fn secret(&self) -> &KeyPair {
-        &self.secret
+    pub fn secret(&self) -> &dyn signature::Signer<ed25519_dalek::Signature> {
+        &*self.secret
     }
 
     /// Given an order, return the list of certificates that are known by this client

diff --git a/fastpay_core/src/unit_tests/client_tests.rs b/fastpay_core/src/unit_tests/client_tests.rs
@@ -137,7 +137,7 @@ async fn extract_cert(
 #[cfg(test)]
 fn order_create(
     src: FastPayAddress,
-    secret: &KeyPair,
+    secret: &dyn signature::Signer<ed25519_dalek::Signature>,
     dest: FastPayAddress,
     value: u64,
     framework_obj_ref: ObjectRef,
@@ -159,14 +159,14 @@ fn order_create(
         Vec::new(),
         pure_arguments,
         GAS_VALUE_FOR_TESTING / 2,
-        secret,
+        &*secret,
     )
 }
 
 #[cfg(test)]
 fn order_transfer(
     src: FastPayAddress,
-    secret: &KeyPair,
+    secret: &dyn signature::Signer<ed25519_dalek::Signature>,
     dest: FastPayAddress,
     object_ref: ObjectRef,
     framework_obj_ref: ObjectRef,
@@ -192,7 +192,7 @@ fn order_transfer(
 #[cfg(test)]
 fn order_set(
     src: FastPayAddress,
-    secret: &KeyPair,
+    secret: &dyn signature::Signer<ed25519_dalek::Signature>,
     object_ref: ObjectRef,
     value: u64,
     framework_obj_ref: ObjectRef,
@@ -217,7 +217,7 @@ fn order_set(
 #[cfg(test)]
 fn order_delete(
     src: FastPayAddress,
-    secret: &KeyPair,
+    secret: &dyn signature::Signer<ed25519_dalek::Signature>,
     object_ref: ObjectRef,
     framework_obj_ref: ObjectRef,
     gas_object_ref: ObjectRef,