예제 #1
0
    def test_main(self):
        with patch.object(WALERestore, 'run', Mock(return_value=0)):
            self.assertEqual(_main(), 0)

        with patch.object(WALERestore, 'run', Mock(return_value=1)), \
             patch('time.sleep', Mock(return_value=None)) as mock_sleep:
            self.assertEqual(_main(), 1)
            assert mock_sleep.call_count == WALE_TEST_RETRIES
예제 #2
0
    def test_main(self):
        self.__thread_ident = current_thread().ident
        sleeps = [0]

        def mock_sleep(*args):
            if current_thread().ident == self.__thread_ident:
                sleeps[0] += 1

        with patch.object(WALERestore, 'run', Mock(return_value=0)):
            self.assertEqual(_main(), 0)

        with patch.object(WALERestore, 'run', Mock(return_value=1)), \
                patch('time.sleep', mock_sleep):
            self.assertEqual(_main(), 1)
            self.assertTrue(sleeps[0], WALE_TEST_RETRIES)
예제 #3
0
 def test_main(self):
     with patch.object(WALERestore, 'run', Mock(return_value=0)):
         self.assertEqual(_main(), 0)
     with patch.object(WALERestore, 'run', Mock(return_value=1)):
         self.assertEqual(_main(), 1)
예제 #4
0
 def test_main(self):
     with patch.object(WALERestore, "run", Mock(return_value=0)):
         self.assertEqual(_main(), 0)
     with patch.object(WALERestore, "run", Mock(return_value=1)):
         self.assertEqual(_main(), 1)
예제 #5
0
 def test_main(self):
     self.assertEqual(_main(), None)
예제 #6
0
 def test_main(self):
     self.assertEqual(_main(), None)