Fix mapping of ids in test.
This commit is contained in:
		
							parent
							
								
									0bab888dc7
								
							
						
					
					
						commit
						51f4683204
					
				| 
						 | 
				
			
			@ -233,7 +233,11 @@ func TestMigrateEBS(t *testing.T) {
 | 
			
		|||
 | 
			
		||||
	ps := v1.PersistentVolumeSpec{}
 | 
			
		||||
	ps.AWSElasticBlockStore = &v1.AWSElasticBlockStoreVolumeSource{}
 | 
			
		||||
	ps.AWSElasticBlockStore.VolumeID = "vol-1111"
 | 
			
		||||
	ps.AWSElasticBlockStore.VolumeID = "ebs-volume-1"
 | 
			
		||||
 | 
			
		||||
	ps2 := v1.PersistentVolumeSpec{}
 | 
			
		||||
	ps2.AWSElasticBlockStore = &v1.AWSElasticBlockStoreVolumeSource{}
 | 
			
		||||
	ps2.AWSElasticBlockStore.VolumeID = "ebs-volume-2"
 | 
			
		||||
 | 
			
		||||
	pvList := &v1.PersistentVolumeList{
 | 
			
		||||
		Items: []v1.PersistentVolume{
 | 
			
		||||
| 
						 | 
				
			
			@ -243,6 +247,12 @@ func TestMigrateEBS(t *testing.T) {
 | 
			
		|||
				},
 | 
			
		||||
				Spec: ps,
 | 
			
		||||
			},
 | 
			
		||||
			{
 | 
			
		||||
				ObjectMeta: metav1.ObjectMeta{
 | 
			
		||||
					Name: "persistent-volume-1",
 | 
			
		||||
				},
 | 
			
		||||
				Spec: ps2,
 | 
			
		||||
			},
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -262,14 +272,24 @@ func TestMigrateEBS(t *testing.T) {
 | 
			
		|||
 | 
			
		||||
	cluster.KubeClient.Pods(namespace).Create(context.TODO(), &pod, metav1.CreateOptions{})
 | 
			
		||||
 | 
			
		||||
	pod = v1.Pod{
 | 
			
		||||
		ObjectMeta: metav1.ObjectMeta{
 | 
			
		||||
			Name:   clusterName + "-1",
 | 
			
		||||
			Labels: filterLabels,
 | 
			
		||||
		},
 | 
			
		||||
		Spec: v1.PodSpec{},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	cluster.KubeClient.Pods(namespace).Create(context.TODO(), &pod, metav1.CreateOptions{})
 | 
			
		||||
 | 
			
		||||
	ctrl := gomock.NewController(t)
 | 
			
		||||
	defer ctrl.Finish()
 | 
			
		||||
 | 
			
		||||
	resizer := mocks.NewMockVolumeResizer(ctrl)
 | 
			
		||||
	resizer.EXPECT().DescribeVolumes(gomock.Eq([]string{"vol-1111"})).Return(
 | 
			
		||||
		[]volumes.VolumeProperties{
 | 
			
		||||
			{VolumeID: "persistent-volume-0", VolumeType: "gp2", Size: 100},
 | 
			
		||||
			{VolumeID: "persistent-volume-1", VolumeType: "gp3", Size: 100}}, nil)
 | 
			
		||||
			{VolumeID: "ebs-volume-1", VolumeType: "gp2", Size: 100},
 | 
			
		||||
			{VolumeID: "ebs-volume-2", VolumeType: "gp3", Size: 100}}, nil)
 | 
			
		||||
 | 
			
		||||
	// expect only gp2 volume to be modified
 | 
			
		||||
	resizer.EXPECT().ModifyVolume(gomock.Eq("persistent-volume-0"), gomock.Eq("gp3"), gomock.Any(), gomock.Any(), gomock.Any()).Return(nil)
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue