add tests

BitcreditProtocol · Nov 26, 2024 · 9372c42 · 9372c42
1 parent 763a58d
commit 9372c42
Show file tree

Hide file tree

Showing 3 changed files with 642 additions and 0 deletions.
diff --git a/src/persistence/company.rs b/src/persistence/company.rs
@@ -14,6 +14,10 @@ use tokio::{
     task,
 };
 
+#[cfg(test)]
+use mockall::automock;
+
+#[cfg_attr(test, automock)]
 #[async_trait]
 pub trait CompanyStoreApi: Send + Sync {
     /// Checks if the given company exists

diff --git a/src/persistence/contact.rs b/src/persistence/contact.rs
@@ -5,6 +5,10 @@ use std::{collections::HashMap, fs, path::Path};
 use super::{file_storage_path, Error, Result};
 use async_trait::async_trait;
 
+#[cfg(test)]
+use mockall::automock;
+
+#[cfg_attr(test, automock)]
 #[async_trait]
 pub trait ContactStoreApi: Send + Sync {
     async fn get_map(&self) -> Result<HashMap<String, IdentityPublicData>>;