From 2f395bd6f999f978ac65be72f36789785265228b Mon Sep 17 00:00:00 2001
From: Santiago Jimenez Giraldo <sago2k8@gmail.com>
Date: Tue, 19 Dec 2023 11:12:10 +0100
Subject: [PATCH] test: test listeners for Redpanda Container

Add test for new WithListener function, validate connectivity and couple
of asserts in the construction of the listener

Signed-off-by: Santiago Jimenez Giraldo <sago2k8@gmail.com>
---
 modules/redpanda/redpanda_test.go | 109 ++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)

diff --git a/modules/redpanda/redpanda_test.go b/modules/redpanda/redpanda_test.go
index 7ad424b3bd..a8cedc1589 100644
--- a/modules/redpanda/redpanda_test.go
+++ b/modules/redpanda/redpanda_test.go
@@ -5,6 +5,7 @@ import (
 	"crypto/tls"
 	"crypto/x509"
 	"fmt"
+	"io"
 	"net/http"
 	"strings"
 	"testing"
@@ -12,6 +13,8 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	"github.com/testcontainers/testcontainers-go"
+	"github.com/testcontainers/testcontainers-go/network"
 	"github.com/twmb/franz-go/pkg/kadm"
 	"github.com/twmb/franz-go/pkg/kerr"
 	"github.com/twmb/franz-go/pkg/kgo"
@@ -278,6 +281,112 @@ func TestRedpandaWithTLS(t *testing.T) {
 	require.Error(t, results.FirstErr(), kerr.UnknownTopicOrPartition)
 }
 
+func TestRedpandaListener_Simple(t *testing.T) {
+	ctx := context.Background()
+	// 1. Create network
+	rpNetwork, err := network.New(ctx, network.WithCheckDuplicate())
+	require.NoError(t, err)
+
+	// 2. Start Redpanda container
+	container, err := RunContainer(ctx,
+		testcontainers.WithImage("redpandadata/redpanda:v23.2.18"),
+		network.WithNetwork([]string{"redpanda-host"}, rpNetwork),
+		WithListener("redpanda:29092"), WithAutoCreateTopics(),
+	)
+	require.NoError(t, err)
+
+	// 3. Start KCat container
+	kcat, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
+		ContainerRequest: testcontainers.ContainerRequest{
+			Image: "confluentinc/cp-kcat:7.4.1",
+			Networks: []string{
+				rpNetwork.Name,
+			},
+			Entrypoint: []string{
+				"sh",
+			},
+			Cmd: []string{
+				"-c",
+				"tail -f /dev/null",
+			},
+		},
+		Started: true,
+	})
+
+	require.NoError(t, err)
+
+	// 4. Copy message to kcat
+	err = kcat.CopyToContainer(ctx, []byte("Message produced by kcat"), "/tmp/msgs.txt", 700)
+	require.NoError(t, err)
+
+	// 5. Produce mesaage to Redpanda
+	_, _, err = kcat.Exec(ctx, []string{"kcat", "-b", "redpanda:29092", "-t", "msgs", "-P", "-l", "/tmp/msgs.txt"})
+
+	require.NoError(t, err)
+
+	// 6. Consume message from Redpanda
+	_, stdout, err := kcat.Exec(ctx, []string{"kcat", "-b", "redpanda:29092", "-C", "-t", "msgs", "-c", "1"})
+	require.NoError(t, err)
+
+	// 7. Read Message from stdou
+	out, err := io.ReadAll(stdout)
+	require.NoError(t, err)
+
+	require.Contains(t, string(out), "Message produced by kcat")
+
+	t.Cleanup(func() {
+		if err := kcat.Terminate(ctx); err != nil {
+			t.Fatalf("failed to terminate kcat container: %s", err)
+		}
+		if err := container.Terminate(ctx); err != nil {
+			t.Fatalf("failed to terminate redpanda container: %s", err)
+		}
+
+		if err := rpNetwork.Remove(ctx); err != nil {
+			t.Fatalf("failed to remove network: %s", err)
+		}
+	})
+}
+
+func TestRedpandaListener_InvalidPort(t *testing.T) {
+	ctx := context.Background()
+
+	// 1. Create network
+	RPNetwork, err := network.New(ctx, network.WithCheckDuplicate())
+	require.NoError(t, err)
+
+	// 2. Attemp Start Redpanda container
+	_, err = RunContainer(ctx,
+		testcontainers.WithImage("redpandadata/redpanda:v23.2.18"),
+		WithListener("redpanda:99092"),
+		network.WithNetwork([]string{"redpanda-host"}, RPNetwork),
+	)
+
+	require.Error(t, err)
+
+	require.Contains(t, err.Error(), "invalid port on listener redpanda:99092")
+
+	t.Cleanup(func() {
+		if err := RPNetwork.Remove(ctx); err != nil {
+			t.Fatalf("failed to remove network: %s", err)
+		}
+	})
+}
+
+func TestRedpandaListener_NoNetwork(t *testing.T) {
+	ctx := context.Background()
+
+	// 1. Attemp Start Redpanda container
+	_, err := RunContainer(ctx,
+		testcontainers.WithImage("redpandadata/redpanda:v23.2.18"),
+		WithListener("redpanda:99092"),
+	)
+
+	require.Error(t, err)
+
+	require.Contains(t, err.Error(), "container must be attached to at least one network")
+}
+
 // localhostCert is a PEM-encoded TLS cert with SAN IPs
 // generated from src/crypto/tls:
 // go run generate_cert.go  --rsa-bits 2048 --host 127.0.0.1,::1,localhost --ca --start-date "Jan 1 00:00:00 1970" --duration=1000000h