Add Deref impl

blockscout · Oct 13, 2023 · 35fbeef · 35fbeef
1 parent e58fe56
commit 35fbeef
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/libs/blockscout-service-launcher/src/test_database.rs b/libs/blockscout-service-launcher/src/test_database.rs
@@ -1,7 +1,7 @@
 use crate::database::{
     ConnectionTrait, Database, DatabaseConnection, DbErr, MigratorTrait, Statement,
 };
-use std::sync::Arc;
+use std::{ops::Deref, sync::Arc};
 
 #[derive(Clone, Debug)]
 pub struct TestDbGuard {
@@ -82,3 +82,10 @@ impl TestDbGuard {
         Migrator::up(db, None).await
     }
 }
+
+impl Deref for TestDbGuard {
+    type Target = DatabaseConnection;
+    fn deref(&self) -> &Self::Target {
+        &*self.conn_with_db
+    }
+}