def test_construct_public_vi():
module = FakeModule(state="present",
id_to_associate=None,
public=True,
name="aaaaaaaa",
vlan=1,
bgp_asn=123,
authentication_key="aaaa",
customer_address="169.254.0.1/30",
amazon_address="169.254.0.2/30",
address_type="ipv4",
cidr=["10.88.0.0/30"],
virtual_gateway_id="xxxx",
direct_connect_gateway_id="yyyy")
vi = aws_direct_connect_virtual_interface.assemble_params_for_creating_vi(
module.params)
assert vi == {
"virtualInterfaceName": "aaaaaaaa",
"vlan": 1,
"asn": 123,
"authKey": "aaaa",
"amazonAddress": "169.254.0.2/30",
"customerAddress": "169.254.0.1/30",
"addressFamily": "ipv4",
"routeFilterPrefixes": [{
"cidr": "10.88.0.0/30"
}]
}
def test_construct_private_vi_with_direct_connect_gateway_id():
module = FakeModule(state="present",
id_to_associate=None,
public=False,
name="aaaaaaaa",
vlan=1,
bgp_asn=123,
authentication_key="aaaa",
customer_address="169.254.0.1/30",
amazon_address="169.254.0.2/30",
address_type="ipv4",
cidr=["10.88.0.0/30"],
virtual_gateway_id=None,
direct_connect_gateway_id="yyyy")
vi = aws_direct_connect_virtual_interface.assemble_params_for_creating_vi(
module.params)
print(vi)
assert vi == {
"virtualInterfaceName": "aaaaaaaa",
"vlan": 1,
"asn": 123,
"authKey": "aaaa",
"amazonAddress": "169.254.0.2/30",
"customerAddress": "169.254.0.1/30",
"addressFamily": "ipv4",
"directConnectGatewayId": "yyyy"
}