Enable using stdin when starting docker compose

compose.go

@@ -54,6 +54,7 @@ type LocalDockerCompose struct {
 	Services             map[string]interface{}
 	waitStrategySupplied bool
 	WaitStrategyMap      map[waitService]wait.Strategy
+	tmpFileName          string
 }
 
 type (
@@ -103,8 +104,16 @@ func NewLocalDockerCompose(filePaths []string, identifier string, opts ...LocalD
 
 	dc.absComposeFilePaths = make([]string, len(filePaths))
 	for i, cfp := range dc.ComposeFilePaths {
-		abs, _ := filepath.Abs(cfp)
-		dc.absComposeFilePaths[i] = abs
+		if cfp == "-" {
+			if err := dc.makeTmpFile(); err != nil {
+				continue
+			}
+			abs, _ := filepath.Abs(dc.tmpFileName)
+			dc.absComposeFilePaths[i] = abs
+		} else {
+			abs, _ := filepath.Abs(cfp)
+			dc.absComposeFilePaths[i] = abs
+		}
 	}
 
 	_ = dc.validate()
@@ -118,6 +127,11 @@ func NewLocalDockerCompose(filePaths []string, identifier string, opts ...LocalD
 
 // Down executes docker-compose down
 func (dc *LocalDockerCompose) Down() ExecError {
+	defer func() {
+		if dc.tmpFileName != "" {
+			_ = os.Remove(dc.tmpFileName)
+		}
+	}()
 	return executeCompose(dc, []string{"down", "--remove-orphans", "--volumes"})
 }
 
@@ -402,3 +416,29 @@ func which(binary string) error {
 
 	return err
 }
+
+func readSTDIN() ([]byte, error) {
+	return ioutil.ReadAll(os.Stdin)
+}
+
+func (dc *LocalDockerCompose) makeTmpFile() error {
+	data, err := readSTDIN()
+	if err != nil {
+		return err
+	}
+
+	file, err := ioutil.TempFile("", "__tmp_")
+	if err != nil {
+		return err
+	}
+
+	defer file.Close()
+
+	if _, err = file.Write(data); err != nil {
+		return err
+	}
+
+	dc.tmpFileName = file.Name()
+
+	return file.Sync()
+}

compose_test.go

@@ -2,6 +2,8 @@ package testcontainers
 
 import (
 	"fmt"
+	"io/ioutil"
+	"os"
 	"os/exec"
 	"strings"
 	"testing"
@@ -468,3 +470,51 @@ func executeAndGetOutput(command string, args []string) (string, ExecError) {
 		StdoutOutput: out,
 	}
 }
+
+func TestLocalDockerCompose_STDIN(t *testing.T) {
+	// mocks stdin
+	oldStdin := os.Stdin
+	data, err := ioutil.ReadFile("./testresources/docker-compose-simple.yml")
+	if err != nil {
+		t.Fatalf("Failed when reading ./testresources/docker-compose-simple.yml: %v", err)
+	}
+	file := mockStdin(t, data)
+	defer func() {
+		os.Stdin = oldStdin
+		file.Close()
+		os.Remove(file.Name())
+	}()
+	// --
+
+	path := "-"
+	identifier := strings.ToLower(uuid.New().String())
+
+	compose := NewLocalDockerCompose([]string{path}, identifier, WithLogger(TestLogger(t)))
+	defer func() {
+		checkIfError(t, compose.Down())
+	}()
+
+	exErr := compose.
+		WithCommand([]string{"up", "-d"}).
+		Invoke()
+	checkIfError(t, exErr)
+}
+
+func mockStdin(t *testing.T, content []byte) *os.File {
+	tmpfile, err := ioutil.TempFile("", "_test_tmp_")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if _, err := tmpfile.Write(content); err != nil {
+		t.Fatal(err)
+	}
+
+	if _, err := tmpfile.Seek(0, 0); err != nil {
+		t.Fatal(err)
+	}
+
+	os.Stdin = tmpfile
+
+	return tmpfile
+}