Skip to content

Commit 74a10f2

Browse files
committed
unit test updates
1 parent 3422ec6 commit 74a10f2

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pkg/asset/manifests/infrastructure_test.go

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,7 @@ func TestGenerateInfrastructure(t *testing.T) {
136136
infraBuild.withVSphereMachineNetworkEntry(configv1.CIDR(ipv4IngressVIP+"/32")),
137137
infraBuild.withVSphereAPIVIP(ipv4APIVIP),
138138
infraBuild.withVSphereIngressVIP(ipv4IngressVIP),
139+
infraBuild.withVSphereNodeNetworkingEntry(configv1.CIDR(defaultMachineNetwork)),
139140
),
140141
expectedFilesGenerated: 1,
141142
}, {
@@ -161,6 +162,8 @@ func TestGenerateInfrastructure(t *testing.T) {
161162
infraBuild.withVSphereIngressVIP(ipv4IngressVIP),
162163
infraBuild.withVSphereAPIVIP(ipv6APIVIP),
163164
infraBuild.withVSphereIngressVIP(ipv6IngressVIP),
165+
infraBuild.withVSphereNodeNetworkingEntry(configv1.CIDR(defaultMachineNetwork)),
166+
infraBuild.withVSphereNodeNetworkingEntry(configv1.CIDR(ipv6MachineNetwork)),
164167
),
165168
expectedFilesGenerated: 1,
166169
},
@@ -180,6 +183,8 @@ func TestGenerateInfrastructure(t *testing.T) {
180183
infraBuild.withVSphereMachineNetworkEntry(configv1.CIDR(ipv4IngressVIP+"/32")),
181184
infraBuild.withVSphereAPIVIP(ipv4APIVIP),
182185
infraBuild.withVSphereIngressVIP(ipv4IngressVIP),
186+
infraBuild.withVSphereNodeNetworkingEntry(configv1.CIDR(defaultMachineNetwork)),
187+
infraBuild.withVSphereNodeNetworkingEntry(configv1.CIDR(ipv4APIVIP+"/32")),
183188
),
184189
expectedFilesGenerated: 1,
185190
},
@@ -197,6 +202,7 @@ func TestGenerateInfrastructure(t *testing.T) {
197202
infraBuild.withVSphereMachineNetworkEntry(configv1.CIDR(ipv4APIVIP+"/32")),
198203
infraBuild.withVSphereAPIVIP(ipv4APIVIP),
199204
infraBuild.withVSphereIngressVIP(ipv4APIVIP),
205+
infraBuild.withVSphereNodeNetworkingEntry(configv1.CIDR(defaultMachineNetwork)),
200206
),
201207
expectedFilesGenerated: 1,
202208
},
@@ -551,6 +557,15 @@ func (b infraBuildNamespace) withVSphereMachineNetworkEntry(cidr configv1.CIDR)
551557
}
552558
}
553559

560+
func (b infraBuildNamespace) withVSphereNodeNetworkingEntry(cidr configv1.CIDR) infraOption {
561+
return func(infra *configv1.Infrastructure) {
562+
b.withVSpherePlatformSpec()(infra)
563+
b.withVSpherePlatformStatus()(infra)
564+
infra.Spec.PlatformSpec.VSphere.NodeNetworking.External.NetworkSubnetCIDR = append(infra.Spec.PlatformSpec.VSphere.NodeNetworking.External.NetworkSubnetCIDR, string(cidr))
565+
infra.Spec.PlatformSpec.VSphere.NodeNetworking.Internal.NetworkSubnetCIDR = append(infra.Spec.PlatformSpec.VSphere.NodeNetworking.Internal.NetworkSubnetCIDR, string(cidr))
566+
}
567+
}
568+
554569
func (b infraBuildNamespace) withVSphereAPIVIP(vip string) infraOption {
555570
return func(infra *configv1.Infrastructure) {
556571
b.withVSpherePlatformSpec()(infra)

0 commit comments

Comments
 (0)