|
5 | 5 | package io.strimzi.operator.cluster.model; |
6 | 6 |
|
7 | 7 | import io.strimzi.api.kafka.model.common.CertSecretSourceBuilder; |
8 | | -import io.strimzi.api.kafka.model.common.ConnectorState; |
9 | 8 | import io.strimzi.api.kafka.model.common.metrics.StrimziMetricsReporterBuilder; |
10 | 9 | import io.strimzi.api.kafka.model.connector.KafkaConnector; |
11 | 10 | import io.strimzi.api.kafka.model.mirrormaker2.KafkaMirrorMaker2; |
|
31 | 30 | import static org.hamcrest.CoreMatchers.nullValue; |
32 | 31 | import static org.hamcrest.MatcherAssert.assertThat; |
33 | 32 |
|
34 | | -@SuppressWarnings("deprecation") // Uses deprecated getPause() field in tests |
35 | 33 | public class KafkaMirrorMaker2ConnectorsTest { |
36 | 34 | private static final String PREFIX = "prefix."; |
37 | 35 |
|
@@ -103,23 +101,20 @@ public void testConnectors() { |
103 | 101 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorSourceConnector")); |
104 | 102 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorSourceConnector")); |
105 | 103 | assertThat(kc.getSpec().getTasksMax(), is(5)); |
106 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
107 | 104 | assertThat(kc.getSpec().getState(), is(nullValue())); |
108 | 105 | assertThat(kc.getSpec().getConfig(), is(expectedSource)); |
109 | 106 |
|
110 | 107 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorCheckpointConnector")).findFirst().orElseThrow(); |
111 | 108 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorCheckpointConnector")); |
112 | 109 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorCheckpointConnector")); |
113 | 110 | assertThat(kc.getSpec().getTasksMax(), is(3)); |
114 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
115 | 111 | assertThat(kc.getSpec().getState(), is(nullValue())); |
116 | 112 | assertThat(kc.getSpec().getConfig(), is(expectedCheckpoint)); |
117 | 113 |
|
118 | 114 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorHeartbeatConnector")).findFirst().orElseThrow(); |
119 | 115 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorHeartbeatConnector")); |
120 | 116 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorHeartbeatConnector")); |
121 | 117 | assertThat(kc.getSpec().getTasksMax(), is(1)); |
122 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
123 | 118 | assertThat(kc.getSpec().getState(), is(nullValue())); |
124 | 119 | assertThat(kc.getSpec().getConfig(), is(expectedAll)); |
125 | 120 | } |
@@ -160,7 +155,6 @@ public void testOverridingSourceAndTargetConfiguration() { |
160 | 155 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorSourceConnector")); |
161 | 156 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorSourceConnector")); |
162 | 157 | assertThat(kc.getSpec().getTasksMax(), is(nullValue())); |
163 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
164 | 158 | assertThat(kc.getSpec().getState(), is(nullValue())); |
165 | 159 | assertThat(kc.getSpec().getConfig(), is(expectedSource)); |
166 | 160 | } |
@@ -234,47 +228,41 @@ public void testConnectorsWithMultipleSources() { |
234 | 228 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorSourceConnector")); |
235 | 229 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorSourceConnector")); |
236 | 230 | assertThat(kc.getSpec().getTasksMax(), is(5)); |
237 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
238 | 231 | assertThat(kc.getSpec().getState(), is(nullValue())); |
239 | 232 | assertThat(kc.getSpec().getConfig(), is(expectedSource)); |
240 | 233 |
|
241 | 234 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorCheckpointConnector")).findFirst().orElseThrow(); |
242 | 235 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorCheckpointConnector")); |
243 | 236 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorCheckpointConnector")); |
244 | 237 | assertThat(kc.getSpec().getTasksMax(), is(3)); |
245 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
246 | 238 | assertThat(kc.getSpec().getState(), is(nullValue())); |
247 | 239 | assertThat(kc.getSpec().getConfig(), is(expectedCheckpoint)); |
248 | 240 |
|
249 | 241 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorHeartbeatConnector")).findFirst().orElseThrow(); |
250 | 242 | assertThat(kc.getMetadata().getName(), is("source->target.MirrorHeartbeatConnector")); |
251 | 243 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorHeartbeatConnector")); |
252 | 244 | assertThat(kc.getSpec().getTasksMax(), is(1)); |
253 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
254 | 245 | assertThat(kc.getSpec().getState(), is(nullValue())); |
255 | 246 | assertThat(kc.getSpec().getConfig(), is(expectedAll)); |
256 | 247 |
|
257 | 248 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("other-source->target.MirrorSourceConnector")).findFirst().orElseThrow(); |
258 | 249 | assertThat(kc.getMetadata().getName(), is("other-source->target.MirrorSourceConnector")); |
259 | 250 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorSourceConnector")); |
260 | 251 | assertThat(kc.getSpec().getTasksMax(), is(15)); |
261 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
262 | 252 | assertThat(kc.getSpec().getState(), is(nullValue())); |
263 | 253 | assertThat(kc.getSpec().getConfig(), is(expectedSource2)); |
264 | 254 |
|
265 | 255 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("other-source->target.MirrorCheckpointConnector")).findFirst().orElseThrow(); |
266 | 256 | assertThat(kc.getMetadata().getName(), is("other-source->target.MirrorCheckpointConnector")); |
267 | 257 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorCheckpointConnector")); |
268 | 258 | assertThat(kc.getSpec().getTasksMax(), is(13)); |
269 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
270 | 259 | assertThat(kc.getSpec().getState(), is(nullValue())); |
271 | 260 | assertThat(kc.getSpec().getConfig(), is(expectedCheckpoint2)); |
272 | 261 |
|
273 | 262 | kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("other-source->target.MirrorHeartbeatConnector")).findFirst().orElseThrow(); |
274 | 263 | assertThat(kc.getMetadata().getName(), is("other-source->target.MirrorHeartbeatConnector")); |
275 | 264 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorHeartbeatConnector")); |
276 | 265 | assertThat(kc.getSpec().getTasksMax(), is(11)); |
277 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
278 | 266 | assertThat(kc.getSpec().getState(), is(nullValue())); |
279 | 267 | assertThat(kc.getSpec().getConfig(), is(expectedAll2)); |
280 | 268 | } |
@@ -303,49 +291,6 @@ public void testConnectorsOnlySome() { |
303 | 291 | assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorHeartbeatConnector")); |
304 | 292 | } |
305 | 293 |
|
306 | | - @Test |
307 | | - public void testConnectorsPauseState() { |
308 | | - KafkaMirrorMaker2 kmm2 = new KafkaMirrorMaker2Builder(KMM2) |
309 | | - .editSpec() |
310 | | - .editMirror(0) |
311 | | - .editSourceConnector() |
312 | | - .withState(ConnectorState.PAUSED) |
313 | | - .endSourceConnector() |
314 | | - .editCheckpointConnector() |
315 | | - .withPause(true) |
316 | | - .endCheckpointConnector() |
317 | | - .editHeartbeatConnector() |
318 | | - .withState(ConnectorState.STOPPED) |
319 | | - .withPause(true) |
320 | | - .endHeartbeatConnector() |
321 | | - .endMirror() |
322 | | - .endSpec() |
323 | | - .build(); |
324 | | - |
325 | | - KafkaMirrorMaker2Connectors connectors = KafkaMirrorMaker2Connectors.fromCrd(Reconciliation.DUMMY_RECONCILIATION, kmm2); |
326 | | - List<KafkaConnector> kcs = connectors.generateConnectorDefinitions(); |
327 | | - |
328 | | - assertThat(kcs.size(), is(3)); |
329 | | - |
330 | | - KafkaConnector kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorSourceConnector")).findFirst().orElseThrow(); |
331 | | - assertThat(kc.getMetadata().getName(), is("source->target.MirrorSourceConnector")); |
332 | | - assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorSourceConnector")); |
333 | | - assertThat(kc.getSpec().getPause(), is(nullValue())); |
334 | | - assertThat(kc.getSpec().getState(), is(ConnectorState.PAUSED)); |
335 | | - |
336 | | - kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorCheckpointConnector")).findFirst().orElseThrow(); |
337 | | - assertThat(kc.getMetadata().getName(), is("source->target.MirrorCheckpointConnector")); |
338 | | - assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorCheckpointConnector")); |
339 | | - assertThat(kc.getSpec().getPause(), is(true)); |
340 | | - assertThat(kc.getSpec().getState(), is(nullValue())); |
341 | | - |
342 | | - kc = kcs.stream().filter(k -> k.getMetadata().getName().contains("source->target.MirrorHeartbeatConnector")).findFirst().orElseThrow(); |
343 | | - assertThat(kc.getMetadata().getName(), is("source->target.MirrorHeartbeatConnector")); |
344 | | - assertThat(kc.getSpec().getClassName(), is("org.apache.kafka.connect.mirror.MirrorHeartbeatConnector")); |
345 | | - assertThat(kc.getSpec().getPause(), is(true)); |
346 | | - assertThat(kc.getSpec().getState(), is(ConnectorState.STOPPED)); |
347 | | - } |
348 | | - |
349 | 294 | @Test |
350 | 295 | public void testConnectorsWithAutoRestart() { |
351 | 296 | KafkaMirrorMaker2 kmm2 = new KafkaMirrorMaker2Builder(KMM2) |
|
0 commit comments