From 787b23404800bdf069c0632a5a89cb4798f58bbd Mon Sep 17 00:00:00 2001
From: Sebastian Pravda <sebastian.pravda@gmail.com>
Date: Wed, 14 Dec 2022 16:41:12 +0100
Subject: [PATCH] feat: test util function under cfg(test)

---
 core/src/utils/db.rs | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/core/src/utils/db.rs b/core/src/utils/db.rs
index cc71e0e..59a754d 100644
--- a/core/src/utils/db.rs
+++ b/core/src/utils/db.rs
@@ -1,6 +1,4 @@
-use entity::{admin, candidate, parent, session};
-use sea_orm::{Schema, Database, DbConn};
-use sea_orm::{sea_query::TableCreateStatement, ConnectionTrait, DbBackend};
+use sea_orm::DbConn;
 
 use crate::Query;
 use crate::error::ServiceError;
@@ -12,8 +10,13 @@ pub async fn get_recipients(db: &DbConn, candidate_pubkey: &str) -> Result<Vec<S
     Ok(recipients)
 }
 
-
+#[cfg(test)]
 pub async fn get_memory_sqlite_connection() -> sea_orm::DbConn {
+    use entity::{admin, candidate, parent, session};
+    use sea_orm::{Schema, Database};
+    use sea_orm::{sea_query::TableCreateStatement, ConnectionTrait, DbBackend};
+
+    
     let base_url = "sqlite::memory:";
     let db: DbConn = Database::connect(base_url).await.unwrap();