Esempio n. 1
0
class TestCategoryDayIndex(unittest.TestCase):
    def setUp(self):
        DBMgr.getInstance().startRequest()
        self.oldIndex = IndexesHolder()._getIdx()["categoryDateLtd"]
        self.newIndex = IndexesHolder()._getIdx()["categoryDate"]
        self.startDate = datetime(2010,
                                  5,
                                  13,
                                  10,
                                  0,
                                  0,
                                  tzinfo=timezone('UTC'))
        self.endDate = datetime(2010, 5, 14, 14, 0, 0, tzinfo=timezone('UTC'))
        self.ch = ConferenceHolder()
        self.categId = '0'

    def tearDownModule(self):
        DBMgr.getInstance().abort()
        DBMgr.getInstance().endRequest()

    def testGetObjectsStartingInDay(self):
        newRes = self.newIndex._idxCategItem[
            self.categId].getObjectsStartingInDay(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[
            self.categId].getObjectsStartingInDay(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))

        assert (oldRes == newRes)

    def testGetObjectsStartingIn(self):
        newRes = self.newIndex.getObjectsStartingIn(self.categId,
                                                    self.startDate,
                                                    self.endDate)
        oldTmp = self.oldIndex.getObjectsStartingIn(self.categId,
                                                    self.startDate,
                                                    self.endDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)

    def testGetObjectsStartingAfterTest(self):
        newRes = self.newIndex._idxCategItem[
            self.categId].getObjectsStartingAfter(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[
            self.categId].getObjectsStartingAfter(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)

    def testGetObjectsEndingAfterTest(self):
        newRes = self.newIndex._idxCategItem[
            self.categId].getObjectsEndingAfter(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[
            self.categId].getObjectsEndingAfter(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)

    def testGetObjectsEndingInDay(self):
        newRes = self.newIndex._idxCategItem[
            self.categId].getObjectsEndingInDay(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[
            self.categId].getObjectsEndingInDay(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)

    def testGetObjectsEndingIn(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsEndingIn(
            self.startDate, self.endDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsEndingIn(
            self.startDate, self.endDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)

    def testGetObjectsInDay(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsInDay(
            self.startDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsInDay(
            self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)

    def testGetObjectsIn(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsIn(
            self.startDate, self.endDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsIn(
            self.startDate, self.endDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert (oldRes == newRes)
Esempio n. 2
0
class TestCategoryDayIndex(unittest.TestCase):

    def setUp(self):
        DBMgr.getInstance().startRequest()
        self.oldIndex = IndexesHolder()._getIdx()["categoryDateLtd"]
        self.newIndex = IndexesHolder()._getIdx()["categoryDate"]
        self.startDate = datetime(2010,5,13, 10, 0, 0, tzinfo=timezone('UTC'))
        self.endDate = datetime(2010,5,14, 14, 0, 0, tzinfo=timezone('UTC'))
        self.ch = ConferenceHolder()
        self.categId = '0'

    def tearDownModule(self):
        DBMgr.getInstance().abort()
        DBMgr.getInstance().endRequest()

    def testGetObjectsStartingInDay(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsStartingInDay(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsStartingInDay(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))

        assert(oldRes == newRes)

    def testGetObjectsStartingIn(self):
        newRes = self.newIndex.getObjectsStartingIn(self.categId, self.startDate, self.endDate)
        oldTmp = self.oldIndex.getObjectsStartingIn(self.categId, self.startDate, self.endDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)

    def testGetObjectsStartingAfterTest(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsStartingAfter(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsStartingAfter(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)

    def testGetObjectsEndingAfterTest(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsEndingAfter(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsEndingAfter(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)

    def testGetObjectsEndingInDay(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsEndingInDay(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsEndingInDay(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)

    def testGetObjectsEndingIn(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsEndingIn(self.startDate,self.endDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsEndingIn(self.startDate,self.endDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)

    def testGetObjectsInDay(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsInDay(self.startDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsInDay(self.startDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)

    def testGetObjectsIn(self):
        newRes = self.newIndex._idxCategItem[self.categId].getObjectsIn(self.startDate,self.endDate)
        oldTmp = self.oldIndex._idxCategItem[self.categId].getObjectsIn(self.startDate,self.endDate)
        oldRes = set()
        for res in oldTmp:
            oldRes.add(self.ch.getById(res))
        assert(oldRes == newRes)