mirror of
https://github.com/zalando/postgres-operator.git
synced 2026-10-04 15:17:45 +02:00
Add support for EBS CSI Driver (#2677)
* Add support for EBS CSI Driver
This commit is contained in:
@@ -151,7 +151,7 @@ func (c *Cluster) populateVolumeMetaData() error {
|
||||
volumeIds := []string{}
|
||||
var volumeID string
|
||||
for _, pv := range pvs {
|
||||
volumeID, err = c.VolumeResizer.ExtractVolumeID(pv.Spec.AWSElasticBlockStore.VolumeID)
|
||||
volumeID, err = c.VolumeResizer.GetProviderVolumeID(pv)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
@@ -216,6 +216,12 @@ func TestMigrateEBS(t *testing.T) {
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-1")).Return("ebs-volume-1", nil)
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-2")).Return("ebs-volume-2", nil)
|
||||
|
||||
resizer.EXPECT().GetProviderVolumeID(gomock.Any()).
|
||||
DoAndReturn(func(pv *v1.PersistentVolume) (string, error) {
|
||||
return resizer.ExtractVolumeID(pv.Spec.AWSElasticBlockStore.VolumeID)
|
||||
}).
|
||||
Times(2)
|
||||
|
||||
resizer.EXPECT().DescribeVolumes(gomock.Eq([]string{"ebs-volume-1", "ebs-volume-2"})).Return(
|
||||
[]volumes.VolumeProperties{
|
||||
{VolumeID: "ebs-volume-1", VolumeType: "gp2", Size: 100},
|
||||
@@ -322,6 +328,12 @@ func TestMigrateGp3Support(t *testing.T) {
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-2")).Return("ebs-volume-2", nil)
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-3")).Return("ebs-volume-3", nil)
|
||||
|
||||
resizer.EXPECT().GetProviderVolumeID(gomock.Any()).
|
||||
DoAndReturn(func(pv *v1.PersistentVolume) (string, error) {
|
||||
return resizer.ExtractVolumeID(pv.Spec.AWSElasticBlockStore.VolumeID)
|
||||
}).
|
||||
Times(3)
|
||||
|
||||
resizer.EXPECT().DescribeVolumes(gomock.Eq([]string{"ebs-volume-1", "ebs-volume-2", "ebs-volume-3"})).Return(
|
||||
[]volumes.VolumeProperties{
|
||||
{VolumeID: "ebs-volume-1", VolumeType: "gp3", Size: 100, Iops: 3000},
|
||||
@@ -377,6 +389,12 @@ func TestManualGp2Gp3Support(t *testing.T) {
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-1")).Return("ebs-volume-1", nil)
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-2")).Return("ebs-volume-2", nil)
|
||||
|
||||
resizer.EXPECT().GetProviderVolumeID(gomock.Any()).
|
||||
DoAndReturn(func(pv *v1.PersistentVolume) (string, error) {
|
||||
return resizer.ExtractVolumeID(pv.Spec.AWSElasticBlockStore.VolumeID)
|
||||
}).
|
||||
Times(2)
|
||||
|
||||
resizer.EXPECT().DescribeVolumes(gomock.Eq([]string{"ebs-volume-1", "ebs-volume-2"})).Return(
|
||||
[]volumes.VolumeProperties{
|
||||
{VolumeID: "ebs-volume-1", VolumeType: "gp2", Size: 150, Iops: 3000},
|
||||
@@ -436,6 +454,12 @@ func TestDontTouchType(t *testing.T) {
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-1")).Return("ebs-volume-1", nil)
|
||||
resizer.EXPECT().ExtractVolumeID(gomock.Eq("aws://eu-central-1b/ebs-volume-2")).Return("ebs-volume-2", nil)
|
||||
|
||||
resizer.EXPECT().GetProviderVolumeID(gomock.Any()).
|
||||
DoAndReturn(func(pv *v1.PersistentVolume) (string, error) {
|
||||
return resizer.ExtractVolumeID(pv.Spec.AWSElasticBlockStore.VolumeID)
|
||||
}).
|
||||
Times(2)
|
||||
|
||||
resizer.EXPECT().DescribeVolumes(gomock.Eq([]string{"ebs-volume-1", "ebs-volume-2"})).Return(
|
||||
[]volumes.VolumeProperties{
|
||||
{VolumeID: "ebs-volume-1", VolumeType: "gp2", Size: 150, Iops: 3000},
|
||||
|
||||
Reference in New Issue
Block a user