Skip to content

Commit c63ebb6

Browse files
committed
fix empty provider
Signed-off-by: zbb88888 <jmdxjsjgcxy@gmail.com>
1 parent e1fb570 commit c63ebb6

File tree

2 files changed

+30
-4
lines changed

2 files changed

+30
-4
lines changed

pkg/util/vpc_nat_gateway.go

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,15 +61,19 @@ func GenNatGwSelectors(selectors []string) map[string]string {
6161
// GenNatGwPodAnnotations returns the Pod annotations for a NAT gateway
6262
// additionalNetworks is optional, used when user specifies extra NADs in gw.Annotations
6363
func GenNatGwPodAnnotations(gw *kubeovnv1.VpcNatGateway, externalNadNamespace, externalNadName, provider, additionalNetworks string) map[string]string {
64+
p := provider
65+
if p == "" {
66+
p = OvnProvider
67+
}
6468
attachedNetworks := fmt.Sprintf("%s/%s", externalNadNamespace, externalNadName)
6569
if additionalNetworks != "" {
6670
attachedNetworks = additionalNetworks + ", " + attachedNetworks
6771
}
6872
return map[string]string{
69-
fmt.Sprintf(VpcNatGatewayAnnotationTemplate, provider): gw.Name,
70-
nadv1.NetworkAttachmentAnnot: attachedNetworks,
71-
fmt.Sprintf(LogicalSwitchAnnotationTemplate, provider): gw.Spec.Subnet,
72-
fmt.Sprintf(IPAddressAnnotationTemplate, provider): gw.Spec.LanIP,
73+
fmt.Sprintf(VpcNatGatewayAnnotationTemplate, p): gw.Name,
74+
nadv1.NetworkAttachmentAnnot: attachedNetworks,
75+
fmt.Sprintf(LogicalSwitchAnnotationTemplate, p): gw.Spec.Subnet,
76+
fmt.Sprintf(IPAddressAnnotationTemplate, p): gw.Spec.LanIP,
7377
}
7478
}
7579

pkg/util/vpc_nat_gateway_test.go

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,28 @@ func TestGenNatGwPodAnnotations(t *testing.T) {
270270
additionalNetworks string
271271
expected map[string]string
272272
}{
273+
{
274+
name: "Empty provider defaults to ovn",
275+
gw: v1.VpcNatGateway{
276+
ObjectMeta: metav1.ObjectMeta{
277+
Name: "test-gateway",
278+
},
279+
Spec: v1.VpcNatGatewaySpec{
280+
Subnet: "internal-subnet",
281+
LanIP: "10.20.30.40",
282+
},
283+
},
284+
externalNadName: "external-subnet",
285+
externalNadNamespace: metav1.NamespaceSystem,
286+
provider: "",
287+
additionalNetworks: "",
288+
expected: map[string]string{
289+
VpcNatGatewayAnnotation: "test-gateway",
290+
nadv1.NetworkAttachmentAnnot: "kube-system/external-subnet",
291+
LogicalSwitchAnnotation: "internal-subnet",
292+
IPAddressAnnotation: "10.20.30.40",
293+
},
294+
},
273295
{
274296
name: "All fields provided with ovn provider",
275297
gw: v1.VpcNatGateway{

0 commit comments

Comments
 (0)