Function TestVolumes

pkg/compose/volumes_test.go:31–83

(t *testing.T)

)

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"},
		},
	}

	listOpts := client.ContainerListOptions{Filters: projectFilter(testProject)}
	volumeListOpts := client.VolumeListOptions{Filters: projectFilter(testProject)}
	volumeReturn := client.VolumeListResult{
		Items: []volume.Volume{vol1, vol2, vol3},
	}
	containerReturn := client.ContainerListResult{
		Items: []container.Summary{c1, c2},
	}

	mockApi.EXPECT().ContainerList(t.Context(), listOpts).Times(2).Return(containerReturn, nil)
	mockApi.EXPECT().VolumeList(t.Context(), volumeListOpts).Times(2).Return(volumeReturn, nil)

	// Test without service filter - should return all project volumes
	volumes, err := tested.Volumes(t.Context(), testProject, api.VolumesOptions{})
	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
	volumes, err = tested.Volumes(t.Context(), testProject, api.VolumesOptions{Services: []string{"service1"}})
	expected = []api.VolumesSummary{vol1, vol2}
	assert.NilError(t, err)
	assert.DeepEqual(t, volumes, expected)
}