fix csi tests

Author: Pearl1594

pkg/cloud/cloud.go

@@ -26,6 +26,7 @@ type Interface interface {
 
 	CreateVolumeFromSnapshot(ctx context.Context, zoneID, name, domainID, projectID, snapshotID string, sizeInGB int64) (*Volume, error)
 	GetSnapshotByID(ctx context.Context, snapshotID string) (*Snapshot, error)
+	GetSnapshotByName(ctx context.Context, name string) (*Snapshot, error)
 	CreateSnapshot(ctx context.Context, volumeID string) (*Snapshot, error)
 	DeleteSnapshot(ctx context.Context, snapshotID string) error
 }

pkg/cloud/fake/fake.go

@@ -4,6 +4,7 @@ package fake
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/hashicorp/go-uuid"
 
@@ -72,6 +73,9 @@ func (f *fakeConnector) ListZonesID(_ context.Context) ([]string, error) {
 }
 
 func (f *fakeConnector) GetVolumeByID(_ context.Context, volumeID string) (*cloud.Volume, error) {
+	if volumeID == "" {
+		return nil, fmt.Errorf("invalid volume ID: empty string")
+	}
 	vol, ok := f.volumesByID[volumeID]
 	if ok {
 		return &vol, nil
@@ -81,6 +85,9 @@ func (f *fakeConnector) GetVolumeByID(_ context.Context, volumeID string) (*clou
 }
 
 func (f *fakeConnector) GetVolumeByName(_ context.Context, name string) (*cloud.Volume, error) {
+	if name == "" {
+		return nil, fmt.Errorf("invalid volume name: empty string")
+	}
 	vol, ok := f.volumesByName[name]
 	if ok {
 		return &vol, nil
@@ -152,3 +159,13 @@ func (f *fakeConnector) CreateSnapshot(ctx context.Context, volumeID string) (*c
 func (f *fakeConnector) DeleteSnapshot(ctx context.Context, snapshotID string) error {
 	return nil
 }
+
+func (f *fakeConnector) GetSnapshotByName(_ context.Context, name string) (*cloud.Snapshot, error) {
+	if name == "" {
+		return nil, fmt.Errorf("invalid snapshot name: empty string")
+	}
+	if f.snapshot != nil && f.snapshot.Name == name {
+		return f.snapshot, nil
+	}
+	return nil, cloud.ErrNotFound
+}

pkg/cloud/snapshots.go

@@ -81,3 +81,40 @@ func (c *client) DeleteSnapshot(ctx context.Context, snapshotID string) error {
 
 	return err
 }
+
+func (c *client) GetSnapshotByName(ctx context.Context, name string) (*Snapshot, error) {
+	logger := klog.FromContext(ctx)
+	if name == "" {
+		return nil, ErrNotFound
+	}
+	p := c.Snapshot.NewListSnapshotsParams()
+	p.SetName(name)
+	if c.projectID != "" {
+		p.SetProjectid(c.projectID)
+	}
+	logger.V(2).Info("CloudStack API call", "command", "ListSnapshots", "params", map[string]string{
+		"name":      name,
+		"projectid": c.projectID,
+	})
+	l, err := c.Snapshot.ListSnapshots(p)
+	if err != nil {
+		return nil, err
+	}
+	if l.Count == 0 {
+		return nil, ErrNotFound
+	}
+	if l.Count > 1 {
+		return nil, ErrTooManyResults
+	}
+	snapshot := l.Snapshots[0]
+	s := Snapshot{
+		ID:        snapshot.Id,
+		Name:      snapshot.Name,
+		DomainID:  snapshot.Domainid,
+		ProjectID: snapshot.Projectid,
+		ZoneID:    snapshot.Zoneid,
+		VolumeID:  snapshot.Volumeid,
+		CreatedAt: snapshot.Created,
+	}
+	return &s, nil
+}

pkg/driver/controller.go

@@ -329,10 +329,27 @@ func (cs *controllerServer) CreateSnapshot(ctx context.Context, req *csi.CreateS
 	klog.V(4).Infof("CreateSnapshot")
 
 	volumeID := req.GetSourceVolumeId()
+	if volumeID == "" {
+		return nil, status.Error(codes.InvalidArgument, "SourceVolumeId missing in request")
+	}
+
+	// Check for existing snapshot with same name but different source volume ID
+	if req.GetName() != "" {
+		// check if the name matches and volumeID differs
+		existingSnap, err := cs.connector.GetSnapshotByName(ctx, req.GetName())
+		if err == nil && existingSnap.VolumeID != volumeID {
+			return nil, status.Errorf(codes.AlreadyExists, "Snapshot with name %s already exists for a different source volume", req.GetName())
+		}
+	}
+
 	volume, err := cs.connector.GetVolumeByID(ctx, volumeID)
-	if errors.Is(err, cloud.ErrNotFound) {
-		return nil, status.Errorf(codes.NotFound, "Volume %v not found", volumeID)
-	} else if err != nil {
+	if err != nil {
+		if err.Error() == "invalid volume ID: empty string" {
+			return nil, status.Error(codes.InvalidArgument, "Invalid volume ID")
+		}
+		if errors.Is(err, cloud.ErrNotFound) {
+			return nil, status.Errorf(codes.NotFound, "Volume %v not found", volumeID)
+		}
 		// Error with CloudStack
 		return nil, status.Errorf(codes.Internal, "Error %v", err)
 	}
@@ -363,6 +380,11 @@ func (cs *controllerServer) CreateSnapshot(ctx context.Context, req *csi.CreateS
 
 }
 
+func (cs *controllerServer) ListSnapshots(ctx context.Context, req *csi.ListSnapshotsRequest) (*csi.ListSnapshotsResponse, error) {
+	// Stub implementation: returns an empty list
+	return &csi.ListSnapshotsResponse{Entries: []*csi.ListSnapshotsResponse_Entry{}}, nil
+}
+
 func (cs *controllerServer) DeleteSnapshot(ctx context.Context, req *csi.DeleteSnapshotRequest) (*csi.DeleteSnapshotResponse, error) {
 	snapshotID := req.GetSnapshotId()