Merge pull request fluent#80 from kiseok7/master

repeatedly · web-flow · commit 45fff4bc3f7a · 2017-04-11T19:16:26.000+09:00
Add context manager to FluentSender in fluent/sender.py
diff --git a/fluent/sender.py b/fluent/sender.py
@@ -198,3 +198,9 @@ def _close(self):
         if self.socket:
             self.socket.close()
         self.socket = None
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, typ, value, traceback):
+        self.close()
diff --git a/tests/test_sender.py b/tests/test_sender.py
@@ -65,6 +65,18 @@ def test_simple(self):
         self.assertTrue(data[0][1])
         self.assertTrue(isinstance(data[0][1], int))
 
+    def test_decorator_simple(self):
+        with self._sender as sender:
+            sender.emit('foo', {'bar': 'baz'})
+        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.assertTrue(data[0][1])
+        self.assertTrue(isinstance(data[0][1], int))
+
     def test_nanosecond(self):
         sender = self._sender
         sender.nanosecond_precision = True
