Merge branch 'master' of github.com:fluent/fluent-logger-python

Author: repeatedly

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()

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