From acb40cc1a32082849211ed92e544a2d5dc1819ac Mon Sep 17 00:00:00 2001
From: Martin Algesten <martin@algesten.se>
Date: Sun, 20 Oct 2019 11:45:37 +0200
Subject: [PATCH] Fail if PoolKey::new() cant find port

---
 src/pool.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/pool.rs b/src/pool.rs
index 6dff887..b611381 100644
--- a/src/pool.rs
+++ b/src/pool.rs
@@ -52,9 +52,20 @@ struct PoolKey {
 
 impl PoolKey {
     fn new(url: &Url) -> Self {
+        let port = if cfg!(test) {
+            if let Some(p) = url.port_or_known_default() {
+                Some(p)
+            } else if url.scheme() == "test" {
+                Some(42)
+            } else {
+                None
+            }
+        } else {
+            url.port_or_known_default()
+        };
         PoolKey {
             hostname: url.host_str().unwrap_or(DEFAULT_HOST).into(),
-            port: url.port_or_known_default().unwrap_or(0),
+            port: port.expect("Failed to get port for pool key"),
         }
     }
 }