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
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)
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)
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)
def test_main(self): self.assertEqual(_main(), None)