testcontainers_test.go

package testcontainers

import (
	"os"
	"os/exec"
	"regexp"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestSessionID(t *testing.T) {
	t.Run("SessionID() returns a non-empty string", func(t *testing.T) {
		sessionID := SessionID()
		require.NotEmptyf(t, sessionID, "SessionID() returned an empty string")
	})

	t.Run("Multiple calls to SessionID() return the same value", func(t *testing.T) {
		sessionID1 := SessionID()
		sessionID2 := SessionID()
		require.Equalf(t, sessionID1, sessionID2, "SessionID() returned different values: %s != %s", sessionID1, sessionID2)
	})

	t.Run("Multiple calls to SessionID() in multiple goroutines return the same value", func(t *testing.T) {
		sessionID1 := ""
		sessionID2 := ""

		done := make(chan bool)
		go func() {
			sessionID1 = SessionID()
			done <- true
		}()

		go func() {
			sessionID2 = SessionID()
			done <- true
		}()

		<-done
		<-done

		require.Equalf(t, sessionID1, sessionID2, "SessionID() returned different values: %s != %s", sessionID1, sessionID2)
	})

	t.Run("SessionID() from different child processes returns the same value", func(t *testing.T) {
		args := []string{"test", "./...", "-v", "-run", "TestSessionIDHelper"}
		env := append(os.Environ(), "TESTCONTAINERS_SESSION_ID_HELPER=1")

		re := regexp.MustCompile(">>>(.*)<<<")

		cmd1 := exec.Command("go", args...)
		cmd1.Env = env
		stdoutStderr1, err := cmd1.CombinedOutput()
		require.NoErrorf(t, err, "cmd1.Run() failed with %s", err)
		sessionID1 := re.FindString(string(stdoutStderr1))

		cmd2 := exec.Command("go", args...)
		cmd2.Env = env
		stdoutStderr2, err := cmd2.CombinedOutput()
		require.NoErrorf(t, err, "cmd2.Run() failed with %s", err)
		sessionID2 := re.FindString(string(stdoutStderr2))

		require.Equalf(t, sessionID1, sessionID2, "SessionID() returned different values: %s != %s", sessionID1, sessionID2)
	})
}

// Not a real test, used to print out the session ID
func TestSessionIDHelper(t *testing.T) {
	if os.Getenv("TESTCONTAINERS_SESSION_ID_HELPER") == "" {
		t.Skip("Not a real test, used as a test helper")
	}

	t.Logf(">>>%s<<<\n", SessionID())
}