mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-28 03:53:33 +00:00
feat: Label new volumes as managed by uncloud (#79)
This commit is contained in:
@@ -345,6 +345,12 @@ func (s *Server) CreateVolume(ctx context.Context, req *pb.CreateVolumeRequest)
|
|||||||
return nil, status.Errorf(codes.InvalidArgument, "unmarshal options: %v", err)
|
return nil, status.Errorf(codes.InvalidArgument, "unmarshal options: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Always add the managed label to the volume to indicate that it is managed by Uncloud.
|
||||||
|
if opts.Labels == nil {
|
||||||
|
opts.Labels = make(map[string]string)
|
||||||
|
}
|
||||||
|
opts.Labels[api.LabelManaged] = ""
|
||||||
|
|
||||||
vol, err := s.client.VolumeCreate(ctx, opts)
|
vol, err := s.client.VolumeCreate(ctx, opts)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, status.Error(codes.Internal, err.Error())
|
return nil, status.Error(codes.Internal, err.Error())
|
||||||
|
|||||||
@@ -121,6 +121,9 @@ func TestComposeDeployment(t *testing.T) {
|
|||||||
})
|
})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Len(t, volumes, 2, "Expected 2 volumes to be created")
|
assert.Len(t, volumes, 2, "Expected 2 volumes to be created")
|
||||||
|
// Check that the volumes have the correct labels
|
||||||
|
assert.Equal(t, map[string]string{"uncloud.managed": ""}, volumes[0].Volume.Labels)
|
||||||
|
assert.Equal(t, map[string]string{"uncloud.managed": ""}, volumes[1].Volume.Labels)
|
||||||
|
|
||||||
data1Volume, data2Volume := volumes[0], volumes[1]
|
data1Volume, data2Volume := volumes[0], volumes[1]
|
||||||
if volumes[0].Volume.Name == "test-compose-volumes-data2" {
|
if volumes[0].Volume.Name == "test-compose-volumes-data2" {
|
||||||
|
|||||||
Reference in New Issue
Block a user