KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use FollowOutput method of testcontainers Package

Best Testcontainers-go code snippet using testcontainers.FollowOutput

test_utils.go

Source:test_utils.go

...79	if err != nil {80		return nil, err81	}82	container.StartLogProducer(ctx)83	container.FollowOutput(&TestLogConsumer{})84	return container, err85}86// RunJMXServiceContainerSSL will start a container running test-server configured with SSL JMX.87func RunJMXServiceContainerSSL(ctx context.Context) (testcontainers.Container, error) {88	var hostnameOpt string89	if !isRunningInDockerContainer() {90		hostnameOpt = "-Djava.rmi.server.hostname=0.0.0.0"91	}92	req := testcontainers.ContainerRequest{93		Image: "test-server:latest",94		ExposedPorts: []string{95			fmt.Sprintf("%[1]s:%[1]s", TestServerPort),96			fmt.Sprintf("%[1]s:%[1]s", TestServerJMXPort),97		},98		Env: map[string]string{99			"JAVA_OPTS": "-Dcom.sun.management.jmxremote.port=" + TestServerJMXPort + " " +100				"-Dcom.sun.management.jmxremote.authenticate=true " +101				"-Dcom.sun.management.jmxremote.ssl=true " +102				"-Dcom.sun.management.jmxremote.ssl.need.client.auth=true " +103				"-Dcom.sun.management.jmxremote.registry.ssl=true " +104				"-Dcom.sun.management.jmxremote=true " +105				"-Dcom.sun.management.jmxremote.rmi.port=" + TestServerJMXPort + " " +106				"-Dcom.sun.management.jmxremote.local.only=false " +107				"-Djavax.net.ssl.keyStore=/keystore  " +108				"-Djavax.net.ssl.keyStorePassword=password " +109				"-Djavax.net.ssl.trustStore=/truststore " +110				"-Djavax.net.ssl.trustStorePassword=password " +111				hostnameOpt,112		},113		WaitingFor: wait.ForListeningPort(TestServerPort),114	}115	container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{116		ContainerRequest: req,117		Started:          true,118	})119	if err != nil {120		return nil, err121	}122	container.StartLogProducer(ctx)123	container.FollowOutput(&TestLogConsumer{})124	return container, err125}126// GetContainerServiceURL will return the url to the test-server running inside the container.127func GetContainerServiceURL(ctx context.Context, container testcontainers.Container, port nat.Port, endpoint string) (string, error) {128	mappedPort, err := container.MappedPort(ctx, port)129	if err != nil {130		return "", err131	}132	var hostIP string133	if isRunningInDockerContainer() {134		if hostIP, err = container.ContainerIP(ctx); err != nil {135			return "", err136		}137	} else {138		if hostIP, err = container.Host(ctx); err != nil {139			return "", err140		}141	}142	return fmt.Sprintf("http://%s:%s%s", hostIP, mappedPort.Port(), endpoint), nil143}144// CleanMBeans will remove all new added MBeans from test-server.145func CleanMBeans(ctx context.Context, container testcontainers.Container) ([]byte, error) {146	url, err := GetContainerServiceURL(ctx, container, TestServerPort, TestServerCleanDataEndpoint)147	if err != nil {148		return nil, err149	}150	return DoHttpRequest(http.MethodPut, url, nil)151}152// AddMBeansBatch will add new MBeans to the test-server.153func AddMBeansBatch(ctx context.Context, container testcontainers.Container, body []map[string]interface{}) ([]byte, error) {154	return addMBeans(ctx, container, body, TestServerAddDataBatchEndpoint)155}156// AddMBeans will add new MBeans to the test-server.157func AddMBeans(ctx context.Context, container testcontainers.Container, body map[string]interface{}) ([]byte, error) {158	return addMBeans(ctx, container, body, TestServerAddDataEndpoint)159}160// AddMBeans will add new MBeans to the test-server.161func AddMCompositeDataBeans(ctx context.Context, container testcontainers.Container, body map[string]interface{}) ([]byte, error) {162	return addMBeans(ctx, container, body, TestServerAddCompositeDataEndpoint)163}164// addMBeans will add new MBeans to the test-server.165func addMBeans(ctx context.Context, container testcontainers.Container, body interface{}, endpointPath string) ([]byte, error) {166	url, err := GetContainerServiceURL(ctx, container, TestServerPort, endpointPath)167	if err != nil {168		return nil, err169	}170	json, err := json.Marshal(body)171	if err != nil {172		return nil, err173	}174	return DoHttpRequest(http.MethodPost, url, json)175}176// TestLogConsumer is used to print container logs to stdout.177type TestLogConsumer struct {178}179func (g *TestLogConsumer) Accept(l testcontainers.Log) {180	fmt.Fprintf(os.Stdout, "[CONTAINER LOG] %s %s\n", time.Now().Format("2006/01/02 15:04:05"), l.Content)181}182// RunJbossStandaloneJMXContainer will start a container running a jboss instace with JMX.183func RunJbossStandaloneJMXContainer(ctx context.Context) (testcontainers.Container, error) {184	req := testcontainers.ContainerRequest{185		Image: "test_jboss",186		ExposedPorts: []string{187			fmt.Sprintf("%[1]s:%[1]s", JbossJMXPort),188		},189		WaitingFor: wait.ForListeningPort(JbossJMXPort),190	}191	container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{192		ContainerRequest: req,193		Started:          true,194	})195	if err != nil {196		return nil, err197	}198	container.StartLogProducer(ctx)199	container.FollowOutput(&TestLogConsumer{})200	return container, err201}202// CopyFileFromContainer will copy a file from a given docker container.203func CopyFileFromContainer(ctx context.Context, container testcontainers.Container, srcPath, dstPath string) error {204	reader, err := container.CopyFileFromContainer(ctx, srcPath)205	if err != nil {206		return err207	}208	defer reader.Close()209	b, err := ioutil.ReadAll(reader)210	if err != nil {211		return err212	}213	return ioutil.WriteFile(dstPath, b, 0644)...

container.go

Source:container.go

...52	c.container = container53	c.Logs = TestLogConsumer{54		Msgs: []string{},55	}56	c.container.FollowOutput(&c.Logs)57	err = c.container.StartLogProducer(c.ctx)58	if err != nil {59		return fmt.Errorf("log producer failed: %s", err)60	}61	c.Address = "localhost"62	err = c.LookupMappedPorts()63	if err != nil {64		_ = c.Terminate()65		return fmt.Errorf("port lookup failed: %s", err)66	}67	return nil68}69// create a lookup table of exposed ports to mapped ports70func (c *Container) LookupMappedPorts() error {...

log.go

Source:log.go

...34		LogChan:   make(chan testcontainers.Log, 10),35		container: c,36	}37	// reversed to avoid "race" since `StartLogProducer` starts a goroutine38	c.FollowOutput(&logger)39	err := c.StartLogProducer(ctx)40	return logger, err41}...

FollowOutput

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		Cmd:          []string{"sh", "-c", "while true; do echo Hello world; sleep 1; done"},6		ExposedPorts: []string{"80/tcp"},7		WaitingFor:   wait.ForLog("Hello world"),8	}9	c, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{10	})11	if err != nil {12		log.Fatal(err)13	}14	defer c.Terminate(ctx)15	ip, err := c.Host(ctx)16	if err != nil {17		log.Fatal(err)18	}19	port, err := c.MappedPort(ctx, "80")20	if err != nil {21		log.Fatal(err)22	}23	fmt.Printf("Container is listening on %s:%s24", ip, port.Port())25	output, err := c.FollowOutput(ctx)26	if err != nil {27		log.Fatal(err)28	}29	go func() {30		for {31			line, err := output.Next()32			if err != nil {33				log.Fatal(err)34			}35			fmt.Printf("%s36		}37	}()38	fmt.Printf("Press enter to stop container")39	_, err = os.Read(0)40	if err != nil {41		log.Fatal(err)42	}43}

FollowOutput

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		ExposedPorts: []string{"80/tcp"},6		WaitingFor:   wait.ForLog("Starting nginx").WithStartupTimeout(time.Second * 5),7	}8	nginxContainer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	if err != nil {11		log.Fatalf("Could not start container: %v", err)12	}13	ip, err := nginxContainer.Host(ctx)14	if err != nil {15		log.Fatalf("Could not get container IP: %v", err)16	}17	port, err := nginxContainer.MappedPort(ctx, "80")18	if err != nil {19		log.Fatalf("Could not get container port: %v", err)20	}21	fmt.Printf("Container IP: %s, Port: %s", ip, port.Port())22	cmd := exec.Command("curl", fmt.Sprintf("%s:%s", ip, port.Port()))23	stdout, err := cmd.StdoutPipe()24	if err != nil {25		log.Fatalf("Could not get stdout pipe: %v", err)26	}27	if err := cmd.Start(); err != nil {28		log.Fatalf("Could not start command: %v", err)29	}30	if _, err := io.Copy(os.Stdout, stdout); err != nil {31		log.Fatalf("Could not copy output to stdout: %v", err)32	}33	if err := cmd.Wait(); err != nil {34		log.Fatalf("Command failed: %v", err)35	}36	if err := nginxContainer.Terminate(ctx); err != nil {37		log.Fatalf("Could not stop container: %v", err)38	}39}

FollowOutput

Using AI Code Generation

1import (2func main() {3	dir, err := filepath.Abs(filepath.Dir(os.Args[0]))4	if err != nil {5		log.Fatal(err)6	}7	dockerfile := filepath.Join(dir, "Dockerfile")8	ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)9	defer cancel()10	req := testcontainers.ContainerRequest{11		ExposedPorts: []string{"8080/tcp"},12		Cmd:          []string{"-p", "8080:8080"},13		WaitingFor:   wait.ForListeningPort("8080/tcp"),14		BindMounts: map[string]string{15		},16	}17	ryuk, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{18	})19	if err != nil {20		log.Fatal(err)21	}22	defer ryuk.Terminate(ctx)23	dockerfile = filepath.Join(dir, "Dockerfile")24	ctx, cancel = context.WithTimeout(context.Background(), 120*time.Second)25	defer cancel()26	req = testcontainers.ContainerRequest{27		ExposedPorts: []string{"8080/tcp"},28		Cmd:          []string{"-p", "8080:8080"},29		WaitingFor:   wait.ForListeningPort("8080/tcp"),30		BindMounts: map[string]string{31		},32	}33	goContainer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{34	})35	if err != nil {36		log.Fatal(err)37	}

FollowOutput

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		Cmd:          []string{"sh", "-c", "while true; do echo hello world; sleep 1; done"},6		ExposedPorts: []string{"80/tcp"},7		WaitingFor:   wait.ForLog("hello world"),8	}9	c, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{10	})11	if err != nil {12		log.Fatal(err)13	}14	defer c.Terminate(ctx)15	ip, err := c.Host(ctx)16	if err != nil {17		log.Fatal(err)18	}19	port, err := c.MappedPort(ctx, "80")20	if err != nil {21		log.Fatal(err)22	}23	fmt.Printf("Container created. IP: %s, Port: %s", ip, port.Port())24	logs, err := c.Logs(ctx)25	if err != nil {26		log.Fatal(err)27	}28	fmt.Println("Printing the logs")29	io.Copy(os.Stdout, logs)30	fmt.Println("Printing the logs in real time")31	go func() {32		time.Sleep(time.Second * 5)33		c.Terminate(ctx)34	}()35	err = c.FollowOutput(ctx, os.Stdout, func(s string) bool {36		fmt.Println("Checking if the container has stopped")37		return strings.Contains(s, "hello world")38	})39	if err != nil {40		log.Fatal(err)41	}42}

FollowOutput

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		Cmd:          []string{"sh", "-c", "while true; do echo hello world; sleep 1; done"},6		ExposedPorts: []string{"80/tcp"},7		WaitingFor:   wait.ForLog("hello world").WithPollInterval(1 * time.Second),8	}

FollowOutput

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		Cmd:          []string{"sh", "-c", "while true; do echo hello; sleep 1; done"},6		ExposedPorts: []string{"80/tcp"},7		WaitingFor:   wait.ForLog("hello"),8	}9	c, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{10	})11	if err != nil {12		panic(err)13	}14	ip, err := c.Host(ctx)15	if err != nil {16		panic(err)17	}18	port, err := c.MappedPort(ctx, "80")19	if err != nil {20		panic(err)21	}22	fmt.Printf("Container IP: %s, mapped port: %s23", ip, port.Port())24	out, err := c.FollowOutput(ctx)25	if err != nil {26		panic(err)27	}28	file, err := os.Create("out.txt")29	if err != nil {30		panic(err)31	}32	defer file.Close()33	_, err = io.Copy(file, out)34	if err != nil {35		panic(err)36	}37	time.Sleep(5 * time.Second)38	err = c.Terminate(ctx)39	if err != nil {40		panic(err)41	}42}

FollowOutput

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		ExposedPorts: []string{"6379/tcp"},6		WaitingFor:   wait.ForListeningPort("6379/tcp"),7	}8	redisContainer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	if err != nil {11		log.Fatalf("Could not create container: %v", err)12	}13	host, err := redisContainer.Host(ctx)14	if err != nil {15		log.Fatalf("Could not get host: %v", err)16	}17	port, err := redisContainer.MappedPort(ctx, "6379/tcp")18	if err != nil {19		log.Fatalf("Could not get port: %v", err)20	}21	fmt.Printf("Host: %s, Port: %s", host, port.Port())22	redisContainer.Start(ctx)23	defer redisContainer.Terminate(ctx)24	redisContainer.FollowOutput(ctx, os.Stdout)25}

FollowOutput

Using AI Code Generation

1import (2func main() {3    ctx := context.Background()4    req := testcontainers.ContainerRequest{5        Cmd:          []string{"echo", "hello world"},6        WaitingFor:   wait.ForLog("hello world"),7    }8    container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9    })10    if err != nil {11        log.Fatal(err)12    }13    defer container.Terminate(ctx)14    out, err := container.FollowOutput(ctx, testcontainers.NewLogConsumer())15    if err != nil {16        log.Fatal(err)17    }18    for {19        select {20            fmt.Println(line)21        case <-time.After(10 * time.Second):22            os.Exit(0)23        }24    }25}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.