From e20372ffff19cc52485c9e1619516073e1659274 Mon Sep 17 00:00:00 2001
From: Joshua Powers <powersj@fastmail.com>
Date: Tue, 24 May 2022 09:29:15 -0600
Subject: [PATCH] test: migrate nats to test-containers (#11170)

---
 docker-compose.yml                |  4 ----
 plugins/outputs/nats/nats_test.go | 17 +++++++++++++++--
 2 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 571e0529c64fc..53e87d3f376f6 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -73,10 +73,6 @@ services:
     image: open62541/open62541
     ports:
       - "4840:4840"
-  nats:
-    image: nats
-    ports:
-      - "4222:4222"
   openldap:
     image: cobaugh/openldap-alpine
     environment:
diff --git a/plugins/outputs/nats/nats_test.go b/plugins/outputs/nats/nats_test.go
index 30004f6ae543d..db560d96e91b3 100644
--- a/plugins/outputs/nats/nats_test.go
+++ b/plugins/outputs/nats/nats_test.go
@@ -1,11 +1,13 @@
 package nats
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/influxdata/telegraf/plugins/serializers"
 	"github.com/influxdata/telegraf/testutil"
 	"github.com/stretchr/testify/require"
+	"github.com/testcontainers/testcontainers-go/wait"
 )
 
 func TestConnectAndWriteIntegration(t *testing.T) {
@@ -13,7 +15,18 @@ func TestConnectAndWriteIntegration(t *testing.T) {
 		t.Skip("Skipping integration test in short mode")
 	}
 
-	server := []string{"nats://" + testutil.GetLocalHost() + ":4222"}
+	container := testutil.Container{
+		Image:        "nats",
+		ExposedPorts: []string{"4222"},
+		WaitingFor:   wait.ForLog("Server is ready"),
+	}
+	err := container.Start()
+	require.NoError(t, err, "failed to start container")
+	defer func() {
+		require.NoError(t, container.Terminate(), "terminating container failed")
+	}()
+
+	server := []string{fmt.Sprintf("nats://%s:%s", container.Address, container.Port)}
 	s, _ := serializers.NewInfluxSerializer()
 	n := &NATS{
 		Servers:    server,
@@ -23,7 +36,7 @@ func TestConnectAndWriteIntegration(t *testing.T) {
 	}
 
 	// Verify that we can connect to the NATS daemon
-	err := n.Connect()
+	err = n.Connect()
 	require.NoError(t, err)
 
 	// Verify that we can successfully write data to the NATS daemon