|
24 | 24 | import org.apache.flink.cdc.composer.definition.SinkDef; |
25 | 25 | import org.apache.flink.cdc.composer.utils.FactoryDiscoveryUtils; |
26 | 26 | import org.apache.flink.cdc.composer.utils.factory.DataSinkFactory1; |
| 27 | +import org.apache.flink.configuration.PipelineOptions; |
| 28 | +import org.apache.flink.configuration.ReadableConfig; |
| 29 | +import org.apache.flink.core.fs.Path; |
27 | 30 |
|
28 | 31 | import org.apache.flink.shaded.guava31.com.google.common.collect.ImmutableMap; |
29 | 32 |
|
30 | 33 | import org.assertj.core.api.Assertions; |
31 | 34 | import org.junit.jupiter.api.Test; |
32 | 35 |
|
| 36 | +import java.time.Duration; |
| 37 | +import java.util.ArrayList; |
| 38 | +import java.util.List; |
| 39 | + |
| 40 | +import static org.apache.flink.configuration.CheckpointingOptions.CHECKPOINTING_INTERVAL; |
| 41 | + |
33 | 42 | /** A test for the {@link FlinkPipelineComposer}. */ |
34 | 43 | class FlinkPipelineComposerTest { |
35 | 44 |
|
@@ -59,4 +68,21 @@ void testCreateDataSinkFromSinkDef() { |
59 | 68 | Assertions.assertThat(((DataSinkFactory1.TestDataSink) dataSink).getHost()) |
60 | 69 | .isEqualTo("0.0.0.0"); |
61 | 70 | } |
| 71 | + |
| 72 | + @Test |
| 73 | + void testOfMiniCluster() { |
| 74 | + org.apache.flink.configuration.Configuration flinkConfig = |
| 75 | + new org.apache.flink.configuration.Configuration(); |
| 76 | + flinkConfig.set(CHECKPOINTING_INTERVAL, Duration.ofSeconds(30)); |
| 77 | + List<Path> additionalJars = new ArrayList<>(); |
| 78 | + additionalJars.add(new Path("/path/to/additionalJars.jar")); |
| 79 | + FlinkPipelineComposer flinkPipelineComposer = |
| 80 | + FlinkPipelineComposer.ofMiniCluster(flinkConfig, additionalJars); |
| 81 | + ReadableConfig configuration = flinkPipelineComposer.getEnv().getConfiguration(); |
| 82 | + |
| 83 | + Assertions.assertThat(configuration.get(CHECKPOINTING_INTERVAL)) |
| 84 | + .isEqualTo(Duration.ofSeconds(30)); |
| 85 | + Assertions.assertThat(configuration.get(PipelineOptions.JARS)) |
| 86 | + .contains("file:/path/to/additionalJars.jar"); |
| 87 | + } |
62 | 88 | } |
0 commit comments