Addded tests for sender.setup

hirokiky · hirokiky · commit 90294220db0e · 2015-06-09T12:13:09.000+09:00
* Cause there wasnt
* To check the new tolerant setup feature
* Added new private function for these tests
diff --git a/fluent/sender.py b/fluent/sender.py
@@ -11,6 +11,13 @@
 _global_sender = None
 
 
+def _set_global_sender(sender):
+    """ [For testing] Function to set global sender directly
+    """
+    global _global_sender
+    _global_sender = sender
+
+
 def setup(tag, **kwargs):
     host = kwargs.get('host', 'localhost')
     port = kwargs.get('port', 24224)
diff --git a/tests/test_sender.py b/tests/test_sender.py
@@ -8,6 +8,36 @@
 from tests import mockserver
 
 
+class TestSetup(unittest.TestCase):
+    def tearDown(self):
+        from fluent.sender import _set_global_sender
+        _set_global_sender(None)
+
+    def test_no_kwargs(self):
+        fluent.sender.setup("tag")
+        actual = fluent.sender.get_global_sender()
+        self.assertEqual(actual.tag, "tag")
+        self.assertEqual(actual.host, "localhost")
+        self.assertEqual(actual.port, 24224)
+        self.assertEqual(actual.timeout, 3.0)
+
+    def test_host_and_port(self):
+        fluent.sender.setup("tag", host="myhost", port=24225)
+        actual = fluent.sender.get_global_sender()
+        self.assertEqual(actual.tag, "tag")
+        self.assertEqual(actual.host, "myhost")
+        self.assertEqual(actual.port, 24225)
+        self.assertEqual(actual.timeout, 3.0)
+
+    def test_tolerant(self):
+        fluent.sender.setup("tag", host="myhost", port=24225, timeout=1.0)
+        actual = fluent.sender.get_global_sender()
+        self.assertEqual(actual.tag, "tag")
+        self.assertEqual(actual.host, "myhost")
+        self.assertEqual(actual.port, 24225)
+        self.assertEqual(actual.timeout, 1.0)
+
+
 class TestSender(unittest.TestCase):
     def setUp(self):
         super(TestSender, self).setUp()
