52 lines
1.6 KiB
Python
Executable File
52 lines
1.6 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
import unittest
|
|
import lmwsip
|
|
import stubSipServer
|
|
|
|
from datetime import datetime, timedelta
|
|
from dateutil import tz
|
|
|
|
class lmwsipTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.sipserver = stubSipServer.sipServer()
|
|
self.sipserver.run()
|
|
self.sip = lmwsip.LmwSip("USER", "PASS", "localhost",
|
|
self.sipserver.port, ssl=False)
|
|
|
|
def tearDown(self):
|
|
self.sipserver.kill()
|
|
self.sip.closesocket()
|
|
|
|
def test_sipobj(self):
|
|
self.assertEqual(type(self.sip), lmwsip.LmwSip)
|
|
|
|
def test_ti(self):
|
|
self.assertEqual(type(self.sip.ti()), str)
|
|
|
|
def test_cmd(self):
|
|
self.assertEqual(type(self.sip.cmd("WN", "DUMMY", "D10", "+00:59", "2020-01-01", "00:00")), str)
|
|
|
|
def test_logout(self):
|
|
self.assertEqual(self.sip.logout(), None)
|
|
|
|
def test_lmwTimeSerie(self):
|
|
timezone = tz.gettz('GMT+1')
|
|
res = self.sip.timeSerie("WN", "DUMMY", "D10",
|
|
datetime.now(timezone)-timedelta(minutes=60),
|
|
datetime.now(timezone))
|
|
self.assertEqual(type(res.ts), list)
|
|
self.assertEqual(len(res.ts), 6)
|
|
self.assertEqual(res.ts[1][1][0], '1')
|
|
|
|
def test_roundtime(self):
|
|
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)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|