Browse Source
Redpanda (https://redpanda.com/) is Kafka-compatible and Testcontainers provides a module. This commit adds support for creating KafkaConnectionDetails from a @ServiceConnection-annotated RedpandaContainer. See gh-34780pull/35090/head
6 changed files with 192 additions and 1 deletions
@ -0,0 +1,65 @@
@@ -0,0 +1,65 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.boot.testcontainers.service.connection.redpanda; |
||||
|
||||
import java.net.URI; |
||||
import java.util.List; |
||||
|
||||
import org.testcontainers.redpanda.RedpandaContainer; |
||||
|
||||
import org.springframework.boot.autoconfigure.kafka.KafkaConnectionDetails; |
||||
import org.springframework.boot.testcontainers.service.connection.ContainerConnectionDetailsFactory; |
||||
import org.springframework.boot.testcontainers.service.connection.ContainerConnectionSource; |
||||
import org.springframework.boot.testcontainers.service.connection.ServiceConnection; |
||||
|
||||
/** |
||||
* {@link ContainerConnectionDetailsFactory} to create {@link KafkaConnectionDetails} from |
||||
* a {@link ServiceConnection @ServiceConnection}-annotated {@link RedpandaContainer}. |
||||
* |
||||
* @author Eddú Meléndez |
||||
*/ |
||||
class RedpandaContainerConnectionDetailsFactory |
||||
extends ContainerConnectionDetailsFactory<KafkaConnectionDetails, RedpandaContainer> { |
||||
|
||||
@Override |
||||
protected KafkaConnectionDetails getContainerConnectionDetails( |
||||
ContainerConnectionSource<RedpandaContainer> source) { |
||||
return new RedpandaContainerConnectionDetails(source); |
||||
} |
||||
|
||||
/** |
||||
* {@link KafkaConnectionDetails} backed by a {@link ContainerConnectionSource}. |
||||
*/ |
||||
private static final class RedpandaContainerConnectionDetails extends ContainerConnectionDetails |
||||
implements KafkaConnectionDetails { |
||||
|
||||
private final RedpandaContainer container; |
||||
|
||||
private RedpandaContainerConnectionDetails(ContainerConnectionSource<RedpandaContainer> source) { |
||||
super(source); |
||||
this.container = source.getContainer(); |
||||
} |
||||
|
||||
@Override |
||||
public List<Node> getBootstrapNodes() { |
||||
URI uri = URI.create(this.container.getBootstrapServers()); |
||||
return List.of(new Node(uri.getHost(), uri.getPort())); |
||||
} |
||||
|
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,20 @@
@@ -0,0 +1,20 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
/** |
||||
* Support for testcontainers Redpanda service connections. |
||||
*/ |
||||
package org.springframework.boot.testcontainers.service.connection.redpanda; |
||||
@ -0,0 +1,93 @@
@@ -0,0 +1,93 @@
|
||||
/* |
||||
* Copyright 2012-2023 the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.boot.testcontainers.service.connection.redpanda; |
||||
|
||||
import java.time.Duration; |
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
import org.awaitility.Awaitility; |
||||
import org.junit.jupiter.api.Test; |
||||
import org.testcontainers.junit.jupiter.Container; |
||||
import org.testcontainers.junit.jupiter.Testcontainers; |
||||
import org.testcontainers.redpanda.RedpandaContainer; |
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.boot.autoconfigure.ImportAutoConfiguration; |
||||
import org.springframework.boot.autoconfigure.kafka.KafkaAutoConfiguration; |
||||
import org.springframework.boot.testcontainers.service.connection.ServiceConnection; |
||||
import org.springframework.boot.testsupport.testcontainers.DockerImageNames; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.kafka.annotation.KafkaListener; |
||||
import org.springframework.kafka.core.KafkaTemplate; |
||||
import org.springframework.test.context.TestPropertySource; |
||||
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
||||
/** |
||||
* Tests for {@link RedpandaContainerConnectionDetailsFactory}. |
||||
* |
||||
* @author Eddú Meléndez |
||||
*/ |
||||
@SpringJUnitConfig |
||||
@Testcontainers(disabledWithoutDocker = true) |
||||
@TestPropertySource(properties = { "spring.kafka.consumer.group-id=test-group", |
||||
"spring.kafka.consumer.auto-offset-reset=earliest" }) |
||||
class RedpandaContainerConnectionDetailsFactoryIntegrationTests { |
||||
|
||||
@Container |
||||
@ServiceConnection |
||||
static final RedpandaContainer redpanda = new RedpandaContainer(DockerImageNames.redpanda()); |
||||
|
||||
@Autowired |
||||
KafkaTemplate<String, String> kafkaTemplate; |
||||
|
||||
@Autowired |
||||
TestListener listener; |
||||
|
||||
@Test |
||||
void connectionCanBeMadeToRedpandaContainer() { |
||||
this.kafkaTemplate.send("test-topic", "test-data"); |
||||
Awaitility.waitAtMost(Duration.ofSeconds(30)) |
||||
.untilAsserted(() -> assertThat(this.listener.messages).containsExactly("test-data")); |
||||
} |
||||
|
||||
@Configuration(proxyBeanMethods = false) |
||||
@ImportAutoConfiguration(KafkaAutoConfiguration.class) |
||||
static class TestConfiguration { |
||||
|
||||
@Bean |
||||
TestListener testListener() { |
||||
return new TestListener(); |
||||
} |
||||
|
||||
} |
||||
|
||||
static class TestListener { |
||||
|
||||
private final List<String> messages = new ArrayList<>(); |
||||
|
||||
@KafkaListener(topics = "test-topic") |
||||
void processMessage(String message) { |
||||
this.messages.add(message); |
||||
} |
||||
|
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue