From 128ea565caec36a04fdc82f0eeb925e15aad2e5c Mon Sep 17 00:00:00 2001
From: matt <mcarey66@gmail.com>
Date: Mon, 18 Feb 2019 13:47:45 -0500
Subject: [PATCH] Add extra log when InfluxDB fails to connected

---
 influxspeedtest/InfluxdbSpeedtest.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/influxspeedtest/InfluxdbSpeedtest.py b/influxspeedtest/InfluxdbSpeedtest.py
index a32cf4b..3daa02b 100644
--- a/influxspeedtest/InfluxdbSpeedtest.py
+++ b/influxspeedtest/InfluxdbSpeedtest.py
@@ -4,7 +4,7 @@ import time
 import speedtest
 from influxdb import InfluxDBClient
 from influxdb.exceptions import InfluxDBClientError, InfluxDBServerError
-from requests import ConnectTimeout
+from requests import ConnectTimeout, ConnectionError
 
 from influxspeedtest.common import log
 from influxspeedtest.config import config
@@ -40,11 +40,13 @@ class InfluxdbSpeedtest():
             log.debug('Testing connection to InfluxDb using provided credentials')
             influx.get_list_users()  # TODO - Find better way to test connection and permissions
             log.debug('Successful connection to InfluxDb')
-        except (ConnectTimeout, InfluxDBClientError) as e:
+        except (ConnectTimeout, InfluxDBClientError, ConnectionError) as e:
             if isinstance(e, ConnectTimeout):
                 log.critical('Unable to connect to InfluxDB at the provided address (%s)', config.influx_address)
             elif e.code == 401:
                 log.critical('Unable to connect to InfluxDB with provided credentials')
+            else:
+                log.critical('Failed to connect to InfluxDB for unknown reason')
 
             sys.exit(1)
 
-- 
GitLab