def setup_common_objects(self):
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
# assert self.vpc1_fixture.verify_on_setup()
if not self.vpc1_fixture.vpc_id:
return False
self.vpc2_fixture = self.useFixture(
VPCFixture(self.vpc2_cidr, connections=self.connections))
if not self.vpc2_fixture.vpc_id:
return False
# assert self.vpc2_fixture.verify_on_setup()
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
# assert self.vpc1_vn1_fixture.verify_on_setup()
if not self.vpc1_vn1_fixture.subnet_id:
return False
self.vpc1_vn2_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn2_cidr,
connections=self.connections))
if not self.vpc1_vn2_fixture.subnet_id:
return False
self.vpc2_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc2_fixture,
subnet_cidr=self.vpc2_vn1_cidr,
connections=self.connections))
if not self.vpc2_vn1_fixture.subnet_id:
return False
# assert self.vpc1_vn2_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
self.vpc1_vn1_vm2_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
self.vpc1_vn2_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn2_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
self.vpc2_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc2_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
def test_ping_between_instances(self):
"""Test ping between instances in subnet """
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
assert self.vpc1_vn1_vm1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
self.vpc1_vn1_vm2_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
assert self.vpc1_vn1_vm2_fixture.verify_on_setup()
self.vpc1_vn1_vm2_fixture.c_vm_fixture.wait_till_vm_is_up()
cidr1 = self.vpc1_cidr
vpc1_fixture = self.vpc1_fixture
vpc1_vn_fixture = self.vpc1_vn1_fixture
vpc1_vn_fixture.verify_on_setup()
assert vpc1_vn_fixture.verify_on_setup(), 'Subnet verification failed'
vm1_fixture = self.vpc1_vn1_vm1_fixture
assert vm1_fixture.verify_on_setup(
), "VPCVMFixture verification failed for VM %s" % (
vm1_fixture.instance_id)
vm2_fixture = self.vpc1_vn1_vm2_fixture
assert vm2_fixture.verify_on_setup(), "VPCVMFixture verification failed for " \
" VM %s" % (vm2_fixture.instance_id)
vm2_ip = vm2_fixture.c_vm_fixture.vm_ip
assert vm1_fixture.c_vm_fixture.ping_with_certainty( vm2_ip, expectation=True ), "Ping " \
"between two vms %s and %s failed!" % (vm1_fixture.instance_id,
vm2_fixture.instance_id)
return True
def test_instance_stop_start(self):
'''
Validate stop and start of VM using EUCA cmds
'''
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
vpc_fixture = self.vpc1_fixture
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
assert self.vpc1_vn1_vm1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
vpc_vn_fixture = self.vpc1_vn1_fixture
vm1_fixture = self.vpc1_vn1_vm1_fixture
result = True
if not vm1_fixture.stop_instance():
self.logger.error('Failed to stop instance!')
result = result and False
if vm1_fixture.verify_on_setup():
self.logger.error(
'VM Fixture verification should have failed after stopping vm!'
)
result = result and False
if not vm1_fixture.start_instance():
self.logger.error('Failed to start instance!')
result = result and False
if not vm1_fixture.verify_on_setup():
self.logger.error('VM Fixture verification failed after start vm')
result = result and False
return result
def test_run_instance(self):
"""Launch a VM in subnet """
cidr = '10.2.3.0/24'
vpc_fixture = self.useFixture(
VPCFixture(cidr, connections=self.connections))
vpc_vn_fixture = self.useFixture(
VPCVNFixture(vpc_fixture,
subnet_cidr=cidr,
connections=self.connections))
vpc_vn_fixture.verify_on_setup()
vm_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu',
connections=self.connections))
vm_fixture.verify_on_setup()
return True
def test_allocate_floating_ip(self):
"""Allocate a floating IP"""
result = True
cidr = '10.2.3.0/24'
floatingIpCidr = self.inputs.fip_pool
pool_name = 'pool1'
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
vpc_fixture = self.vpc1_fixture
assert vpc_fixture.verify_on_setup(), " VPC %s verification failed" % (
cidr)
self.logger.info(
'Adding rules to default SG of %s to reach public vm' %
(vpc_fixture.vpc_id))
default_sg_name = 'default'
rule1 = {
'protocol': 'icmp',
'direction': 'ingress',
'cidr': floatingIpCidr,
}
rule2 = {
'protocol': 'icmp',
'direction': 'egress',
'cidr': floatingIpCidr,
}
default_sg_id = vpc_fixture.get_security_group_id(default_sg_name)
if not (self.createSgRule(vpc_fixture, default_sg_id, rule1)
and self.createSgRule(vpc_fixture, default_sg_id, rule2)):
self.logger.error('Unable to create allow in SG %s ' %
(default_sg_name))
result = result and False
# create public VN for floating ip pool
ec2_base = EC2Base(logger=self.inputs.logger,
inputs=self.admin_inputs,
tenant=self.inputs.project_name)
public_vn_fixture = self.public_vn_obj.public_vn_fixture
assert public_vn_fixture.verify_on_setup(),\
"Public VN Fixture verification failed, Check logs"
# Assign floating IP. Internet GW is just dummy
ec2_base = EC2Base(logger=self.inputs.logger,
inputs=self.inputs,
tenant=vpc_fixture.vpc_id)
vpc_fip_fixture = self.useFixture(
VPCFIPFixture(public_vn_obj=self.public_vn_obj,
connections=self.connections,
ec2_base=ec2_base))
assert vpc_fip_fixture.verify_on_setup(
), "FIP pool verification failed, Pls check logs"
# Add rules in public VM's SG to reach the private VM"
self.set_sec_group_for_allow_all(self.inputs.stack_tenant, 'default')
fip_vm_fixture = self.useFixture(
VMFixture(connections=self.admin_connections,
vn_obj=public_vn_fixture.obj,
vm_name='fip_vm1'))
assert fip_vm_fixture.verify_on_setup(
), "VM verification in FIP VN failed"
assert fip_vm_fixture.wait_till_vm_is_up(),\
"VM verification in FIP VN failed"
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
assert self.vpc1_vn1_vm1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
vm1_fixture = self.vpc1_vn1_vm1_fixture
assert vm1_fixture.verify_on_setup(), "VPCVMFixture verification failed " \
"for VM %s" % (vm1_fixture.instance_id)
assert vm1_fixture.wait_till_vm_is_up(),\
"VM verification failed"
(fip, fip_alloc_id) = vpc_fip_fixture.create_and_assoc_fip(
vm1_fixture.instance_id)
if fip is None or fip_alloc_id is None:
self.logger.error('FIP creation and/or association failed! ')
result = result and False
if result:
self.addCleanup(vpc_fip_fixture.disassoc_and_delete_fip,
fip_alloc_id, fip)
assert vpc_fip_fixture.verify_fip(
fip), " FIP %s, %s verification failed" % (fip, fip_alloc_id)
assert vm1_fixture.c_vm_fixture.ping_with_certainty(
fip_vm_fixture.vm_ip), "Ping from FIP IP failed"
assert fip_vm_fixture.ping_with_certainty(
fip), "Ping to FIP IP failed"
return result
def test_sg_tcp_udp(self):
'''
Validate TCP File transfer between VMs by creating rules in a SG
'''
result = True
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
cidr = self.vpc1_cidr
sg1_name = 'sg1'
rule1 = {
'protocol': 'tcp',
'direction': 'ingress',
'cidr': cidr,
'port': '0-65535'
}
rule2 = {
'protocol': 'icmp',
'direction': 'ingress',
'cidr': cidr,
}
rule3 = {
'protocol': 'udp',
'direction': 'ingress',
'cidr': cidr,
'port': '0-65535'
}
default_sg_name = 'default'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
self.vpc1_vn1_vm2_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
assert self.vpc1_vn1_vm2_fixture.verify_on_setup()
self.vpc1_vn1_vm2_fixture.c_vm_fixture.wait_till_vm_is_up()
vpc_fixture = self.vpc1_fixture
default_sg_id = vpc_fixture.get_security_group_id(default_sg_name)
vpc_vn_fixture = self.vpc1_vn1_fixture
vm2_fixture = self.vpc1_vn1_vm2_fixture
sg1_id = self.createSecurityGroup(vpc_fixture, sg1_name)
if not (sg1_id):
self.logger.error('Creation of SG %s failed' % (sg1_name))
result = result and False
else:
self.addCleanup(vpc_fixture.delete_security_group, sg1_id)
# create rule-1 and rule-2 in SG
self.logger.info('Test create new rules')
if not (self.createSgRule(vpc_fixture, sg1_id, rule1)
and self.createSgRule(vpc_fixture, sg1_id, rule2)):
self.logger.error('Unable to create rule1/rule2 in SG %s ' %
(sg1_id))
result = result and False
vm1_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu-traffic',
connections=self.connections,
sg_ids=[sg1_name]))
assert vm1_fixture.verify_on_setup(
), "VPC1 VM fixture verification failed, check logs"
vm3_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
assert vm3_fixture.verify_on_setup(
), "VPC1 VM3 fixture verification failed, check logs"
vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
vm3_fixture.c_vm_fixture.wait_till_vm_is_up()
vm1_fixture.c_vm_fixture.put_pub_key_to_vm()
vm3_fixture.c_vm_fixture.put_pub_key_to_vm()
if not vm2_fixture.c_vm_fixture.ping_with_certainty(
vm1_fixture.c_vm_fixture.vm_ip):
self.logger.error("With SG rule to allow ping, ping failed!")
result = result and False
if not vm1_fixture.c_vm_fixture.ping_with_certainty(
vm2_fixture.c_vm_fixture.vm_ip, expectation=False):
self.logger.error("With SG rule to deny ping, ping passed!")
result = result and False
transfer_result = vm3_fixture.c_vm_fixture.check_file_transfer(
dest_vm_fixture=vm1_fixture.c_vm_fixture,
mode='scp',
size=str(random.randint(100, 1000000)))
if not transfer_result:
self.logger.error('File transfer step failed. Pls check logs')
result = result and False
# Validate tftp transfer fails without a rule
transfer_result = vm1_fixture.c_vm_fixture.check_file_transfer(
dest_vm_fixture=vm3_fixture.c_vm_fixture,
mode='tftp',
size=str(random.randint(100, 1000000)))
if transfer_result:
self.logger.error(
'File transfer step passed, expected it to fail. Pls check logs'
)
result = result and False
self.logger.info(
'Deleting the SG rule which allowed TCP and validate if transfer fails'
)
self.deleteSgRule(vpc_fixture, sg1_id, rule1)
transfer_result = vm3_fixture.c_vm_fixture.check_file_transfer(
dest_vm_fixture=vm1_fixture.c_vm_fixture,
mode='scp',
size=str(random.randint(100, 1000000)))
if transfer_result:
self.logger.error(
'File transfer step passed which should have failed!!. Pls check logs'
)
result = result and False
self.logger.info(
'Adding an SG rule to allow UDP and validate that transfer passes')
self.createSgRule(vpc_fixture, sg1_id, rule3)
self.createSgRule(vpc_fixture, default_sg_id, rule3)
transfer_result = vm3_fixture.c_vm_fixture.check_file_transfer(
dest_vm_fixture=vm1_fixture.c_vm_fixture,
mode='tftp',
size=str(random.randint(100, 1000000)))
if not transfer_result:
self.logger.error('File transfer step failed. Pls check logs')
result = result and False
def test_route_using_nat_instance(self):
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
vpc1_fixture = self.vpc1_fixture
vpc1_id = vpc1_fixture.vpc_id
public_vn_subnet = self.inputs.fip_pool
public_ip_to_ping = '8.8.8.8'
public_vn_rt = self.inputs.mx_rt
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
vpc1_vn1_fixture = self.vpc1_vn1_fixture
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
assert self.vpc1_vn1_vm1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
self.vpc1_vn1_vm2_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
assert self.vpc1_vn1_vm2_fixture.verify_on_setup()
self.vpc1_vn1_vm2_fixture.c_vm_fixture.wait_till_vm_is_up()
vm1_fixture = self.vpc1_vn1_vm1_fixture
result = True
# Just Read the existing vpc as a fixture
vpc1_contrail_fixture = self.useFixture(
ProjectFixture(project_name=vpc1_id,
username=self.admin_inputs.stack_user,
password=self.admin_inputs.stack_password,
connections=self.connections))
vpc1_contrail_fixture.get_project_connections()
public_vn_fixture = self.useFixture(
VNFixture(project_name=vpc1_id,
connections=vpc1_contrail_fixture.project_connections,
inputs=self.inputs,
vn_name='public',
subnets=[public_vn_subnet],
rt_number=public_vn_rt))
assert public_vn_fixture.verify_on_setup(),\
"Public VN Fixture verification failed, Check logs"
nat_instance_fixture = self.useFixture(
VPCVMFixture(
vpc1_vn1_fixture,
image_name='nat-service',
connections=vpc1_contrail_fixture.project_connections,
instance_type='nat',
public_vn_fixture=public_vn_fixture,
))
# Create Route table
rtb_id = vpc1_fixture.create_route_table()
self.addCleanup(vpc1_fixture.delete_route_table, rtb_id)
assert vpc1_fixture.verify_route_table(rtb_id),\
"Verification of Routetable %s failed!" % (rtb_id)
# Associate route table with subnet
subnet_id = vpc1_vn1_fixture.subnet_id
assoc_id = vpc1_fixture.associate_route_table(rtb_id, subnet_id)
if not assoc_id:
self.logger.error('Association of Subnet %s with RTB %s failed' \
%(subnet_id, rtb_id))
return False
# end if
self.addCleanup(vpc1_fixture.disassociate_route_table, assoc_id)
# Add route
prefix = '0.0.0.0/0'
c_result = vpc1_fixture.create_route(prefix, rtb_id,
nat_instance_fixture.instance_id)
if not c_result:
self.logger.error('Unable to create default route in RTB %s with \
instance %s ' % (rtb_id, vm1_fixture.instance_id))
return False
self.addCleanup(vpc1_fixture.delete_route, rtb_id, prefix)
# Check if route is installed in agent
c_vm1_fixture = vm1_fixture.c_vm_fixture
vm1_node_ip = c_vm1_fixture.vm_node_ip
agent_path = self.agent_inspect_h[vm1_node_ip].get_vna_active_route(
vrf_id=c_vm1_fixture.agent_vrf_id[c_vm1_fixture.vn_fq_name],
ip=prefix.split('/')[0],
prefix=prefix.split('/')[1])
if not agent_path:
self.logger.error('Route %s added is not seen in agent!' %
(prefix))
result = result and False
if not c_vm1_fixture.ping_with_certainty(public_ip_to_ping,
expectation=True):
self.logger.error('Ping to Public IP %s failed!' %
(public_ip_to_ping))
result = result and False
return result
def test_sg_inside_group(self):
'''
Validate that SG rules to allow traffic within an SG
Have VMs vm1,vm2,vm3 and SGs SG1, SG2.
SG1 to allow traffic from SG1 only (VM1)
SG2 to allow traffic from SG1,SG3 (VM2)
SG3 to allow traffic from SG3 only (VM3)
VM1<->VM3 ping should fail
VM3<->VM2 ping should pass
VM1<->VM2 ping should pass
'''
result = True
sg1_name = 'sg1'
sg2_name = 'sg2'
sg3_name = 'sg3'
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
vpc_fixture = self.vpc1_fixture
vpc_vn_fixture = self.vpc1_vn1_fixture
sg1_id = self.createSecurityGroup(vpc_fixture, sg1_name)
sg2_id = self.createSecurityGroup(vpc_fixture, sg2_name)
sg3_id = self.createSecurityGroup(vpc_fixture, sg3_name)
cidr = self.vpc1_cidr
sg1_rule1 = {
'protocol': 'icmp',
'direction': 'ingress',
'source-group': sg1_id
}
sg2_rule1 = {
'protocol': 'icmp',
'direction': 'ingress',
'source-group': sg1_id
}
sg2_rule2 = {
'protocol': 'icmp',
'direction': 'ingress',
'source-group': sg3_id
}
sg3_rule1 = {
'protocol': 'icmp',
'direction': 'ingress',
'source-group': sg3_id
}
if not sg1_id or not sg2_id or not sg3_id:
self.logger.error('Creation of SG %s/%s/%s failed' %
(sg1_name, sg2_name, sg3_name))
result = result and False
else:
self.addCleanup(vpc_fixture.delete_security_group, sg1_id)
self.addCleanup(vpc_fixture.delete_security_group, sg2_id)
self.addCleanup(vpc_fixture.delete_security_group, sg3_id)
self.createSgRule(vpc_fixture, sg1_id, sg1_rule1)
self.createSgRule(vpc_fixture, sg2_id, sg2_rule1)
self.createSgRule(vpc_fixture, sg2_id, sg2_rule2)
self.createSgRule(vpc_fixture, sg3_id, sg3_rule1)
# Create VMs using the SGs
vm1_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu',
connections=self.connections,
sg_ids=[sg1_name]))
assert vm1_fixture.verify_on_setup(
), "VPC VM1 fixture verification failed, check logs"
vm2_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu',
connections=self.connections,
sg_ids=[sg2_name]))
assert vm2_fixture.verify_on_setup(
), "VPC VM2 fixture verification failed, check logs"
vm3_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu',
connections=self.connections,
sg_ids=[sg3_name]))
assert vm3_fixture.verify_on_setup(
), "VPC1 VM3 fixture verification failed, check logs"
vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
vm2_fixture.c_vm_fixture.wait_till_vm_is_up()
vm3_fixture.c_vm_fixture.wait_till_vm_is_up()
# Ping between VM1 and VM3 should fail
if not vm1_fixture.c_vm_fixture.ping_with_certainty(
vm3_fixture.c_vm_fixture.vm_ip, expectation=False):
self.logger.error(
'SG rule should have disallowed ping between Vm1,Vm3')
result = result and False
# ping between Vm3 and VM2 should pass
if not vm3_fixture.c_vm_fixture.ping_with_certainty(
vm2_fixture.c_vm_fixture.vm_ip):
self.logger.error(
"SG rule should have allowed ping between Vm2,Vm3")
result = result and False
# ping between Vm1 and VM2 should pass
if not vm1_fixture.c_vm_fixture.ping_with_certainty(
vm2_fixture.c_vm_fixture.vm_ip):
self.logger.error(
"SG rule should have allowed ping between Vm1,Vm2")
result = result and False
return result
def test_security_group(self):
"""Create Security Groups, Add and Delete Rules """
result = True
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
cidr = self.vpc1_cidr
sg_name = 'sg1'
default_sg_name = 'default'
rule1 = {
'protocol': 'icmp',
'direction': 'egress',
'cidr': cidr,
}
rule2 = {
'protocol': 'tcp',
'direction': 'ingress',
'cidr': cidr,
'port': '100-200'
}
rule3 = {
'protocol': 'icmp',
'direction': 'ingress',
'cidr': cidr,
}
rule4 = {
'protocol': 'icmp',
'direction': 'egress',
'cidr': cidr,
}
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
vpc_fixture = self.vpc1_fixture
vpc_vn_fixture = self.vpc1_vn1_fixture
sg_id = self.createSecurityGroup(vpc_fixture, sg_name)
if not (sg_id): # and self.verifySecurityGroup()):
self.logger.error('Creation of SG %s failed' % (sg_name))
result = result and False
else:
self.addCleanup(self.deleteSecurityGroup, vpc_fixture, sg_id)
default_sg_id = vpc_fixture.get_security_group_id(default_sg_name)
# create rule-1 and rule-2 in SG
self.logger.info('Test create new rules')
if not (self.createSgRule(vpc_fixture, sg_id, rule1)
and self.createSgRule(vpc_fixture, sg_id, rule2)):
self.logger.error('Unable to create rule1 and rule2 in SG %s ' %
(sg_id))
result = result and False
# test create existing rule
self.logger.info('Test create existing rule')
if self.createSgRule(vpc_fixture, sg_id, rule1):
self.logger.error('Able to create an existing rule in SG %s' %
(sg_id))
result = result and False
else:
self.logger.info(
'Unable to create an already existing rule rule1..OK')
# Create egress rule on default SG so that ping packets can reach vm in
# sg1
self.logger.info(
'Adding egress rule on default SG so that ping packets can reach vm in sg1'
)
if not (self.createSgRule(vpc_fixture, default_sg_id, rule4)):
self.logger.error('Unable to create rule4 in SG %s ' %
(default_sg_name))
result = result and False
vm1_fixture = self.useFixture(
VPCVMFixture(vpc_vn_fixture,
image_name='ubuntu',
connections=self.connections,
sg_ids=[sg_name]))
assert vm1_fixture.verify_on_setup(
), "VPC1 VM fixture verification failed, check logs"
vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
self.vpc1_vn1_vm2_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
assert self.vpc1_vn1_vm2_fixture.verify_on_setup()
self.vpc1_vn1_vm2_fixture.c_vm_fixture.wait_till_vm_is_up()
vm2_fixture = self.vpc1_vn1_vm2_fixture
# Without a rule for icmp, SG should drop ping packets
if not vm2_fixture.c_vm_fixture.ping_with_certainty(
vm1_fixture.c_vm_fixture.vm_ip, expectation=False):
self.logger.error("With no SG rule to allow ping, ping passed!")
result = result and False
self.createSgRule(vpc_fixture, sg_id, rule3)
time.sleep(5)
# With a rule for icmp, SG should pass ping packets
if not vm2_fixture.c_vm_fixture.ping_with_certainty(
vm1_fixture.c_vm_fixture.vm_ip):
self.logger.error("With SG rule to allow ping, ping failed!")
result = result and False
# test delete existing rules
self.logger.info('Test delete existing rule')
if not (self.deleteSgRule(vpc_fixture, sg_id, rule1)
and self.deleteSgRule(vpc_fixture, sg_id, rule2)):
self.logger.error(
'Unable to delete rules rule1 and rule2 in SG %s' % (sg_id))
result = result and False
else:
self.logger.info('Deleted rules rule1 and rul2 in SG %s' % (sg_id))
# test delete non-existing rule
self.logger.info('Test delete non-existing rule')
if self.deleteSgRule(vpc_fixture, sg_id, rule2):
self.logger.error(
'Got success while deleting a non-existing rule rule2 in SG %s'
% (sg_id))
result = result and False
else:
self.logger.info(
'Unable to delete a non-existing rule rule2 in SG %s' %
(sg_id))
return result
def test_acl_with_association(self):
"""Create ACL, associate it with a subnet, add and replace rules """
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
cidr = self.vpc1_vn1_cidr
rule1 = {
'number': '100',
'protocol': 'tcp',
'direction': 'egress',
'action': 'pass',
'cidr': cidr,
'fromPort': '100',
'toPort': '200'
}
rule2 = {
'number': '200',
'protocol': 'udp',
'direction': 'ingress',
'action': 'deny',
'cidr': cidr,
'fromPort': '100',
'toPort': '200'
}
rule3 = {
'number': '100',
'protocol': 'tcp',
'direction': 'egress',
'action': 'pass',
'cidr': cidr,
'fromPort': '1000',
'toPort': '2000'
}
rule4 = {
'number': '101',
'protocol': 'tcp',
'direction': 'egress',
'action': 'pass',
'cidr': cidr,
'fromPort': '1000',
'toPort': '2000'
}
rule5 = {
'number': '99',
'protocol': 'icmp',
'direction': 'egress',
'action': 'deny',
'cidr': cidr,
}
rule6 = {
'number': '98',
'protocol': 'icmp',
'direction': 'egress',
'action': 'pass',
'cidr': cidr,
}
result = True
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
assert self.vpc1_vn1_vm1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
self.vpc1_vn1_vm2_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu-traffic',
connections=self.connections))
assert self.vpc1_vn1_vm2_fixture.verify_on_setup()
self.vpc1_vn1_vm2_fixture.c_vm_fixture.wait_till_vm_is_up()
vpc_fixture = self.vpc1_fixture
vpc_vn_fixture = self.vpc1_vn1_fixture
vm1_fixture = self.vpc1_vn1_vm1_fixture
vm2_fixture = self.vpc1_vn1_vm2_fixture
acl_id = self.createAcl(vpc_fixture)
if not (acl_id):
self.logger.error('ACL %s not seen ' % (acl_id))
return False
if not (vpc_vn_fixture.associate_acl(acl_id) and
(vpc_vn_fixture.verify_acl_binding(acl_id))):
self.logger.error('ACL %s association with Subnet %s failed' %
(acl_id, vpc_vn_fixture.subnet_id))
result = result and False
# create rule-1 and rule-2 in acl
self.logger.info('Test create new rules')
if not (self.createAclRule(vpc_fixture, acl_id, rule1)
and self.createAclRule(vpc_fixture, acl_id, rule2)):
self.logger.error('Creation of rules rule-1 and/or rule-2 failed')
result = result and False
self.logger.info('Test replace existing rules')
# replace existing rule-1 with rule-3
if not self.replaceAclRule(vpc_fixture, acl_id, rule3):
self.logger.error('Replacing rule1 with rule3 failed')
result = result and False
self.logger.info('Test replace non-existing rules')
# test replaceing not exixting rule-4
if self.replaceAclRule(vpc_fixture, acl_id, rule4):
self.logger.error('Replacing non-existant rule rule4 passed!')
result = result and False
self.logger.info('Test delete existing rules')
# delete existing rule-3 and rule-2
if not (self.deleteAclRule(vpc_fixture, acl_id, rule3)
and self.deleteAclRule(vpc_fixture, acl_id, rule2)):
self.logger.error('Deletion of rule2 and/or rule3 failed')
result = result and False
self.logger.info('Test delete non-existing rules')
# test deleting non-exixting rule-4
if self.deleteAclRule(vpc_fixture, acl_id, rule4):
self.logger.error('Deletion of non-existant rule rule4 passed!')
result = result and False
# Test traffic now with deny on icmp and with allow on icmp
self.createAclRule(vpc_fixture, acl_id, rule5)
assert vm1_fixture.c_vm_fixture.ping_with_certainty(
vm2_fixture.c_vm_fixture.vm_ip, expectation=False), \
"With rule to deny ping, ping passed!"
self.createAclRule(vpc_fixture, acl_id, rule6)
assert vm1_fixture.c_vm_fixture.ping_with_certainty(
vm2_fixture.c_vm_fixture.vm_ip), \
"With rule to allow ping, ping failed!"
if not (vpc_vn_fixture.associate_acl('acl-default')
and vpc_vn_fixture.verify_acl_binding('acl-default')):
self.logger.error('Unable to associate acl-default to subnet %s' %
(vpc_vn_fixture.subnet_id))
result = result and False
# Cleanup
vpc_fixture.delete_acl(acl_id)
return result
def test_route_using_gateway(self):
self.vpc1_cidr = '10.2.5.0/24'
self.vpc1_vn1_cidr = '10.2.5.0/25'
self.vpc1_vn2_cidr = '10.2.5.128/25'
self.vpc2_cidr = '10.2.50.0/24'
self.vpc2_vn1_cidr = '10.2.50.0/25'
self.vpc1_fixture = self.useFixture(
VPCFixture(self.vpc1_cidr, connections=self.connections))
assert self.vpc1_fixture.verify_on_setup()
vpc1_fixture = self.vpc1_fixture
vpc1_id = vpc1_fixture.vpc_id
public_vn_subnet = self.inputs.fip_pool
public_ip_to_ping = '8.8.8.8'
public_vn_rt = self.inputs.mx_rt
self.vpc1_vn1_fixture = self.useFixture(
VPCVNFixture(self.vpc1_fixture,
subnet_cidr=self.vpc1_vn1_cidr,
connections=self.connections))
assert self.vpc1_vn1_fixture.verify_on_setup()
vpc1_vn1_fixture = self.vpc1_vn1_fixture
self.vpc1_vn1_vm1_fixture = self.useFixture(
VPCVMFixture(self.vpc1_vn1_fixture,
image_name='ubuntu',
connections=self.connections))
assert self.vpc1_vn1_vm1_fixture.verify_on_setup()
self.vpc1_vn1_vm1_fixture.c_vm_fixture.wait_till_vm_is_up()
vm1_fixture = self.vpc1_vn1_vm1_fixture
result = True
# Just Read the existing vpc as a fixture
vpc1_contrail_fixture = self.useFixture(
ProjectFixture(project_name=vpc1_id,
username=self.admin_inputs.stack_user,
password=self.admin_inputs.stack_password,
connections=self.connections))
vpc1_contrail_fixture.get_project_connections()
public_vn_fixture = self.public_vn_obj.public_vn_fixture
assert public_vn_fixture.verify_on_setup(),\
"Public VN Fixture verification failed, Check logs"
# Assign floating IP. Internet GW is just dummy
ec2_base = EC2Base(logger=self.inputs.logger,
inputs=self.inputs,
tenant=vpc1_id)
vpc_fip_fixture = self.useFixture(
VPCFIPFixture(public_vn_obj=self.public_vn_obj,
connections=self.connections,
ec2_base=ec2_base))
assert vpc_fip_fixture.verify_on_setup(
), "FIP pool verification failed, Pls check logs"
(fip, fip_alloc_id) = vpc_fip_fixture.create_and_assoc_fip(
vm1_fixture.instance_id)
if fip is None or fip_alloc_id is None:
self.logger.error('FIP creation and/or association failed! ')
result = result and False
if result:
self.addCleanup(vpc_fip_fixture.disassoc_and_delete_fip,
fip_alloc_id, fip)
# Create Internet gateway
gw_id = vpc1_fixture.create_gateway()
self.addCleanup(vpc1_fixture.delete_gateway, gw_id)
# Create Route table
rtb_id = vpc1_fixture.create_route_table()
self.addCleanup(vpc1_fixture.delete_route_table, rtb_id)
assert vpc1_fixture.verify_route_table(rtb_id),\
"Verification of Routetable %s failed!" % (rtb_id)
# Associate route table with subnet
subnet_id = vpc1_vn1_fixture.subnet_id
assoc_id = vpc1_fixture.associate_route_table(rtb_id, subnet_id)
if not assoc_id:
self.logger.error('Association of Subnet %s with RTB %s failed' %
(subnet_id, rtb_id))
return False
# end if
self.addCleanup(vpc1_fixture.disassociate_route_table, assoc_id)
# Add route
prefix = '0.0.0.0/0'
c_result = vpc1_fixture.create_route(prefix, rtb_id, gw_id=gw_id)
if not c_result:
self.logger.error('Unable to create default route in RTB %s with \
gateway %s ' % (rtb_id, gw_id))
return False
self.addCleanup(vpc1_fixture.delete_route, rtb_id, prefix)
# No need to check if this route is installed in agent
c_vm1_fixture = vm1_fixture.c_vm_fixture
if not c_vm1_fixture.ping_with_certainty(public_ip_to_ping,
expectation=True):
self.logger.error('Ping to Public IP %s failed!' %
(public_ip_to_ping))
result = result and False
return result
