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

How to use WaitForService method of testcontainers Package

Best Testcontainers-go code snippet using testcontainers.WaitForService

compose_api_test.go

Source:compose_api_test.go

...27	ctx, cancel := context.WithCancel(context.Background())28	t.Cleanup(cancel)29	err = compose.30		// Appending with _1 as given in the Java Test-Containers Example31		WaitForService("mysql-1", wait.NewLogStrategy("started").WithStartupTimeout(10*time.Second).WithOccurrence(1)).32		Up(ctx, Wait(true))33	assert.Error(t, err, "Expected error to be thrown because service with wait strategy is not running")34	assert.Equal(t, err.Error(), "no container found for service name mysql-1")35	serviceNames := compose.Services()36	assert.Equal(t, 1, len(serviceNames))37	assert.Contains(t, serviceNames, "nginx")38}39func TestDockerComposeAPIWithWaitLogStrategy(t *testing.T) {40	compose, err := NewDockerCompose("./testresources/docker-compose-complex.yml")41	assert.NoError(t, err, "NewDockerCompose()")42	t.Cleanup(func() {43		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")44	})45	ctx, cancel := context.WithCancel(context.Background())46	t.Cleanup(cancel)47	err = compose.48		WaitForService("mysql", wait.NewLogStrategy("started").WithStartupTimeout(10*time.Second).WithOccurrence(1)).49		Up(ctx, Wait(true))50	assert.NoError(t, err, "compose.Up()")51	serviceNames := compose.Services()52	assert.Equal(t, 2, len(serviceNames))53	assert.Contains(t, serviceNames, "nginx")54	assert.Contains(t, serviceNames, "mysql")55}56func TestDockerComposeAPIWithRunServices(t *testing.T) {57	compose, err := NewDockerCompose("./testresources/docker-compose-complex.yml")58	assert.NoError(t, err, "NewDockerCompose()")59	t.Cleanup(func() {60		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")61	})62	ctx, cancel := context.WithCancel(context.Background())63	t.Cleanup(cancel)64	err = compose.65		WaitForService("nginx", wait.NewHTTPStrategy("/").WithPort("80/tcp").WithStartupTimeout(10*time.Second)).66		Up(ctx, Wait(true), RunServices("nginx"))67	assert.NoError(t, err, "compose.Up()")68	serviceNames := compose.Services()69	_, err = compose.ServiceContainer(context.Background(), "mysql")70	assert.Error(t, err, "Make sure there is no mysql container")71	assert.Equal(t, 1, len(serviceNames))72	assert.Contains(t, serviceNames, "nginx")73}74func TestDockerComposeAPIWithWaitForService(t *testing.T) {75	compose, err := NewDockerCompose("./testresources/docker-compose-simple.yml")76	assert.NoError(t, err, "NewDockerCompose()")77	t.Cleanup(func() {78		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")79	})80	ctx, cancel := context.WithCancel(context.Background())81	t.Cleanup(cancel)82	err = compose.83		WithEnv(map[string]string{84			"bar": "BAR",85		}).86		WaitForService("nginx", wait.NewHTTPStrategy("/").WithPort("80/tcp").WithStartupTimeout(10*time.Second)).87		Up(ctx, Wait(true))88	assert.NoError(t, err, "compose.Up()")89	serviceNames := compose.Services()90	assert.Equal(t, 1, len(serviceNames))91	assert.Contains(t, serviceNames, "nginx")92}93func TestDockerComposeAPIWithWaitHTTPStrategy(t *testing.T) {94	compose, err := NewDockerCompose("./testresources/docker-compose-simple.yml")95	assert.NoError(t, err, "NewDockerCompose()")96	t.Cleanup(func() {97		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")98	})99	ctx, cancel := context.WithCancel(context.Background())100	t.Cleanup(cancel)101	err = compose.102		WithEnv(map[string]string{103			"bar": "BAR",104		}).105		WaitForService("nginx", wait.NewHTTPStrategy("/").WithPort("80/tcp").WithStartupTimeout(10*time.Second)).106		Up(ctx, Wait(true))107	assert.NoError(t, err, "compose.Up()")108	serviceNames := compose.Services()109	assert.Equal(t, 1, len(serviceNames))110	assert.Contains(t, serviceNames, "nginx")111}112func TestDockerComposeAPIWithContainerName(t *testing.T) {113	compose, err := NewDockerCompose("./testresources/docker-compose-container-name.yml")114	assert.NoError(t, err, "NewDockerCompose()")115	t.Cleanup(func() {116		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")117	})118	ctx, cancel := context.WithCancel(context.Background())119	t.Cleanup(cancel)120	err = compose.121		WithEnv(map[string]string{122			"bar": "BAR",123		}).124		WaitForService("nginx", wait.NewHTTPStrategy("/").WithPort("80/tcp").WithStartupTimeout(10*time.Second)).125		Up(ctx, Wait(true))126	assert.NoError(t, err, "compose.Up()")127	serviceNames := compose.Services()128	assert.Equal(t, 1, len(serviceNames))129	assert.Contains(t, serviceNames, "nginx")130}131func TestDockerComposeAPIWithWaitStrategy_NoExposedPorts(t *testing.T) {132	compose, err := NewDockerCompose("./testresources/docker-compose-no-exposed-ports.yml")133	assert.NoError(t, err, "NewDockerCompose()")134	t.Cleanup(func() {135		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")136	})137	ctx, cancel := context.WithCancel(context.Background())138	t.Cleanup(cancel)139	err = compose.140		WaitForService("nginx", wait.ForLog("Configuration complete; ready for start up")).141		Up(ctx, Wait(true))142	assert.NoError(t, err, "compose.Up()")143	serviceNames := compose.Services()144	assert.Equal(t, 1, len(serviceNames))145	assert.Contains(t, serviceNames, "nginx")146}147func TestDockerComposeAPIWithMultipleWaitStrategies(t *testing.T) {148	compose, err := NewDockerCompose("./testresources/docker-compose-complex.yml")149	assert.NoError(t, err, "NewDockerCompose()")150	t.Cleanup(func() {151		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")152	})153	ctx, cancel := context.WithCancel(context.Background())154	t.Cleanup(cancel)155	err = compose.156		WaitForService("mysql", wait.NewLogStrategy("started").WithStartupTimeout(10*time.Second)).157		WaitForService("nginx", wait.NewHTTPStrategy("/").WithPort("80/tcp").WithStartupTimeout(10*time.Second)).158		Up(ctx, Wait(true))159	assert.NoError(t, err, "compose.Up()")160	serviceNames := compose.Services()161	assert.Equal(t, 2, len(serviceNames))162	assert.Contains(t, serviceNames, "nginx")163	assert.Contains(t, serviceNames, "mysql")164}165func TestDockerComposeAPIWithFailedStrategy(t *testing.T) {166	compose, err := NewDockerCompose("./testresources/docker-compose-simple.yml")167	assert.NoError(t, err, "NewDockerCompose()")168	t.Cleanup(func() {169		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")170	})171	ctx, cancel := context.WithCancel(context.Background())172	t.Cleanup(cancel)173	err = compose.174		WithEnv(map[string]string{175			"bar": "BAR",176		}).177		WaitForService("nginx_1", wait.NewHTTPStrategy("/").WithPort("8080/tcp").WithStartupTimeout(5*time.Second)).178		Up(ctx, Wait(true))179	// Verify that an error is thrown and not nil180	// A specific error message matcher is not asserted since the docker library can change the return message, breaking this test181	assert.Error(t, err, "Expected error to be thrown because of a wrong suplied wait strategy")182	serviceNames := compose.Services()183	assert.Equal(t, 1, len(serviceNames))184	assert.Contains(t, serviceNames, "nginx")185}186func TestDockerComposeAPIComplex(t *testing.T) {187	compose, err := NewDockerCompose("./testresources/docker-compose-complex.yml")188	assert.NoError(t, err, "NewDockerCompose()")189	t.Cleanup(func() {190		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")191	})192	ctx, cancel := context.WithCancel(context.Background())193	t.Cleanup(cancel)194	assert.NoError(t, compose.Up(ctx, Wait(true)), "compose.Up()")195	serviceNames := compose.Services()196	assert.Equal(t, 2, len(serviceNames))197	assert.Contains(t, serviceNames, "nginx")198	assert.Contains(t, serviceNames, "mysql")199}200func TestDockerComposeAPIWithEnvironment(t *testing.T) {201	identifier := testNameHash(t.Name())202	compose, err := NewDockerComposeWith(WithStackFiles("./testresources/docker-compose-simple.yml"), identifier)203	assert.NoError(t, err, "NewDockerCompose()")204	t.Cleanup(func() {205		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")206	})207	ctx, cancel := context.WithCancel(context.Background())208	t.Cleanup(cancel)209	err = compose.210		WithEnv(map[string]string{211			"bar": "BAR",212		}).213		Up(ctx, Wait(true))214	assert.NoError(t, err, "compose.Up()")215	serviceNames := compose.Services()216	assert.Equal(t, 1, len(serviceNames))217	assert.Contains(t, serviceNames, "nginx")218	present := map[string]string{219		"bar": "BAR",220	}221	absent := map[string]string{}222	assertContainerEnvironmentVariables(t, identifier.String(), "nginx", present, absent)223}224func TestDockerComposeAPIWithMultipleComposeFiles(t *testing.T) {225	composeFiles := ComposeStackFiles{226		"testresources/docker-compose-simple.yml",227		"testresources/docker-compose-postgres.yml",228		"testresources/docker-compose-override.yml",229	}230	identifier := testNameHash(t.Name())231	compose, err := NewDockerComposeWith(composeFiles, identifier)232	assert.NoError(t, err, "NewDockerCompose()")233	t.Cleanup(func() {234		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")235	})236	ctx, cancel := context.WithCancel(context.Background())237	t.Cleanup(cancel)238	err = compose.239		WithEnv(map[string]string{240			"bar": "BAR",241			"foo": "FOO",242		}).243		Up(ctx, Wait(true))244	assert.NoError(t, err, "compose.Up()")245	serviceNames := compose.Services()246	assert.Equal(t, 3, len(serviceNames))247	assert.Contains(t, serviceNames, "nginx")248	assert.Contains(t, serviceNames, "mysql")249	assert.Contains(t, serviceNames, "postgres")250	present := map[string]string{251		"bar": "BAR",252		"foo": "FOO",253	}254	absent := map[string]string{}255	assertContainerEnvironmentVariables(t, identifier.String(), "nginx", present, absent)256}257func TestDockerComposeAPIWithVolume(t *testing.T) {258	compose, err := NewDockerCompose("./testresources/docker-compose-volume.yml")259	assert.NoError(t, err, "NewDockerCompose()")260	t.Cleanup(func() {261		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")262	})263	ctx, cancel := context.WithCancel(context.Background())264	t.Cleanup(cancel)265	err = compose.Up(ctx, Wait(true))266	assert.NoError(t, err, "compose.Up()")267}268func TestDockerComposeAPIWithBuild(t *testing.T) {269	compose, err := NewDockerCompose("./testresources/docker-compose-build.yml")270	assert.NoError(t, err, "NewDockerCompose()")271	t.Cleanup(func() {272		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")273	})274	ctx, cancel := context.WithCancel(context.Background())275	t.Cleanup(cancel)276	err = compose.277		WaitForService("echo", wait.ForHTTP("/env").WithPort("8080/tcp")).278		Up(ctx, Wait(true))279	assert.NoError(t, err, "compose.Up()")280}281func TestDockerComposeApiWithWaitForShortLifespanService(t *testing.T) {282	compose, err := NewDockerCompose("./testresources/docker-compose-short-lifespan.yml")283	assert.NoError(t, err, "NewDockerCompose()")284	t.Cleanup(func() {285		assert.NoError(t, compose.Down(context.Background(), RemoveOrphans(true), RemoveImagesLocal), "compose.Down()")286	})287	ctx, cancel := context.WithCancel(context.Background())288	t.Cleanup(cancel)289	err = compose.290		//Assumption: tzatziki service wait logic will run before falafel, so that falafel service will exit before291		WaitForService("tzatziki", wait.ForExit().WithExitTimeout(10*time.Second)).292		WaitForService("falafel", wait.ForExit().WithExitTimeout(10*time.Second)).293		Up(ctx)294	assert.NoError(t, err, "compose.Up()")295	services := compose.Services()296	assert.Equal(t, 2, len(services))297	assert.Contains(t, services, "falafel")298	assert.Contains(t, services, "tzatziki")299}300func testNameHash(name string) StackIdentifier {301	return StackIdentifier(fmt.Sprintf("%x", fnv.New32a().Sum([]byte(name))))302}...

environment.go

Source:environment.go

...49	composeFilePaths []string,50	identifier string,51) {52	c.compose = testcontainers.NewLocalDockerCompose(composeFilePaths, identifier).53		WaitForService(PostgresContainerName, wait.ForLog("database system is ready to accept connections")).54		WaitForService(ZooKeeperContainerName, wait.ForLog("binding to port 0.0.0.0/0.0.0.0:"+ZooKeeperPort)).55		WaitForService(KafkaContainerName, wait.ForLog("[KafkaServer id=1] started"))56	if len(composeFilePaths) > 1 { // this is little tricky hack here. :)57		// if we have one docker-compose file for app container, that add wait strategy.58		c.compose = c.compose.WaitForService(AppName, wait.ForLog("App starting successfully! Ready for hard work!"))59	}60	require.Nil(t, c.compose.WithCommand([]string{"up", "--force-recreate", "-d"}).Invoke().Error)61}62// FinishedDockerComposeEnvironment - finished containers (env) which we created by second way.63func (c *ContainersEnvironment) FinishedDockerComposeEnvironment(t *testing.T) {64	require.Nil(t, c.compose.Down().Error, "docker compose must down without errors")65}...

cnc_test.go

Source:cnc_test.go

...17func (i *IntegrationTestSuite) SetupSuite() {18	composeFilePaths := []string{"docker/test-docker-compose.yml"}19	identifier := strings.ToLower(uuid.New().String())20	i.dockerCompose = testcontainers.NewLocalDockerCompose(composeFilePaths, identifier)21	i.dockerCompose.WaitForService("bridge0",22		wait.ForHTTP("/balance").WithPort("26658").23			WithStartupTimeout(60*time.Second).24			WithPollInterval(3*time.Second))25	execError := i.dockerCompose.WithCommand([]string{"up", "-d"}).Invoke()26	err := execError.Error27	if err != nil {28		i.Fail("failed to execute docker compose up:", "err", err.Error(), "stdout", execError.Stdout, "stderr", execError.Stderr)29	}30}31func (i *IntegrationTestSuite) TearDownSuite() {32	execError := i.dockerCompose.Down()33	if err := execError.Error; err != nil {34		i.Fail("failed to execute docker compose down: %v\n", err, execError.Stdout, execError.Stderr)35	}...

WaitForService

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, _ := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	redisHost, _ := redisContainer.Host(ctx)11	redisPort, _ := redisContainer.MappedPort(ctx, "6379/tcp")12	fmt.Println(redisHost)13	fmt.Println(redisPort.Int())14}

WaitForService

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	req := testcontainers.ContainerRequest{5		Env: map[string]string{6		},7		ExposedPorts: []string{"5432/tcp"},8		WaitingFor:   wait.ForListeningPort("5432/tcp"),9	}10	postgresContainer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{11	})12	if err != nil {13		log.Fatal(err)14	}15	ip, err := postgresContainer.Host(ctx)16	if err != nil {17		log.Fatal(err)18	}19	port, err := postgresContainer.MappedPort(ctx, "5432/tcp")20	if err != nil {21		log.Fatal(err)22	}23	fmt.Printf("Postgres is available on %s:%s24", ip, port.Port())25	postgresContainer.Terminate(ctx)26}

WaitForService

Using AI Code Generation

1import (2func main() {3	req := testcontainers.ContainerRequest{4		ExposedPorts: []string{"5432/tcp"},5		WaitingFor:   wait.ForLog("database system is ready to accept connections").WithStartupTimeout(60 * time.Second),6	}7	ctx := context.Background()8	postgres, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	if err != nil {11		log.Fatal(err)12	}13	defer postgres.Terminate(ctx)14	ip, err := postgres.Host(ctx)15	if err != nil {16		log.Fatal(err)17	}18	mappedPort, err := postgres.MappedPort(ctx, "5432")19	if err != nil {20		log.Fatal(err)21	}22	fmt.Println(ip, mappedPort.Int())23}

WaitForService

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.Fatal(err)12	}13	defer redisContainer.Terminate(ctx)14	redisHost, err := redisContainer.Host(ctx)15	if err != nil {16		log.Fatal(err)17	}18	redisPort, err := redisContainer.MappedPort(ctx, "6379/tcp")19	if err != nil {20		log.Fatal(err)21	}22	fmt.Println(redisURL)23}24import (25func main() {26	ctx := context.Background()27	req := testcontainers.ContainerRequest{28		ExposedPorts: []string{"6379/tcp"},29		WaitingFor:   wait.ForLog("Ready to accept connections"),30	}31	redisContainer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{32	})33	if err != nil {34		log.Fatal(err)35	}36	defer redisContainer.Terminate(ctx)37	redisHost, err := redisContainer.Host(ctx)38	if err != nil {39		log.Fatal(err)40	}41	redisPort, err := redisContainer.MappedPort(ctx, "6379/tcp")42	if err != nil {43		log.Fatal(err)44	}45	fmt.Println(redisURL)46}

WaitForService

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.Fatal(err)12	}13	defer redisContainer.Terminate(ctx)14	ip, err := redisContainer.Host(ctx)15	if err != nil {16		log.Fatal(err)17	}18	port, err := redisContainer.MappedPort(ctx, "6379/tcp")19	if err != nil {20		log.Fatal(err)21	}22	fmt.Println(ip, port.Int())23}

WaitForService

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	redis, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	if err != nil {11		log.Fatal(err)12	}13	ip, err := redis.Host(ctx)14	if err != nil {15		log.Fatal(err)16	}17	port, err := redis.MappedPort(ctx, "6379/tcp")18	if err != nil {19		log.Fatal(err)20	}21	fmt.Printf("Redis is available at %s:%s22", ip, port.Port())23	if err := redis.Terminate(ctx); err != nil {24		log.Fatal(err)25	}26}27import (28func main() {29	ctx := context.Background()30	req := testcontainers.ContainerRequest{31		ExposedPorts: []string{"6379/tcp"},32		WaitingFor:   wait.ForLog("Ready to accept connections"),33	}34	redis, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{35	})36	if err != nil {37		log.Fatal(err)38	}39	ip, err := redis.Host(ctx)40	if err != nil {41		log.Fatal(err)42	}

WaitForService

Using AI Code Generation

1import (2func main() {3	req := testcontainers.ContainerRequest{4		ExposedPorts: []string{"27017/tcp"},5		WaitingFor:   wait.ForListeningPort("27017/tcp"),6	}7	ctx := context.Background()8	mongo, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	if err != nil {11		log.Fatal(err)12	}13	defer mongo.Terminate(ctx)14	ip, err := mongo.Host(ctx)15	if err != nil {16		log.Fatal(err)17	}18	port, err := mongo.MappedPort(ctx, "27017/tcp")19	if err != nil {20		log.Fatal(err)21	}22	fmt.Printf("Container IP: %s, Container Port: %s23", ip, port.Port())

WaitForService

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	redis, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{9	})10	if err != nil {11		panic(err)12	}13	ip, err := redis.Host(ctx)14	if err != nil {15		panic(err)16	}17	port, err := redis.MappedPort(ctx, "6379/tcp")18	if err != nil {19		panic(err)20	}21	fmt.Printf("Redis is available at %s on port %s22", ip, port.Port())23	redis.Terminate(ctx)24}

WaitForService

Using AI Code Generation

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

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.