def test_set_get_restart_strategy(self):
self.env.set_restart_strategy(RestartStrategies.no_restart())
restart_strategy = self.env.get_restart_strategy()
self.assertEqual(restart_strategy, RestartStrategies.no_restart())
def test_get_set_restart_strategy(self):
self.execution_config.set_restart_strategy(
RestartStrategies.no_restart())
self.assertEqual(self.execution_config.get_restart_strategy(),
RestartStrategies.no_restart())
self.execution_config.set_restart_strategy(
RestartStrategies.failure_rate_restart(5, 10000, 5000))
self.assertIsInstance(
self.execution_config.get_restart_strategy(),
RestartStrategies.FailureRateRestartStrategyConfiguration)
self.execution_config.set_restart_strategy(
RestartStrategies.fixed_delay_restart(4, 10000))
self.assertIsInstance(
self.execution_config.get_restart_strategy(),
RestartStrategies.FixedDelayRestartStrategyConfiguration)
self.execution_config.set_restart_strategy(
RestartStrategies.fall_back_restart())
self.assertEqual(self.execution_config.get_restart_strategy(),
RestartStrategies.fall_back_restart())
def setUp(self):
self.env = StreamExecutionEnvironment.get_execution_environment()
self._load_dependency_jars()
config = Configuration(
j_configuration=get_j_env_configuration(self.env._j_stream_execution_environment))
config.set_boolean("execution.checkpointing.checkpoints-after-tasks-finish.enabled", True)
self.env.set_parallelism(4)
self.env.enable_checkpointing(100)
self.env.set_restart_strategy(RestartStrategies.no_restart())
self.t_env = StreamTableEnvironment.create(self.env)
self.temp_dir = tempfile.mkdtemp()
