package dockerutils import ( "context" "fmt" "net/http" "strings" "testing" "github.com/docker/docker/api/types" "github.com/docker/docker/api/types/volume" uuid "github.com/satori/go.uuid" "github.com/stretchr/testify/require" ) // TestContainerCRUD // Test the following API functions: // - ContainerListByLabels // - ContainerListByNames // - ContainerRemoveByNames // - ContainerRemoveByLabels // - ContainerRemoveByIDs func TestContainerCRUD(t *testing.T) { var ( ctx = context.Background() require = require.New(t) iterations = 5 cleanupLabels = map[string]string{ uuid.NewV4().String(): uuid.NewV4().String(), } ) dockerClient, err := New() require.NoError(err) t.Cleanup(func() { dockerClient.ContainerRemoveByLabels(ctx, cleanupLabels) }) // Create Containers containerIDs := make([]string, 0) containerNames := make([]string, 0) for i := 0; i < iterations; i++ { containerName := uuid.NewV4().String() containerID, err := dockerClient.NewBuilder("nginx:alpine"). Labels(cleanupLabels). Port("80"). Pull(). WithName(containerName). Start(ctx) require.NoError(err) containerNames = append(containerNames, containerName) containerIDs = append(containerIDs, containerID) } // ListByLabels containers, err := dockerClient.ContainerListByLabels(ctx, true, cleanupLabels) require.NoError(err) require.Len(containers, iterations) for _, container := range containers { require.Contains(containerIDs, container.ID) require.Contains(containerNames, strings.Split(container.Names[0], "/")[1]) } // ListByNames containers, err = dockerClient.ContainerListByNames(ctx, true, containerNames...) require.NoError(err) require.Len(containers, iterations) for _, container := range containers { require.Contains(containerIDs, container.ID) require.Contains(containerNames, strings.Split(container.Names[0], "/")[1]) } // RemoveByLabels err = dockerClient.ContainerRemoveByLabels(ctx, cleanupLabels) require.NoError(err) containers, err = dockerClient.ContainerListByLabels(ctx, true, cleanupLabels) require.NoError(err) require.Len(containers, 0) // Create containerIDs = make([]string, 0) containerNames = make([]string, 0) for i := 0; i < iterations; i++ { containerName := uuid.NewV4().String() containerID, err := dockerClient.NewBuilder("nginx:alpine"). Labels(cleanupLabels). Port("80"). Pull(). WithName(containerName). Start(ctx) require.NoError(err) containerNames = append(containerNames, containerName) containerIDs = append(containerIDs, containerID) } // RemoveByNames err = dockerClient.ContainerRemoveByNames(ctx, containerNames...) require.NoError(err) containers, err = dockerClient.ContainerListByNames(ctx, true, containerNames...) require.NoError(err) require.Len(containers, 0) // Create containerIDs = make([]string, 0) containerNames = make([]string, 0) for i := 0; i < iterations; i++ { containerName := uuid.NewV4().String() containerID, err := dockerClient.NewBuilder("nginx:alpine"). Labels(cleanupLabels). Port("80"). Pull(). WithName(containerName). Start(ctx) require.NoError(err) containerNames = append(containerNames, containerName) containerIDs = append(containerIDs, containerID) } // RemoveByID err = dockerClient.ContainerRemoveByIDs(ctx, containerIDs...) require.NoError(err) containers, err = dockerClient.ContainerListByLabels(ctx, true, cleanupLabels) require.NoError(err) require.Len(containers, 0) } // TestNetworkCRUD // Test the following API functions: // - NetworkListByLabels // - NetworkListByNames // - NetworkRemoveByLabels // - NetworkRemoveByNames // - NetworkRemoveByIDs func TestNetworkCRUD(t *testing.T) { var ( ctx = context.Background() require = require.New(t) iterations = 5 cleanupLabels = map[string]string{ uuid.NewV4().String(): uuid.NewV4().String(), } ) dockerClient, err := New() require.NoError(err) t.Cleanup(func() { dockerClient.NetworkRemoveByLabels(ctx, cleanupLabels) }) // Create Networks networkIDs := make([]string, 0) networkNames := make([]string, 0) for i := 0; i < iterations; i++ { networkName := uuid.NewV4().String() resp, err := dockerClient.NetworkCreate(ctx, networkName, types.NetworkCreate{ Labels: cleanupLabels, }) require.NoError(err) networkNames = append(networkNames, networkName) networkIDs = append(networkIDs, resp.ID) } // ListByLabels networks, err := dockerClient.NetworkListByLabels(ctx, cleanupLabels) require.NoError(err) require.Len(networks, iterations) for _, network := range networks { require.Contains(networkIDs, network.ID) require.Contains(networkNames, network.Name) } // ListByLabels, network with label does not exist networks, err = dockerClient.NetworkListByLabels(ctx, map[string]string{uuid.NewV4().String(): uuid.NewV4().String()}) require.NoError(err) require.Len(networks, 0) // ListByNames networks, err = dockerClient.NetworkListByNames(ctx, networkNames...) require.NoError(err) require.Len(networks, iterations) for _, network := range networks { require.Contains(networkIDs, network.ID) require.Contains(networkNames, network.Name) } // ListByNames, network with names does not exist networks, err = dockerClient.NetworkListByNames(ctx, uuid.NewV4().String(), uuid.NewV4().String()) require.Error(err) require.Nil(networks) // RemoveByLabels err = dockerClient.NetworkRemoveByLabels(ctx, cleanupLabels) require.NoError(err) networks, err = dockerClient.NetworkListByLabels(ctx, cleanupLabels) require.NoError(err) require.Len(networks, 0) // RemoveByLabels, label does not exists err = dockerClient.NetworkRemoveByLabels(ctx, map[string]string{uuid.NewV4().String(): uuid.NewV4().String()}) require.NoError(err) // Create Networks networkIDs = make([]string, 0) networkNames = make([]string, 0) for i := 0; i < iterations; i++ { networkName := uuid.NewV4().String() resp, err := dockerClient.NetworkCreate(ctx, networkName, types.NetworkCreate{ Labels: cleanupLabels, }) require.NoError(err) networkNames = append(networkNames, networkName) networkIDs = append(networkIDs, resp.ID) } // RemoveByNames err = dockerClient.NetworkRemoveByNames(ctx, networkNames...) require.NoError(err) networks, err = dockerClient.NetworkListByNames(ctx, networkNames...) require.Error(err) require.Nil(networks) // RemoveByNames, name does not exists err = dockerClient.NetworkRemoveByNames(ctx, uuid.NewV4().String()) require.Error(err) // Create Networks networkIDs = make([]string, 0) networkNames = make([]string, 0) for i := 0; i < iterations; i++ { networkName := uuid.NewV4().String() resp, err := dockerClient.NetworkCreate(ctx, networkName, types.NetworkCreate{ Labels: cleanupLabels, }) require.NoError(err) networkNames = append(networkNames, networkName) networkIDs = append(networkIDs, resp.ID) } // RemoveByIDs err = dockerClient.NetworkRemoveByIDs(ctx, networkIDs...) require.NoError(err) networks, err = dockerClient.NetworkListByNames(ctx, networkNames...) require.Error(err) require.Nil(networks) // RemoveByID, id does not exists err = dockerClient.NetworkRemoveByIDs(ctx, uuid.NewV4().String()) require.Error(err) } func TestVolumeCRUD(t *testing.T) { var ( ctx = context.Background() require = require.New(t) iterations = 5 cleanupLabels = map[string]string{ uuid.NewV4().String(): uuid.NewV4().String(), } ) dockerClient, err := New() require.NoError(err) t.Cleanup(func() { dockerClient.VolumeRemoveByLabels(ctx, cleanupLabels) }) // Create Volumes volumeNames := make([]string, 0) for i := 0; i < iterations; i++ { volumeName := uuid.NewV4().String() volume, err := dockerClient.VolumeCreate(ctx, volume.VolumesCreateBody{ Name: volumeName, Labels: cleanupLabels, }) require.NoError(err) volumeNames = append(volumeNames, volume.Name) } // ListByLabels volumes, err := dockerClient.VolumeListByLabels(ctx, cleanupLabels) require.NoError(err) require.Len(volumes.Volumes, iterations) for _, volume := range volumes.Volumes { require.Contains(volumeNames, volume.Name) } // ListByLabels, network with label does not exist volumes, err = dockerClient.VolumeListByLabels(ctx, map[string]string{uuid.NewV4().String(): uuid.NewV4().String()}) require.NoError(err) require.Len(volumes.Volumes, 0) // ListByNames volumes, err = dockerClient.VolumeListByNames(ctx, volumeNames...) require.NoError(err) require.Len(volumes.Volumes, iterations) for _, volume := range volumes.Volumes { require.Contains(volumeNames, volume.Name) } // ListByNames, network with names does not exist volumes, err = dockerClient.VolumeListByNames(ctx, uuid.NewV4().String(), uuid.NewV4().String()) require.Error(err) require.Nil(volumes.Volumes) // RemoveByLabels err = dockerClient.VolumeRemoveByLabels(ctx, cleanupLabels) require.NoError(err) volumes, err = dockerClient.VolumeListByLabels(ctx, cleanupLabels) require.NoError(err) require.Len(volumes.Volumes, 0) // RemoveByLabels, labels does not exists err = dockerClient.NetworkRemoveByLabels(ctx, map[string]string{uuid.NewV4().String(): uuid.NewV4().String()}) require.NoError(err) // Create Volumes volumeNames = make([]string, 0) for i := 0; i < iterations; i++ { volumeName := uuid.NewV4().String() volume, err := dockerClient.VolumeCreate(ctx, volume.VolumesCreateBody{ Name: volumeName, Labels: cleanupLabels, }) require.NoError(err) volumeNames = append(volumeNames, volume.Name) } // RemoveByNames err = dockerClient.VolumeRemoveByNames(ctx, volumeNames...) require.NoError(err) volumes, err = dockerClient.VolumeListByNames(ctx, volumeNames...) require.Error(err) require.Nil(volumes.Volumes) // RemoveByNames, name does not exists err = dockerClient.NetworkRemoveByNames(ctx, uuid.NewV4().String()) require.Error(err) } // TestContainerMultipleNetworks // Test if a container can be accessed ofer multiple networks/ips. func TestContainerMultipleNetworks(t *testing.T) { var ( ctx = context.Background() require = require.New(t) iterations = 5 cleanupLabels = map[string]string{ uuid.NewV4().String(): uuid.NewV4().String(), } ) dockerClient, err := New() require.NoError(err) t.Cleanup(func() { dockerClient.ContainerRemoveByLabels(ctx, cleanupLabels) dockerClient.NetworkRemoveByLabels(ctx, cleanupLabels) }) // Create Containers containerIDs := make([]string, 0) containerNames := make([]string, 0) containersNetworks := make(map[string]map[string][]string, 0) for i := 0; i < iterations; i++ { containerName := uuid.NewV4().String() containerNetworks := map[string][]string{ uuid.NewV4().String(): { uuid.NewV4().String(), uuid.NewV4().String(), }, uuid.NewV4().String(): { uuid.NewV4().String(), uuid.NewV4().String(), }, } builder := dockerClient.NewBuilder("nginx:alpine"). Labels(cleanupLabels). Port("80"). Pull(). WithName(containerName) for networkName, aliasses := range containerNetworks { _, err := dockerClient.NetworkCreate(ctx, networkName, types.NetworkCreate{ Labels: cleanupLabels, }) require.NoError(err) builder.Network(networkName, aliasses...) } containerID, err := builder.Start(ctx) require.NoError(err) containerNames = append(containerNames, containerName) containerIDs = append(containerIDs, containerID) containersNetworks[containerID] = containerNetworks } for containerID, containerNetworks := range containersNetworks { for networkName := range containerNetworks { networks, err := dockerClient.NetworkListByNames(ctx, networkName) require.NoError(err) for _, network := range networks { if _, present := network.Containers[containerID]; !present { require.Fail("Container %v not found in network %v", containerID, network.ID) } networkIPParts := strings.Split(network.Containers[containerID].IPv4Address, "/") url := fmt.Sprintf("http://%v", networkIPParts[0]) resp, err := http.Get(url) require.NoError(err) defer resp.Body.Close() require.Equal(http.StatusOK, resp.StatusCode) } } } }