Extra test voor lmwsip.run.run()

test/__init__.py

@@ -1,37 +1,52 @@
 #!/usr/bin/python
 
+import sys
+import io
 import unittest
 import lmwsip
 import stubSipServer
 
+from lmwsip.run import run
 from datetime import datetime, timedelta
 from dateutil import tz
 
+class myTestArgs():
+    pass
+
 class lmwsipTest(unittest.TestCase):
 
     def setUp(self):
         self.sipserver = stubSipServer.sipServer()
+        self.sip       = None
         self.sipserver.run()
-        self.sip       = lmwsip.LmwSip("USER", "PASS", "localhost",
-                                       self.sipserver.port, ssl=False)
+
+    def login(self):
+        self.sip = lmwsip.LmwSip("USER", "PASS", "localhost",
+                                  self.sipserver.port, ssl=False)
 
     def tearDown(self):
+        if self.sip:
+            self.sip.closesocket()
         self.sipserver.kill()
-        self.sip.closesocket()
 
     def test_sipobj(self):
+        self.login()
         self.assertEqual(type(self.sip), lmwsip.LmwSip)
 
     def test_ti(self):
+        self.login()
         self.assertEqual(type(self.sip.ti()), str)
 
     def test_cmd(self):
+        self.login()
         self.assertEqual(type(self.sip.cmd("WN", "DUMMY", "D10", "+00:59", "2020-01-01", "00:00")), str)
 
     def test_logout(self):
+        self.login()
         self.assertEqual(self.sip.logout(), None)
 
     def test_lmwTimeSerie(self):
+        self.login()
         timezone = tz.gettz('GMT+1')
         res = self.sip.timeSerie("WN", "DUMMY", "D10",
                                   datetime.now(timezone)-timedelta(minutes=60),
@@ -41,11 +56,30 @@ class lmwsipTest(unittest.TestCase):
         self.assertEqual(res.ts[1][1][0], '1')
 
     def test_roundtime(self):
+        self.login()
         timezone = tz.gettz('GMT+1')
         t1 = datetime(2020, 1, 1, 0, 10, 0, 0, timezone)
         t2 = datetime(2020, 1, 1, 0,  0, 0, 1, timezone)
         self.assertEqual(self.sip._roundtime_(t1, "D10"), t1)
         self.assertEqual(self.sip._roundtime_(t2, "D10"), t1)
 
+    def test_run(self):
+        capturedOutput = io.StringIO() 
+        sys.stdout = capturedOutput
+        args = myTestArgs()
+        args.debug       = "WARN"
+        args.host        = "localhost"
+        args.port        = self.sipserver.port
+        args.unencrypted = True
+        args.cleartelnet = False
+        args.time        = "+00:59"
+        args.date        = "2020-01-01"
+        args.files       = [open("test/test.sip")]
+        run(args)
+        args.files[0].close()
+        self.assertEqual(capturedOutput.getvalue().find("!")>= 0, True)
+        self.assertEqual(capturedOutput.getvalue().find("?"), -1)
+        sys.stdout = sys.__stdout__
+
 if __name__ == '__main__':
     unittest.main()