From 5b7ab5b2672fb7157d8f5d65d3f2a3ce6562f690 Mon Sep 17 00:00:00 2001
From: Chris Lee <chrissslee93@gmail.com>
Date: Sat, 23 Oct 2021 10:52:56 -0600
Subject: [PATCH] fix with unit-test

---
 docker.go           |   2 +-
 logconsumer_test.go | 101 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+), 1 deletion(-)

diff --git a/docker.go b/docker.go
index 0f869bf952..efc8439ad5 100644
--- a/docker.go
+++ b/docker.go
@@ -435,7 +435,7 @@ func (c *DockerContainer) StartLogProducer(ctx context.Context) error {
 				_, err := r.Read(h)
 				if err != nil {
 					// proper type matching requires https://go-review.googlesource.com/c/go/+/250357/ (go 1.16)
-					if strings.Contains(err.Error(), "use of closed connection") {
+					if strings.Contains(err.Error(), "use of closed network connection") {
 						now := time.Now()
 						since = fmt.Sprintf("%d.%09d", now.Unix(), int64(now.Nanosecond()))
 						goto BEGIN
diff --git a/logconsumer_test.go b/logconsumer_test.go
index b251349e40..854b449468 100644
--- a/logconsumer_test.go
+++ b/logconsumer_test.go
@@ -7,6 +7,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/docker/docker/client"
 	"github.com/testcontainers/testcontainers-go/wait"
 	"gotest.tools/assert"
 )
@@ -181,3 +182,103 @@ func Test_ShouldRecognizeLogTypes(t *testing.T) {
 	c.Terminate(ctx)
 
 }
+
+func TestContainerLogWithErrClosed(t *testing.T) {
+	// First spin up a docker-in-docker container, then spin up an inner container within that dind container
+	// Logs are being read from the inner container via the dind container's tcp port, which can be briefly
+	// closed to test behaviour in connection-closed situations.
+	ctx := context.Background()
+
+	dind, err := GenericContainer(ctx, GenericContainerRequest{
+		Started: true,
+		ContainerRequest: ContainerRequest{
+			Image:        "docker:dind",
+			ExposedPorts: []string{"2375/tcp"},
+			Env:          map[string]string{"DOCKER_TLS_CERTDIR": ""},
+			WaitingFor:   wait.ForListeningPort("2375/tcp"),
+			Privileged:   true,
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer dind.Terminate(ctx)
+
+	remoteDocker, err := dind.Endpoint(ctx, "tcp")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	client, err := client.NewClientWithOpts(client.WithHost(remoteDocker))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer client.Close()
+
+	client.NegotiateAPIVersion(ctx)
+	provider := &DockerProvider{client: client}
+
+	nginx, err := provider.CreateContainer(ctx, ContainerRequest{Image: "nginx", ExposedPorts: []string{"80/tcp"}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := nginx.Start(ctx); err != nil {
+		t.Fatal(err)
+	}
+	defer nginx.Terminate(ctx)
+
+	port, err := nginx.MappedPort(ctx, "80/tcp")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var consumer TestLogConsumer
+	if err = nginx.StartLogProducer(ctx); err != nil {
+		t.Fatal(err)
+	}
+	defer nginx.StopLogProducer()
+	nginx.FollowOutput(&consumer)
+
+	// Gather the initial container logs
+	time.Sleep(time.Second * 1)
+	existingLogs := len(consumer.Msgs)
+
+	hitNginx := func() {
+		i, err := dind.Exec(ctx, []string{"wget", "--spider", "localhost:" + port.Port()})
+		if err != nil || i > 0 {
+			t.Fatalf("Can't make request to nginx container from dind container")
+		}
+	}
+
+	hitNginx()
+	time.Sleep(time.Second * 1)
+	if len(consumer.Msgs)-existingLogs != 1 {
+		t.Fatalf("logConsumer should have 1 new log message, instead has: %v", consumer.Msgs[existingLogs:])
+	}
+	existingLogs = len(consumer.Msgs)
+
+	iptableArgs := []string{
+		"INPUT", "-p", "tcp", "--dport", "2375",
+		"-j", "REJECT", "--reject-with", "tcp-reset",
+	}
+	// Simulate a transient closed connection to the docker daemon
+	i, err := dind.Exec(ctx, append([]string{"iptables", "-A"}, iptableArgs...))
+	if err != nil || i > 0 {
+		t.Fatalf("Failed to close connection to dind daemon")
+	}
+	i, err = dind.Exec(ctx, append([]string{"iptables", "-D"}, iptableArgs...))
+	if err != nil || i > 0 {
+		t.Fatalf("Failed to re-open connection to dind daemon")
+	}
+	time.Sleep(time.Second * 3)
+
+	hitNginx()
+	hitNginx()
+	time.Sleep(time.Second * 1)
+	if len(consumer.Msgs)-existingLogs != 2 {
+		t.Fatalf(
+			"LogConsumer should have 2 new log messages after detecting closed connection and"+
+				" re-requesting logs. Instead has:\n%s", consumer.Msgs[existingLogs:],
+		)
+	}
+}