Fix more

Author: avandras

tests/test_citus.py

@@ -3,7 +3,7 @@
 import unittest
 
 from unittest.mock import Mock, patch, PropertyMock
-from patroni.postgresql.mpp.citus import CitusHandler
+from patroni.postgresql.mpp.citus import CitusHandler, PgDistNode, PgDistGroup
 from patroni.psycopg import ProgrammingError
 
 from . import BaseTestPostgresql, MockCursor, psycopg_connect, SleepException

tests/test_ctl.py

@@ -2,6 +2,7 @@
 import etcd
 import os
 import unittest
+import urllib3
 
 from datetime import datetime, timedelta
 from unittest import mock

tests/test_patroni.py

@@ -6,7 +6,9 @@
 import unittest
 
 from http.server import HTTPServer
+from threading import Thread
 from unittest.mock import Mock, PropertyMock, patch
+
 from patroni.api import RestApiServer
 from patroni.async_executor import AsyncExecutor
 from patroni.dcs import Cluster, ClusterConfig, Member