Add unit test for unix domain sockets support in WhiteListRoundRobinPolicy

Author: sylwiaszunejko

tests/unit/test_policies.py

@@ -34,7 +34,7 @@
                                 LoadBalancingPolicy, ConvictionPolicy, ReconnectionPolicy, FallthroughRetryPolicy,
                                 IdentityTranslator, EC2MultiRegionTranslator, HostFilterPolicy)
 from cassandra.pool import Host
-from cassandra.connection import DefaultEndPoint
+from cassandra.connection import DefaultEndPoint, UnixSocketEndPoint
 from cassandra.query import Statement
 
 from six.moves import xrange
@@ -1254,6 +1254,17 @@ def test_hosts_with_hostname(self):
         self.assertEqual(sorted(qplan), [host])
 
         self.assertEqual(policy.distance(host), HostDistance.LOCAL)
+    
+    def test_hosts_with_socket_hostname(self):
+        hosts = [UnixSocketEndPoint('/tmp/scylla-workdir/cql.m')]
+        policy = WhiteListRoundRobinPolicy(hosts)
+        host = Host(UnixSocketEndPoint('/tmp/scylla-workdir/cql.m'), SimpleConvictionPolicy)
+        policy.populate(None, [host])
+
+        qplan = list(policy.make_query_plan())
+        self.assertEqual(sorted(qplan), [host])
+
+        self.assertEqual(policy.distance(host), HostDistance.LOCAL)
 
 
 class AddressTranslatorTest(unittest.TestCase):