source: geniuswebcore/test/geniusweb/progress/ProgressTimeTest.py

from datetime import datetime
import json
from typing import List
import unittest

from pyson.ObjectMapper import ObjectMapper
from unitpy.GeneralTests import GeneralTests

from geniusweb.progress.Progress import Progress
from geniusweb.progress.ProgressTime import ProgressTime


class ProgressTimeTest (unittest.TestCase, GeneralTests[Progress]):
        # we set T0 quite high because Python has trouble handling small datetime
        T0 = 100000
        TIMEZERO = datetime.fromtimestamp(T0 / 1000)
        TESTTIME = 10  # milliseconds!
        progress1 = ProgressTime(TESTTIME, TIMEZERO)
        progress1a = ProgressTime(TESTTIME, TIMEZERO)
        progress2 = ProgressTime(200, TIMEZERO)

        jackson = ObjectMapper()
        progressstring = "{\"ProgressTime\":{\"duration\":10,\"start\":100000}}"

        start = 1234
        pr = ProgressTime(TESTTIME, datetime.fromtimestamp(start / 1000))

        def getGeneralTestData(self) -> List[List[Progress]]:
                return [[self.progress1, self.progress1a], [self.progress2]]
        
        def getGeneralTestStrings(self) -> List[str]:
                return ["ProgressTime\\[100000.*10ms\\]",
                                "ProgressTime\\[100000.*200ms\\]"]

        def testInit(self):
                self.assertEqual(0, self.progress1.get(0))

        def testAdvancingTime(self):
                for n in range(0, self.TESTTIME):
                        self.assertEqual(float(n) / self.TESTTIME,
                                        self.progress1.get(self.T0 + n))

        def testAdvanceWhenDeadlineReached(self):
                progress = ProgressTime(self.TESTTIME, self.TIMEZERO)
                self.assertEqual(1, self.progress1.get(self.T0 + self.TESTTIME));

        def testSerialize(self):
                actual = self.jackson.toJson(self.progress1)
                print(actual)
                self.assertEqual(json.loads(self.progressstring), actual)

        def testDeserialize(self):
                newprog = self.jackson.parse(json.loads(self.progressstring),   ProgressTime)
                # we can't directly compare with progress since that's a hacked
                # object...
                self.assertEqual(self.TESTTIME, newprog.getDuration())
                self.assertEqual(self.T0, 1000 * newprog.getStart().timestamp())

        def pastDeadlinetest(self):
                self.assertTrue(self.pr.isPastDeadline(self.start + self.TESTTIME + 1))
                self.assertFalse(self.pr.isPastDeadline(self.start + self.TESTTIME - 1))

        def terminationTimetest(self):
                self.assertEqual(self.start + self.TESTTIME, self.pr.getTerminationTime().getTime())

        def gettest(self):
                self.assertEqual(0, self.pr.get(self.start))
                self.assertEqual(1, self.pr.get(self.start + self.TESTTIME));
                self.assertEqual(0.5, self.pr.get(self.start + self.TESTTIME / 2));
                self.assertEqual(0, self.pr.get(self.start - 10))
                self.assertEqual(1, self.pr.get(self.start + self.TESTTIME + 10))