76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
from __future__ import unicode_literals
|
|
import re
|
|
import sure # noqa
|
|
import xmltodict
|
|
|
|
import moto.server as server
|
|
from tests import EXAMPLE_AMI_ID
|
|
|
|
"""
|
|
Test the different server responses
|
|
"""
|
|
|
|
|
|
def test_ec2_server_get():
|
|
backend = server.create_backend_app("ec2")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.get(
|
|
"/?Action=RunInstances&ImageId=" + EXAMPLE_AMI_ID,
|
|
headers={"Host": "ec2.us-east-1.amazonaws.com"},
|
|
)
|
|
|
|
groups = re.search("<instanceId>(.*)</instanceId>", res.data.decode("utf-8"))
|
|
instance_id = groups.groups()[0]
|
|
|
|
res = test_client.get("/?Action=DescribeInstances")
|
|
res.data.decode("utf-8").should.contain(instance_id)
|
|
|
|
|
|
def test_ec2_get_unknown_vpc():
|
|
"""
|
|
Ensure that this call returns the error format in the right format
|
|
Terraform will throw errors when destroying a VPC otherwise
|
|
:return:
|
|
"""
|
|
backend = server.create_backend_app("ec2")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.get(
|
|
"/?Action=DescribeVpcs&VpcId.1=vpc-unknown",
|
|
headers={"Host": "ec2.us-east-1.amazonaws.com"},
|
|
)
|
|
|
|
res.status_code.should.equal(400)
|
|
body = xmltodict.parse(res.data.decode("utf-8"), dict_constructor=dict)
|
|
body.should.have.key("Response")
|
|
body["Response"].should.have.key("Errors")
|
|
body["Response"]["Errors"].should.have.key("Error")
|
|
error = body["Response"]["Errors"]["Error"]
|
|
error["Code"].should.equal("InvalidVpcID.NotFound")
|
|
error["Message"].should.equal("VpcID {'vpc-unknown'} does not exist.")
|
|
|
|
|
|
def test_ec2_get_unknown_route_table():
|
|
"""
|
|
Ensure that this call returns the error format in the right format
|
|
Terraform will throw errors when destroying a RouteTable otherwise
|
|
:return:
|
|
"""
|
|
backend = server.create_backend_app("ec2")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.get(
|
|
"/?Action=DescribeRouteTables&RouteTableId.1=rtb-unknown",
|
|
headers={"Host": "ec2.us-east-1.amazonaws.com"},
|
|
)
|
|
|
|
res.status_code.should.equal(400)
|
|
body = xmltodict.parse(res.data.decode("utf-8"), dict_constructor=dict)
|
|
body.should.have.key("Response")
|
|
body["Response"].should.have.key("Errors")
|
|
body["Response"]["Errors"].should.have.key("Error")
|
|
error = body["Response"]["Errors"]["Error"]
|
|
error["Code"].should.equal("InvalidRouteTableID.NotFound")
|
|
error["Message"].should.equal("The routeTable ID 'rtb-unknown' does not exist")
|