diff --git a/pkg/compose/volumes_test.go b/pkg/compose/volumes_test.go new file mode 100644 index 000000000..8e149159b --- /dev/null +++ b/pkg/compose/volumes_test.go @@ -0,0 +1,89 @@ +/* + Copyright 2020 Docker Compose CLI authors + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. +*/ + +package compose + +import ( + "context" + "testing" + + "github.com/compose-spec/compose-go/v2/types" + "github.com/docker/compose/v2/pkg/api" + "github.com/docker/docker/api/types/container" + "github.com/docker/docker/api/types/filters" + "github.com/docker/docker/api/types/volume" + "go.uber.org/mock/gomock" + "gotest.tools/v3/assert" +) + +func TestVolumes(t *testing.T) { + mockCtrl := gomock.NewController(t) + defer mockCtrl.Finish() + + mockApi, mockCli := prepareMocks(mockCtrl) + tested := composeService{ + dockerCli: mockCli, + } + + // Create test volumes + vol1 := &volume.Volume{Name: testProject + "_vol1"} + vol2 := &volume.Volume{Name: testProject + "_vol2"} + vol3 := &volume.Volume{Name: testProject + "_vol3"} + + // Create test containers with volume mounts + c1 := container.Summary{ + Labels: map[string]string{api.ServiceLabel: "service1"}, + Mounts: []container.MountPoint{ + {Name: testProject + "_vol1"}, + {Name: testProject + "_vol2"}, + }, + } + c2 := container.Summary{ + Labels: map[string]string{api.ServiceLabel: "service2"}, + Mounts: []container.MountPoint{ + {Name: testProject + "_vol3"}, + }, + } + + ctx := context.Background() + project := &types.Project{Name: testProject} + args := filters.NewArgs(projectFilter(testProject)) + listOpts := container.ListOptions{Filters: args} + volumeListArgs := filters.NewArgs(projectFilter(testProject)) + volumeListOpts := volume.ListOptions{Filters: volumeListArgs} + volumeReturn := volume.ListResponse{ + Volumes: []*volume.Volume{vol1, vol2, vol3}, + } + containerReturn := []container.Summary{c1, c2} + + // Mock API calls + mockApi.EXPECT().ContainerList(ctx, listOpts).Times(2).Return(containerReturn, nil) + mockApi.EXPECT().VolumeList(ctx, volumeListOpts).Times(2).Return(volumeReturn, nil) + + // Test without service filter - should return all project volumes + volumeOptions := api.VolumesOptions{} + volumes, err := tested.Volumes(ctx, project, volumeOptions) + expected := []api.VolumesSummary{vol1, vol2, vol3} + assert.NilError(t, err) + assert.DeepEqual(t, volumes, expected) + + // Test with service filter - should only return volumes used by service1 + volumeOptions = api.VolumesOptions{Services: []string{"service1"}} + volumes, err = tested.Volumes(ctx, project, volumeOptions) + expected = []api.VolumesSummary{vol1, vol2} + assert.NilError(t, err) + assert.DeepEqual(t, volumes, expected) +}