From b30a103b89198f6aad262dcd0a1c2fe8a7c7372b Mon Sep 17 00:00:00 2001
From: Chris Tsang <tyt2y7@gmail.com>
Date: Sat, 18 Sep 2021 22:38:57 +0800
Subject: [PATCH] Fixup

---
 src/database/connection.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/database/connection.rs b/src/database/connection.rs
index c3390b338..31e047d2b 100644
--- a/src/database/connection.rs
+++ b/src/database/connection.rs
@@ -141,14 +141,14 @@ impl ConnectionTrait for DatabaseConnection {
 
 #[cfg(feature = "mock")]
 impl DatabaseConnection {
-    fn as_mock_connection(&self) -> &crate::MockDatabaseConnection {
+    pub fn as_mock_connection(&self) -> &crate::MockDatabaseConnection {
         match self {
             DatabaseConnection::MockDatabaseConnection(mock_conn) => mock_conn,
             _ => panic!("not mock connection"),
         }
     }
 
-    fn into_transaction_log(&self) -> Vec<crate::Transaction> {
+    pub fn into_transaction_log(&self) -> Vec<crate::Transaction> {
         let mut mocker = self.as_mock_connection().get_mocker_mutex().lock().unwrap();
         mocker.drain_transaction_log()
     }