def setUp(self):
     self.history = []
     self.clock = Clock()
     self.pps = PreProcessingService(
         self.fakeServiceCreator, None, "store",
         None, "storageService", reactor=self.clock
     )
Exemple #2
0
 def setUp(self):
     self.history = []
     self.clock = Clock()
     self.pps = PreProcessingService(
         self.fakeServiceCreator, None, "store",
         None, "storageService", reactor=self.clock
     )
class PreProcessingServiceTestCase(TestCase):
    def fakeServiceCreator(self, cp, store, lo, storageService):
        self.history.append(("serviceCreator", store, storageService))

    def setUp(self):
        self.history = []
        self.clock = Clock()
        self.pps = PreProcessingService(
            self.fakeServiceCreator, None, "store", None, "storageService", reactor=self.clock
        )

    def _record(self, value, failure):
        self.history.append(value)

    def test_allSuccess(self):
        self.pps.addStep(StepOne(self._record, False)).addStep(StepTwo(self._record, False)).addStep(
            StepThree(self._record, False)
        ).addStep(StepFour(self._record, False))
        self.pps.startService()
        self.assertEquals(
            self.history,
            [
                "one success",
                "two success",
                "three success",
                "four success",
                ("serviceCreator", "store", "storageService"),
            ],
        )

    def test_allFailure(self):
        self.pps.addStep(StepOne(self._record, True)).addStep(StepTwo(self._record, True)).addStep(
            StepThree(self._record, True)
        ).addStep(StepFour(self._record, True))
        self.pps.startService()
        self.assertEquals(
            self.history,
            ["one success", "two failure", "three failure", "four failure", ("serviceCreator", None, "storageService")],
        )

    def test_partialFailure(self):
        self.pps.addStep(StepOne(self._record, True)).addStep(StepTwo(self._record, False)).addStep(
            StepThree(self._record, True)
        ).addStep(StepFour(self._record, False))
        self.pps.startService()
        self.assertEquals(
            self.history,
            [
                "one success",
                "two failure",
                "three success",
                "four failure",
                ("serviceCreator", "store", "storageService"),
            ],
        )
class PreProcessingServiceTestCase(TestCase):

    def fakeServiceCreator(self, cp, store, lo, storageService):
        self.history.append(("serviceCreator", store, storageService))


    def setUp(self):
        self.history = []
        self.clock = Clock()
        self.pps = PreProcessingService(self.fakeServiceCreator, None, "store",
            None, "storageService", reactor=self.clock)


    def _record(self, value, failure):
        self.history.append(value)


    def test_allSuccess(self):
        self.pps.addStep(
            StepOne(self._record, False)
        ).addStep(
            StepTwo(self._record, False)
        ).addStep(
            StepThree(self._record, False)
        ).addStep(
            StepFour(self._record, False)
        )
        self.pps.startService()
        self.assertEquals(self.history,
            ['one success', 'two success', 'three success', 'four success',
            ('serviceCreator', 'store', 'storageService')])


    def test_allFailure(self):
        self.pps.addStep(
            StepOne(self._record, True)
        ).addStep(
            StepTwo(self._record, True)
        ).addStep(
            StepThree(self._record, True)
        ).addStep(
            StepFour(self._record, True)
        )
        self.pps.startService()
        self.assertEquals(self.history,
            ['one success', 'two failure', 'three failure', 'four failure',
            ('serviceCreator', None, 'storageService')])


    def test_partialFailure(self):
        self.pps.addStep(
            StepOne(self._record, True)
        ).addStep(
            StepTwo(self._record, False)
        ).addStep(
            StepThree(self._record, True)
        ).addStep(
            StepFour(self._record, False)
        )
        self.pps.startService()
        self.assertEquals(self.history,
            ['one success', 'two failure', 'three success', 'four failure',
            ('serviceCreator', 'store', 'storageService')])


    def test_quitAfterUpgradeStep(self):
        triggerFileName = "stop_after_upgrade"
        triggerFile = FilePath(triggerFileName)
        self.pps.addStep(
            StepOne(self._record, False)
        ).addStep(
            StepTwo(self._record, False)
        ).addStep(
            QuitAfterUpgradeStep(triggerFile.path, reactor=self.clock)
        ).addStep(
            StepFour(self._record, True)
        )
        triggerFile.setContent("")
        self.pps.startService()
        self.assertEquals(self.history,
            ['one success', 'two success', 'four failure',
            ('serviceCreator', None, 'storageService')])
        self.assertFalse(triggerFile.exists())
Exemple #5
0
class PreProcessingServiceTestCase(TestCase):
    def fakeServiceCreator(self, cp, store, lo, storageService):
        self.history.append(("serviceCreator", store, storageService))

    def setUp(self):
        self.history = []
        self.clock = Clock()
        self.pps = PreProcessingService(self.fakeServiceCreator,
                                        None,
                                        "store",
                                        None,
                                        "storageService",
                                        reactor=self.clock)

    def _record(self, value, failure):
        self.history.append(value)

    def test_allSuccess(self):
        self.pps.addStep(StepOne(self._record, False)).addStep(
            StepTwo(self._record,
                    False)).addStep(StepThree(self._record, False)).addStep(
                        StepFour(self._record, False))
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two success', 'three success', 'four success',
            ('serviceCreator', 'store', 'storageService')
        ])

    def test_allFailure(self):
        self.pps.addStep(StepOne(self._record, True)).addStep(
            StepTwo(self._record,
                    True)).addStep(StepThree(self._record, True)).addStep(
                        StepFour(self._record, True))
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two failure', 'three failure', 'four failure',
            ('serviceCreator', None, 'storageService')
        ])

    def test_partialFailure(self):
        self.pps.addStep(StepOne(self._record, True)).addStep(
            StepTwo(self._record, False)).addStep(StepThree(
                self._record, True)).addStep(StepFour(self._record, False))
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two failure', 'three success', 'four failure',
            ('serviceCreator', 'store', 'storageService')
        ])
class PreProcessingServiceTestCase(TestCase):
    def fakeServiceCreator(self, cp, store, lo, storageService):
        self.history.append(("serviceCreator", store, storageService))

    def setUp(self):
        self.history = []
        self.clock = Clock()
        self.pps = PreProcessingService(self.fakeServiceCreator,
                                        None,
                                        "store",
                                        None,
                                        "storageService",
                                        reactor=self.clock)

    def _record(self, value, failure):
        self.history.append(value)

    def test_allSuccess(self):
        self.pps.addStep(StepOne(self._record, False)).addStep(
            StepTwo(self._record,
                    False)).addStep(StepThree(self._record, False)).addStep(
                        StepFour(self._record, False))
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two success', 'three success', 'four success',
            ('serviceCreator', 'store', 'storageService')
        ])

    def test_allFailure(self):
        self.pps.addStep(StepOne(self._record, True)).addStep(
            StepTwo(self._record,
                    True)).addStep(StepThree(self._record, True)).addStep(
                        StepFour(self._record, True))
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two failure', 'three failure', 'four failure',
            ('serviceCreator', None, 'storageService')
        ])

    def test_partialFailure(self):
        self.pps.addStep(StepOne(self._record, True)).addStep(
            StepTwo(self._record, False)).addStep(StepThree(
                self._record, True)).addStep(StepFour(self._record, False))
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two failure', 'three success', 'four failure',
            ('serviceCreator', 'store', 'storageService')
        ])

    def test_quitAfterUpgradeStep(self):
        triggerFileName = "stop_after_upgrade"
        triggerFile = FilePath(triggerFileName)
        self.pps.addStep(StepOne(self._record, False)).addStep(
            StepTwo(self._record, False)).addStep(
                QuitAfterUpgradeStep(triggerFile.path,
                                     reactor=self.clock)).addStep(
                                         StepFour(self._record, True))
        triggerFile.setContent("")
        self.pps.startService()
        self.assertEquals(self.history, [
            'one success', 'two success', 'four failure',
            ('serviceCreator', None, 'storageService')
        ])
        self.assertFalse(triggerFile.exists())