Add ReplaceRouteTable methods & tests

alex0z1 · Apr 2, 2014 · b309429 · b309429
1 parent 6f54e50
commit b309429
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 0 deletions.
diff --git a/ec2/responses_test.go b/ec2/responses_test.go
@@ -822,3 +822,10 @@ var AssociateRouteTableExample = `
    <associationId>rtbassoc-f8ad4891</associationId>
 </AssociateRouteTableResponse>
 `
+
+var ReplaceRouteTableAssociationExample = `
+<ReplaceRouteTableAssociationResponse xmlns="http://ec2.amazonaws.com/doc/2014-02-01/">
+   <requestId>59dbff89-35bd-4eac-88ed-be587EXAMPLE</requestId>
+   <newAssociationId>rtbassoc-faad2958</newAssociationId>
+</ReplaceRouteTableAssociationResponse>
+`
diff --git a/ec2/vpc.go b/ec2/vpc.go
@@ -95,3 +95,26 @@ func (ec2 *EC2) AssociateRouteTable(routeTableId, subnetId string) (resp *Associ
 	}
 	return
 }
+
+// ReplaceRouteTableAssociationResp represents a response from a ReplaceRouteTableAssociation call
+//
+// See  for more details.
+type ReplaceRouteTableAssociationResp struct {
+	RequestId        string `xml:"requestId"`
+	NewAssociationId string `xml:"newAssociationId"`
+}
+
+// ReplaceRouteTableAssociation changes the route table associated with a given subnet in a VPC.
+//
+// See  for more details.
+func (ec2 *EC2) ReplaceRouteTableAssociation(associationId, routeTableId string) (resp *ReplaceRouteTableAssociationResp, err error) {
+	params := makeParams("ReplaceRouteTableAssociation")
+	params["AssociationId"] = associationId
+	params["RouteTableId"] = routeTableId
+	resp = &ReplaceRouteTableAssociationResp{}
+	err = ec2.query(params, resp)
+	if err != nil {
+		return nil, err
+	}
+	return
+}
diff --git a/ec2/vpc_test.go b/ec2/vpc_test.go
@@ -59,3 +59,18 @@ func (s *S) TestAssociateRouteTable(c *gocheck.C) {
 	c.Assert(resp.RequestId, gocheck.Equals, "59dbff89-35bd-4eac-99ed-be587EXAMPLE")
 	c.Assert(resp.AssociationId, gocheck.Equals, "rtbassoc-f8ad4891")
 }
+
+func (s *S) TestReplaceRouteTableAssociation(c *gocheck.C) {
+	testServer.Response(200, nil, ReplaceRouteTableAssociationExample)
+
+	resp, err := s.ec2.ReplaceRouteTableAssociation("rtbassoc-f8ad4891", "rtb-f9ad4890")
+
+	req := testServer.WaitRequest()
+	c.Assert(req.Form["Action"], gocheck.DeepEquals, []string{"ReplaceRouteTableAssociation"})
+	c.Assert(req.Form["RouteTableId"], gocheck.DeepEquals, []string{"rtb-f9ad4890"})
+	c.Assert(req.Form["AssociationId"], gocheck.DeepEquals, []string{"rtbassoc-f8ad4891"})
+
+	c.Assert(err, gocheck.IsNil)
+	c.Assert(resp.RequestId, gocheck.Equals, "59dbff89-35bd-4eac-88ed-be587EXAMPLE")
+	c.Assert(resp.NewAssociationId, gocheck.Equals, "rtbassoc-faad2958")
+}