From bfe868ef5520a34eff8706baafb21b62f83aea43 Mon Sep 17 00:00:00 2001
From: Taylor Neely <tneely417@gmail.com>
Date: Tue, 27 Aug 2024 19:30:20 -0700
Subject: [PATCH] Update test_drop_on_broken to use BrokenConnectionManager

---
 bb8/tests/test.rs | 25 ++++---------------------
 1 file changed, 4 insertions(+), 21 deletions(-)

diff --git a/bb8/tests/test.rs b/bb8/tests/test.rs
index 38e3198..330636e 100644
--- a/bb8/tests/test.rs
+++ b/bb8/tests/test.rs
@@ -251,27 +251,10 @@ async fn test_drop_on_broken() {
         }
     }
 
-    struct Handler;
-
-    #[async_trait]
-    impl ManageConnection for Handler {
-        type Connection = Connection;
-        type Error = Error;
-
-        async fn connect(&self) -> Result<Self::Connection, Self::Error> {
-            Ok(Default::default())
-        }
-
-        async fn is_valid(&self, _conn: &mut Self::Connection) -> Result<(), Self::Error> {
-            Ok(())
-        }
-
-        fn has_broken(&self, _: &mut Self::Connection) -> bool {
-            true
-        }
-    }
-
-    let pool = Pool::builder().build(Handler).await.unwrap();
+    let pool = Pool::builder()
+        .build(BrokenConnectionManager::<Connection>::new())
+        .await
+        .unwrap();
     {
         let _ = pool.get().await.unwrap();
     }