def setUp(self):
self.workspace = Workspace()
self.aws = self.workspace.add_aws(region='eu-west-1')
self.vpc = self.aws.add_vpc(name='test-vpc')
def main(ctx, debug):
g = {"workspace": Workspace()}
with open("Touchdownfile") as f:
code = compile(f.read(), "Touchdownfile", "exec")
exec(code, g)
ctx.obj = g['workspace']
class TestNetworkAclRules(unittest.TestCase):
def setUp(self):
self.workspace = Workspace()
self.aws = self.workspace.add_aws(region='eu-west-1')
self.vpc = self.aws.add_vpc(name='test-vpc')
def test_simple_rule_with_all_ports(self):
acl = self.vpc.add_network_acl(
name='test-acl',
inbound=[dict(
network='10.0.0.0/20',
protocol='tcp',
port='*',
)]
)
assert acl.inbound[0].port.start == 1
assert acl.inbound[0].port.end == 65535
def test_simple_rule_with_single_port(self):
acl = self.vpc.add_network_acl(
name='test-acl',
inbound=[dict(
network='10.0.0.0/20',
protocol='tcp',
port=20,
)]
)
assert acl.inbound[0].port.start == 20
assert acl.inbound[0].port.end == 20
def test_simple_rule_with_port_range(self):
acl = self.vpc.add_network_acl(
name='test-acl',
inbound=[dict(
network='10.0.0.0/20',
protocol='tcp',
port__start=20,
port__end=40,
)]
)
assert acl.inbound[0].port.start == 20
assert acl.inbound[0].port.end == 40
def test_mixing_port_and_port__start(self):
self.assertRaises(
errors.InvalidParameter,
self.vpc.add_network_acl,
name='test-acl',
inbound=[dict(
network='10.0.0.0/20',
protocol='tcp',
port=20,
port__start=20,
)]
)
def test_mixing_port_and_port__end(self):
self.assertRaises(
errors.InvalidParameter,
self.vpc.add_network_acl,
name='test-acl',
inbound=[dict(
network='10.0.0.0/20',
protocol='tcp',
port=20,
port__end=20,
)]
)
