Add test for sender.

methane · methane · commit c63361536a23 · 2011-10-04T17:32:04.000+09:00
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -2,4 +2,4 @@
 sys.path = ['..'] + sys.path
 
 from test_handler import *
-
+from test_sender import *
diff --git a/tests/mockserver.py b/tests/mockserver.py
@@ -0,0 +1,39 @@
+import socket
+import threading
+import time
+from cStringIO import StringIO
+from msgpack import Unpacker
+
+class MockRecvServer(threading.Thread):
+    """
+    Single threaded server accepts one connection and recv until EOF.
+    """
+    def __init__(self, port):
+        self._sock = socket.socket()
+        self._sock.bind(('localhost', port))
+        self._buf = StringIO()
+
+        threading.Thread.__init__(self)
+        self.start()
+
+    def run(self):
+        s = self._sock
+        s.listen(1)
+        con, _ = s.accept()
+        while True:
+            d = con.recv(4096)
+            if not d:
+                break
+            self._buf.write(d)
+        con.close()
+        s.close()
+        self._sock = None
+
+    def wait(self):
+        while self._sock:
+            time.sleep(0.1)
+
+    def get_recieved(self):
+        self.wait()
+        self._buf.seek(0)
+        return list(Unpacker(self._buf))
diff --git a/tests/test_sender.py b/tests/test_sender.py
@@ -0,0 +1,35 @@
+import unittest
+import mockserver
+import fluent.sender
+import msgpack
+
+class TestSender(unittest.TestCase):
+    def setUp(self):
+        super(TestSender, self).setUp()
+        for port in xrange(10000, 20000):
+            try:
+                self._server = mockserver.MockRecvServer(port)
+                break
+            except IOError, e:
+                print e
+                pass
+        self._sender = fluent.sender.FluentSender(
+                tag='test',
+                port=port,
+                )
+
+    def get_data(self):
+        return self._server.get_recieved()
+
+    def test_simple(self):
+        sender = self._sender
+        sender.emit('foo', {'bar': 'baz'})
+        sender._close()
+        data = self.get_data()
+        eq = self.assertEqual
+        eq(1, len(data))
+        eq(3, len(data[0]))
+        eq('test.foo', data[0][0])
+        eq({'bar':'baz'}, data[0][2])
+        self.assert_(data[0][1])
+        self.assert_(isinstance(data[0][1], int))
