class ClusterBasics(unittest.TestCase):
def setUp(self):
if options.config_file:
self.tester = Eucaops(config_file=options.config_file, password=options.clc_password)
else:
print "\tNeed to pass --config_file option. Try --help for more information\n"
exit(1)
## If specific image wants to be tested, use that; if not, use any instance-store backed image.
if options.image:
self.image = self.tester.get_emi(emi=options.image)
else:
self.image = self.tester.get_emi(root_device_type="instance-store")
self.keypair = self.tester.add_keypair(options.prefix + "-" + str(time.time()))
self.keypath = os.curdir + "/" + self.keypair.name + ".pem"
### Identify type of instance to run
if options.type == "random":
options.type = random.choice(["m1.small", "c1.medium", "m1.large", "m1.xlarge", "c1.xlarge"])
### Identify number of instances to run (i.e. number of security groups)
self.num_vms = self.tester.get_available_vms(options.type)
if self.num_vms >= options.number:
self.available = options.number
else:
options.type = "m1.small"
avail_vms = self.tester.get_available_vms(options.type)
if avail_vms < options.number:
self.tester.fail("Not enough m1.small vm types to run test with minimal of 3 security groups.")
else:
self.available = options.number
if options.print_debug:
self.tester.start_euca_logs()
self.security_groups = []
self.reservations = []
def tearDown(self):
### Clean up after running test case
for reservation in self.reservations:
self.tester.terminate_instances(reservation)
for security_group in security_groups:
self.tester.delete_group(security_group.name)
self.tester.delete_keypair(self.keypair)
os.remove(self.keypath)
self.keypath = None
self.keypair = None
self.image = None
self.num_vms = None
self.available = None
self.security_groups = None
if options.print_debug:
self.tester.stop_euca_logs()
self.tester.save_euca_logs()
self.tester = None
### Test Cases ###
def iptables_Cruft(self):
### Launch number of instances based upon number of security groups wanting to be tested.
### Take snapshot of iptables before creating security groups and launching instances.
### Use service manager to get to enabled CC to get iptables rules
partition = self.tester.service_manager.partitions.keys()
part = list(partition)[0]
main_part = self.tester.service_manager.partitions.get(part)
cc_machine = main_part.get_enabled_cc()
cc_shell = self.tester.create_ssh(hostname=cc_machine.hostname, password=options.cc_password)
pre_stdin, pre_iptables, pre_stderr = cc_shell.exec_command(
'iptables-save | grep -v "#" | grep -v "\:PRE" | grep -v "\:POST" | grep -v "\:INPUT" | grep -v "\:FORWARD" | grep -v "\:OUT"'
)
self.pre_iptables = list(pre_iptables)
self.assertTrue(pre_stderr, "pre_Iptables_Snapshot failed.")
### Create security group for number of security groups we want to test.
while self.available > 0:
### Create unique security group and authorize SSH and PING
sec_group = self.tester.add_group(group_name=options.prefix + "-" + str(time.time()))
self.assertNotEqual(len(sec_group.name), 0, "Could not create group.")
self.assertTrue(
self.tester.authorize_group_by_name(group_name=sec_group.name), "Could not authorize group for SSH"
)
self.assertTrue(
self.tester.authorize_group_by_name(group_name=sec_group.name, port=-1, protocol="icmp"),
"Could not authorize group for PING",
)
self.security_groups.append(sec_group)
### Launch instance for the unique security group
try:
reservation = self.tester.run_instance(
self.image, keypair=self.keypair.name, group=sec_group.name, type=options.type
)
except Exception, e:
self.fail("Caught an exception when running the instance: " + str(e))
self.reservations.append(reservation)
### Decrement count of security groups and instances left to create
self.available -= 1
### Loop through and terminate instances
### Grab total number of instances ran from by test case
### Terminate each instance
for reservation in self.reservations:
self.assertTrue(self.tester.terminate_instances(reservation), "Failure when terminating instance.")
### Loop through and delete security groups
for group in self.security_groups:
self.assertTrue(self.tester.delete_group(group), "Failure when deleting group " + group.name)
### Take snapshot of iptables after deleting security groups and terminating instances.
### Use service manager to get to enabled CC to get iptables rules
partition = self.tester.service_manager.partitions.keys()
part = list(partition)[0]
main_part = self.tester.service_manager.partitions.get(part)
cc_machine = main_part.get_enabled_cc()
cc_shell = self.tester.create_ssh(hostname=cc_machine.hostname, password=options.cc_password)
post_stdin, post_iptables, post_stderr = cc_shell.exec_command(
'iptables-save | grep -v "#" | grep -v "\:PRE" | grep -v "\:POST" | grep -v "\:INPUT" | grep -v "\:FORWARD" | grep -v "\:OUT"'
)
self.post_iptables = list(post_iptables)
self.assertTrue(post_stderr, "post_Iptables_Snapshot failed.")
### Evaluate pre and post iptables outputs to see if there is a difference.
if len(self.pre_iptables) != len(self.post_iptables):
## Get different lines and print them
iptables_diff = set(self.post_iptables) - set(self.pre_iptables)
pp = pprint.PrettyPrinter(indent=4)
self.tester.critical("\n======================================\n")
self.tester.critical("Diffences between iptables snapshots: ")
self.tester.critical("PRE-IPTABLES SNAPSHOT LENGTH: %i", len(self.pre_iptables))
self.tester.critical("POST-IPTABLES SNAPSHOT LENGTH: %i", len(self.post_iptables))
self.tester.critical("\n---------------------------------------\n")
pp.pprint(list(iptables_diff))
self.tester.critical("\n======================================\n")
class ClusterBasics(unittest.TestCase):
def setUp(self):
if options.config_file:
self.tester = Eucaops(config_file=options.config_file,
password=options.clc_password)
else:
print "\tNeed to pass --config_file option. Try --help for more information\n"
exit(1)
## If specific image wants to be tested, use that; if not, use any instance-store backed image.
if options.image:
self.image = self.tester.get_emi(emi=options.image)
else:
self.image = self.tester.get_emi(root_device_type="instance-store")
self.keypair = self.tester.add_keypair(options.prefix + "-" +
str(time.time()))
self.keypath = os.curdir + "/" + self.keypair.name + ".pem"
### Identify type of instance to run
if options.type == "random":
options.type = random.choice([
"m1.small", "c1.medium", "m1.large", "m1.xlarge", "c1.xlarge"
])
### Identify number of instances to run (i.e. number of security groups)
self.num_vms = self.tester.get_available_vms(options.type)
if self.num_vms >= options.number:
self.available = options.number
else:
options.type = "m1.small"
avail_vms = self.tester.get_available_vms(options.type)
if avail_vms < options.number:
self.tester.fail(
"Not enough m1.small vm types to run test with minimal of 3 security groups."
)
else:
self.available = options.number
self.security_groups = []
self.reservations = []
def tearDown(self):
### Clean up after running test case
for reservation in self.reservations:
self.tester.terminate_instances(reservation)
for security_group in self.security_groups:
self.tester.delete_group(security_group)
self.tester.delete_keypair(self.keypair)
os.remove(self.keypath)
self.keypath = None
self.keypair = None
self.image = None
self.num_vms = None
self.available = None
self.security_groups = None
self.tester = None
### Test Cases ###
def iptables_Cruft(self):
### Launch number of instances based upon number of security groups wanting to be tested.
### Take snapshot of iptables before creating security groups and launching instances.
### Use service manager to get to enabled CC to get iptables rules
partition = self.tester.service_manager.partitions.keys()
part = list(partition)[0]
main_part = self.tester.service_manager.partitions.get(part)
cc_machine = main_part.get_enabled_cc()
cc_shell = self.tester.create_ssh(hostname=cc_machine.hostname,
password=options.cc_password)
pre_stdin, pre_iptables, pre_stderr = cc_shell.exec_command(
"iptables-save | grep -v \"#\" | grep -v \"\:PRE\" | grep -v \"\:POST\" | grep -v \"\:INPUT\" | grep -v \"\:FORWARD\" | grep -v \"\:OUT\""
)
self.pre_iptables = list(pre_iptables)
self.assertTrue(pre_stderr, "pre_Iptables_Snapshot failed.")
### Create security group for number of security groups we want to test.
while self.available > 0:
### Create unique security group and authorize SSH and PING
sec_group = self.tester.add_group(group_name=options.prefix + "-" +
str(time.time()))
self.assertNotEqual(len(sec_group.name), 0,
"Could not create group.")
self.assertTrue(
self.tester.authorize_group_by_name(group_name=sec_group.name),
"Could not authorize group for SSH")
self.assertTrue(
self.tester.authorize_group_by_name(group_name=sec_group.name,
port=-1,
protocol="icmp"),
"Could not authorize group for PING")
self.security_groups.append(sec_group)
### Launch instance for the unique security group
try:
reservation = self.tester.run_instance(
self.image,
keypair=self.keypair.name,
group=sec_group.name,
type=options.type)
except Exception, e:
self.fail("Caught an exception when running the instance: " +
str(e))
self.reservations.append(reservation)
### Decrement count of security groups and instances left to create
self.available -= 1
### Take snapshot of iptables after deleting security groups and terminating instances.
### Use service manager to get to enabled CC to get iptables rules
partition = self.tester.service_manager.partitions.keys()
part = list(partition)[0]
main_part = self.tester.service_manager.partitions.get(part)
cc_machine = main_part.get_enabled_cc()
cc_shell = self.tester.create_ssh(hostname=cc_machine.hostname,
password=options.cc_password)
post_stdin, post_iptables, post_stderr = cc_shell.exec_command(
"iptables-save | grep -v \"#\" | grep -v \"\:PRE\" | grep -v \"\:POST\" | grep -v \"\:INPUT\" | grep -v \"\:FORWARD\" | grep -v \"\:OUT\""
)
self.post_iptables = list(post_iptables)
self.assertTrue(post_stderr, "post_Iptables_Snapshot failed.")
### Evaluate pre and post iptables outputs to see if there is a difference.
if (len(self.pre_iptables) != len(self.post_iptables)):
## Get different lines and print them
iptables_diff = set(self.post_iptables) - set(self.pre_iptables)
pp = pprint.PrettyPrinter(indent=4)
self.tester.critical("\n======================================\n")
self.tester.critical("Diffences between iptables snapshots: ")
self.tester.critical("PRE-IPTABLES SNAPSHOT LENGTH: %i",
len(self.pre_iptables))
self.tester.critical("POST-IPTABLES SNAPSHOT LENGTH: %i",
len(self.post_iptables))
self.tester.critical("\n---------------------------------------\n")
pp.pprint(list(iptables_diff))
self.tester.critical("\n======================================\n")
