Fix fake cloud to support snapshot not found case

kubernetes-sigs · May 13, 2019 · d6030c0 · d6030c0
1 parent d7e003a
commit d6030c0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/sanity/fake_cloud_provider.go b/tests/sanity/fake_cloud_provider.go
@@ -64,6 +64,11 @@ func (c *fakeCloudProvider) GetMetadata() cloud.MetadataService {
 
 func (c *fakeCloudProvider) CreateDisk(ctx context.Context, volumeName string, diskOptions *cloud.DiskOptions) (*cloud.Disk, error) {
 	r1 := rand.New(rand.NewSource(time.Now().UnixNano()))
+	if len(diskOptions.SnapshotID) > 0 {
+		if _, ok := c.snapshots[diskOptions.SnapshotID]; !ok {
+			return nil, cloud.ErrNotFound
+		}
+	}
 	d := &fakeDisk{
 		Disk: &cloud.Disk{
 			VolumeID:         fmt.Sprintf("vol-%d", r1.Uint64()),