partner-network-connect: Add redundancy zone. (#840)

wez470 · wcarlson-digitalocean · web-flow · commit 9fe1e48fccda · 2025-05-27T10:23:07.000+05:30
Co-authored-by: Weston Carlson &lt;wcarlson@digitalocean.com&gt;
diff --git a/partner_network_connect.go b/partner_network_connect.go
@@ -47,6 +47,8 @@ type PartnerAttachmentCreateRequest struct {
 	VPCIDs []string `json:"vpc_ids,omitempty"`
 	// BGP is the BGP configuration of the Partner Attachment
 	BGP BGP `json:"bgp,omitempty"`
+	// RedundancyZone is the redundancy zone of the Partner Attachment
+	RedundancyZone string `json:"redundancy_zone,omitempty"`
 }
 
 type partnerAttachmentRequestBody struct {
@@ -62,6 +64,8 @@ type partnerAttachmentRequestBody struct {
 	VPCIDs []string `json:"vpc_ids,omitempty"`
 	// BGP is the BGP configuration of the Partner Attachment
 	BGP *BGPInput `json:"bgp,omitempty"`
+	// RedundancyZone is the redundancy zone of the Partner Attachment
+	RedundancyZone string `json:"redundancy_zone,omitempty"`
 }
 
 func (req *PartnerAttachmentCreateRequest) buildReq() *partnerAttachmentRequestBody {
@@ -71,6 +75,7 @@ func (req *PartnerAttachmentCreateRequest) buildReq() *partnerAttachmentRequestB
 		Region:                    req.Region,
 		NaaSProvider:              req.NaaSProvider,
 		VPCIDs:                    req.VPCIDs,
+		RedundancyZone:            req.RedundancyZone,
 	}
 
 	if req.BGP != (BGP{}) {
@@ -189,6 +194,8 @@ type PartnerAttachment struct {
 	VPCIDs []string `json:"vpc_ids,omitempty"`
 	// BGP is the BGP configuration of the Partner Attachment
 	BGP BGP `json:"bgp,omitempty"`
+	// RedundancyZone is the redundancy zone of the Partner Attachment
+	RedundancyZone string `json:"redundancy_zone,omitempty"`
 	// CreatedAt is time when this Partner Attachment was first created
 	CreatedAt time.Time `json:"created_at,omitempty"`
 }
diff --git a/partner_network_connect_test.go b/partner_network_connect_test.go
@@ -26,7 +26,8 @@ var vPartnerAttachmentTestObj = &PartnerAttachment{
 		PeerRouterIP:  "169.250.0.6",
 		AuthKey:       "my-auth-key",
 	},
-	CreatedAt: time.Date(2024, 12, 26, 21, 48, 40, 995304079, time.UTC),
+	CreatedAt:      time.Date(2024, 12, 26, 21, 48, 40, 995304079, time.UTC),
+	RedundancyZone: "MEGAPORT_RED",
 }
 
 var vPartnerAttachmentNoBGPTestObj = &PartnerAttachment{
@@ -38,6 +39,7 @@ var vPartnerAttachmentNoBGPTestObj = &PartnerAttachment{
 	NaaSProvider:              "MEGAPORT",
 	VPCIDs:                    []string{"f5a0c5e4-7537-47de-bb8d-46c766f89ffb"},
 	CreatedAt:                 time.Date(2024, 12, 26, 21, 48, 40, 995304079, time.UTC),
+	RedundancyZone:            "MEGAPORT_RED",
 }
 
 var vPartnerAttachmentTestJSON = `
@@ -56,7 +58,8 @@ var vPartnerAttachmentTestJSON = `
 			"peer_router_ip":"169.250.0.6",
 			"auth_key":"my-auth-key"
 			},
-		"created_at":"2024-12-26T21:48:40.995304079Z"
+		"created_at":"2024-12-26T21:48:40.995304079Z",
+		"redundancy_zone": "MEGAPORT_RED"
 	}
 `
 
@@ -69,7 +72,8 @@ var vPartnerAttachmentNoBGPTestJSON = `
 		"region":"NYC",
 		"naas_provider":"MEGAPORT",
 		"vpc_ids":["f5a0c5e4-7537-47de-bb8d-46c766f89ffb"],
-		"created_at":"2024-12-26T21:48:40.995304079Z"
+		"created_at":"2024-12-26T21:48:40.995304079Z",
+		"redundancy_zone": "MEGAPORT_RED"
 	}
 `
 
@@ -139,6 +143,7 @@ func TestPartnerAttachment_Create(t *testing.T) {
 			PeerASN:       133937,
 			PeerRouterIP:  "169.250.0.6",
 		},
+		RedundancyZone: "MEGAPORT_BLUE",
 	}
 	jsonBlob := `
 {

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,8 @@ var vPartnerAttachmentTestObj = &PartnerAttachment{`
`26`	`26`	`PeerRouterIP: "169.250.0.6",`
`27`	`27`	`AuthKey: "my-auth-key",`
`28`	`28`	`},`
`29`		`- CreatedAt: time.Date(2024, 12, 26, 21, 48, 40, 995304079, time.UTC),`
	`29`	`+ CreatedAt: time.Date(2024, 12, 26, 21, 48, 40, 995304079, time.UTC),`
	`30`	`+ RedundancyZone: "MEGAPORT_RED",`
`30`	`31`	`}`
`31`	`32`
`32`	`33`	`var vPartnerAttachmentNoBGPTestObj = &PartnerAttachment{`
`@@ -38,6 +39,7 @@ var vPartnerAttachmentNoBGPTestObj = &PartnerAttachment{`
`38`	`39`	`NaaSProvider: "MEGAPORT",`
`39`	`40`	`VPCIDs: []string{"f5a0c5e4-7537-47de-bb8d-46c766f89ffb"},`
`40`	`41`	`CreatedAt: time.Date(2024, 12, 26, 21, 48, 40, 995304079, time.UTC),`
	`42`	`+ RedundancyZone: "MEGAPORT_RED",`
`41`	`43`	`}`
`42`	`44`
`43`	`45`	var vPartnerAttachmentTestJSON = `
@@ -56,7 +58,8 @@ var vPartnerAttachmentTestJSON = `
`56`	`58`	`"peer_router_ip":"169.250.0.6",`
`57`	`59`	`"auth_key":"my-auth-key"`
`58`	`60`	`},`
`59`		`- "created_at":"2024-12-26T21:48:40.995304079Z"`
	`61`	`+ "created_at":"2024-12-26T21:48:40.995304079Z",`
	`62`	`+ "redundancy_zone": "MEGAPORT_RED"`
`60`	`63`	`}`
`61`	`64`	`
`62`	`65`
@@ -69,7 +72,8 @@ var vPartnerAttachmentNoBGPTestJSON = `
`69`	`72`	`"region":"NYC",`
`70`	`73`	`"naas_provider":"MEGAPORT",`
`71`	`74`	`"vpc_ids":["f5a0c5e4-7537-47de-bb8d-46c766f89ffb"],`
`72`		`- "created_at":"2024-12-26T21:48:40.995304079Z"`
	`75`	`+ "created_at":"2024-12-26T21:48:40.995304079Z",`
	`76`	`+ "redundancy_zone": "MEGAPORT_RED"`
`73`	`77`	`}`
`74`	`78`	`
`75`	`79`
`@@ -139,6 +143,7 @@ func TestPartnerAttachment_Create(t *testing.T) {`
`139`	`143`	`PeerASN: 133937,`
`140`	`144`	`PeerRouterIP: "169.250.0.6",`
`141`	`145`	`},`
	`146`	`+ RedundancyZone: "MEGAPORT_BLUE",`
`142`	`147`	`}`
`143`	`148`	jsonBlob := `
`144`	`149`	`{`