36 lines
948 B
Python
Executable File
36 lines
948 B
Python
Executable File
import unittest
|
|
import sender
|
|
|
|
class Test_Compute_Relative_Time(unittest.TestCase) :
|
|
def test_basic(self) :
|
|
class MockDateTime() :
|
|
t = 1000000
|
|
def __init__(self) :
|
|
self.datetime = self
|
|
def fromtimestamp(self, t) :
|
|
self.t = t
|
|
def timestamp(self) :
|
|
return self.t
|
|
def utcnow(self) :
|
|
return self
|
|
mock_date_time = MockDateTime()
|
|
was = sender.datetime
|
|
sender.datetime = mock_date_time
|
|
self.case("1s", 1)
|
|
self.case("5s", 5)
|
|
self.case("0s", 0)
|
|
self.case("0m", 0)
|
|
self.case("1m", 60)
|
|
self.case("9m", 9*60)
|
|
self.case("0h", 0)
|
|
self.case("1h", 60*60)
|
|
self.case("9h", 9*60*60)
|
|
sender.datetime = was
|
|
|
|
def case(self, duration, expected) :
|
|
seconds = sender.parse_go_duration(duration)
|
|
self.assertEqual(seconds, expected)
|
|
|
|
if __name__ == "__main__" :
|
|
unittest.main()
|