Function TestRunContainerWithService

agent/pkg/docker/docker_api_test.go:197–292

(t *testing.T)

const testServiceImage = "nginx:latest"

func TestRunContainerWithService(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	agentID := uuid.NewString()
	containerID := uuid.NewString()

	t.Log("building client")
	rawCl, err := dclient.NewClientWithOpts(dclient.WithAPIVersionNegotiation(), dclient.FromEnv)
	require.NoError(t, err)
	defer func() {
		if cErr := rawCl.Close(); cErr != nil {
			t.Logf("closing docker client: %s", cErr)
		}
	}()
	cl := docker.NewClient(rawCl)

	t.Log("pull test image")
	evs := make(chan docker.Event, 1024)
	pub := events.ChannelPublisher(evs)
	if err = cl.PullImage(ctx, docker.PullImage{Name: testServiceImage}, pub); err != nil {
		t.Errorf("pulling image: %s", err.Error())
		return
	}
	close(evs)
	_, _, err = rawCl.ImageInspectWithRaw(ctx, testImage)
	require.NoError(t, err)

	t.Log("creating simple container")
	evs = make(chan docker.Event, 64)
	pub = events.ChannelPublisher(evs)
	dockerID, err := cl.CreateContainer(ctx, cproto.NewID(), cproto.RunSpec{
		ContainerConfig: container.Config{
			Image: testServiceImage,
			Labels: map[string]string{
				docker.AgentLabel:       agentID,
				docker.ContainerIDLabel: containerID,
			},
		},
	}, pub)
	require.NoError(t, err)

	t.Log("running simple container")
	c, err := cl.RunContainer(ctx, ctx, dockerID, pub)
	require.NoError(t, err)

	close(evs)
	defer func() {
		if rErr := rawCl.ContainerRemove(
			ctx,
			c.ContainerInfo.ID,
			container.RemoveOptions{Force: true},
		); rErr != nil {
			t.Errorf("failed to cleanup container %s: %s", c.ContainerInfo.ID, rErr)
		}
	}()

	t.Log("ensure it is listed when searching docker for our containers")
	containers, err := cl.ListRunningContainers(ctx, docker.LabelFilter(docker.AgentLabel, agentID))