Add test for IPv6 functionality

fluent/sender.py

@@ -120,6 +120,13 @@ def close(self):
             self._close()
             self.pendings = None
 
+    def _is_ipv4_host(self):
+        try:
+            socket.getaddrinfo(self.host, None, socket.AF_INET)
+            return True
+        except socket.error:
+            return False
+
     def _make_packet(self, label, timestamp, data):
         if label:
             tag = '.'.join((self.tag, label))
@@ -201,7 +208,12 @@ def _reconnect(self):
                     sock.settimeout(self.timeout)
                     sock.connect(self.host[len('unix://'):])
                 else:
-                    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                    if self._is_ipv4_host():
+                        sock = socket.socket(socket.AF_INET,
+                                             socket.SOCK_STREAM)
+                    else:
+                        sock = socket.socket(socket.AF_INET6,
+                                             socket.SOCK_STREAM)
                     sock.settimeout(self.timeout)
                     # This might be controversial and may need to be removed
                     sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)

tests/mockserver.py

@@ -16,21 +16,21 @@ class MockRecvServer(threading.Thread):
     Single threaded server accepts one connection and recv until EOF.
     """
 
-    def __init__(self, host='localhost', port=0):
+    def __init__(self, host='localhost', port=0, inet_family=socket.AF_INET):
         super(MockRecvServer, self).__init__()
 
         if host.startswith('unix://'):
             self.socket_proto = socket.AF_UNIX
             self.socket_type = socket.SOCK_STREAM
             self.socket_addr = host[len('unix://'):]
         else:
-            self.socket_proto = socket.AF_INET
+            self.socket_proto = inet_family
             self.socket_type = socket.SOCK_STREAM
             self.socket_addr = (host, port)
 
         self._sock = socket.socket(self.socket_proto, self.socket_type)
         self._sock.bind(self.socket_addr)
-        if self.socket_proto == socket.AF_INET:
+        if self.socket_proto == inet_family:
             self.port = self._sock.getsockname()[1]
 
         self._sock.listen(1)
@@ -76,7 +76,7 @@ def close(self):
             pass
 
         try:
-            conn = socket.socket(socket.AF_INET,
+            conn = socket.socket(self.socket_proto,
                                  socket.SOCK_STREAM)
             try:
                 conn.connect((self.socket_addr[0], self.port))

tests/test_sender.py

@@ -8,6 +8,7 @@
 import unittest
 from shutil import rmtree
 from tempfile import mkdtemp
+from unittest.mock import patch
 
 import msgpack
 
@@ -291,6 +292,37 @@ def recv(self, bufsize, flags=0):
             finally:
                 self._sender.socket = old_sock
 
+    def test_ipv6_only(self):
+        # Test if our host supports IPv6 before running this test
+        try:
+            socket.gethostbyaddr('::1')
+        except socket.herror:
+            self.skipTest("Host does not support IPv6, cannot run this test")
+
+        self.tearDown()
+
+        real_getaddrinfo = socket.getaddrinfo
+
+        def _fake_getaddrinfo(host, port, family=0, type=0, proto=0, flags=0):
+            if family == socket.AF_INET:
+                raise socket.gaierror("mock: IPv6 Only")
+            else:
+                return real_getaddrinfo(host, port, family, type, proto, flags)
+
+        self._server = mockserver.MockRecvServer(host='localhost',
+                                                 inet_family=socket.AF_INET6)
+
+
+        with patch('socket.getaddrinfo', side_effect=_fake_getaddrinfo):
+            sender = fluent.sender.FluentSender(tag='test',
+                                                host='localhost',
+                                                port=self._server.port)
+            sender.emit('foo', {'bar': 'baz'})
+            sender._close()
+            data = self.get_data()
+            self.assertEqual(len(data), 1)
+            self.assertEqual(data[0][2], {'bar': 'baz'})
+
     @unittest.skipIf(sys.platform == "win32", "Unix socket not supported")
     def test_unix_socket(self):
         self.tearDown()