def test_service_state(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'graphql', 'graphql', True, 'v2-beta', logger.level) self.assertEqual('active', rancher.get_service_state())
def test_stack_doesnt_exists(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'ranchertool', '', True, 'v2-beta', logger.level) self.assertEqual(False, rancher.stack_exists(), "Something weird happened.")
def test_stack_exists(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'graphql', '', True, 'v2-beta', logger.level) self.assertEqual(True, rancher.stack_exists(), "Didn't find a stack that should exist.")
def test_get_launchconfig(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'graphql', 'graphql', True, 'v2-beta', logger.level) self.assertIsNotNone(rancher.get_launch_config(), "Failed to get launch config")
def test_create_stack(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'aaa-test-stack', 'aaa-test-service', True, 'v2-beta', logger.level) self.assertEqual(True, rancher.create_stack(), "Something weird happened.") self.assertEqual(True, rancher.stack_exists(), "Didn't find a stack that should exist.")
def test_labels_processing(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'ranchertool', '', True, 'v2-beta', logger.level) labels_str = 'label1=value1,label2=value2' labels_tup = [('label3', 'value3'), ('label4', 'This is my label'), ('label5', '"this is a test";')] rancher.set_labels(labels_str) rancher.set_labels(labels_tup) self.assertEqual( True, rancher.get_labels() is not None, "There should be labels defined on the Rancher object.") self.assertEqual( 5, len(rancher.get_labels()), "There should be 2 labels defined on the Rancher object.") self.assertEqual('value1', rancher.get_labels()['label1'], "The first label should have value of value1") self.assertEqual('value3', rancher.get_labels()['label3'], "Label 3 added via tuple is wrong.")
def test_service_links_processing(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'ranchertool', '', True, 'v2-beta', logger.level) service_links = 'kafka1=kafka/kafka1,kafka2=kafka/kafka2'
def test_string_variables_processing(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'ranchertool', '', True, 'v2-beta', logger.level) variables_str = 'var1=val1|var2=val2 val3|var3="this is a test";' rancher.set_variables(variables_str) self.assertEqual(True, rancher.get_variables() is not None, 'There should be some variables') self.assertEqual(rancher.get_variables()['var1'], 'val1', 'variable 1 should have value of val1') self.assertEqual(rancher.get_variables()['var2'], 'val2 val3', "Variable 2 doesn't have the correct value.") self.assertEqual(rancher.get_variables()['var3'], '"this is a test";', "Variable 3 doesn't have the correct value.")
def test_tuple_variables_processing(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'ranchertool', '', True, 'v2-beta', logger.level) variables_tup = [('var4', 'val4'), ('var5', 'val5 val6'), ('var6', '"this is a test";')] rancher.set_variables(variables_tup) self.assertEqual(True, rancher.get_variables() is not None, 'There should be some variables') self.assertEqual(rancher.get_variables()['var4'], 'val4', 'variable 4 should have value of val4') self.assertEqual(rancher.get_variables()['var5'], 'val5 val6', "Variable 5 doesn't have the correct value.") self.assertEqual(rancher.get_variables()['var6'], '"this is a test";', "Variable 6 doesn't have the correct value.")
def test_do_upgrade(self): import os from helpers import RancherConnection rancher = RancherConnection('https://rancher.dev.cu.edu', os.environ.get('RANCHER_ACCESS_KEY'), os.environ.get('RANCHER_SECRET_KEY'), 'ODIN-DEV', 'graphql', 'sarsen-graphql', True, 'v2-beta', LogLevel.TRACE) upgrade = { 'inServiceStrategy': { 'batchSize': 1, 'intervalMillis': 2 * 1000, # rancher expects milliseconds 'startFirst': False, 'startOnCreate': False, 'launchConfig': {}, 'secondaryLaunchConfigs': [] } } upgrade['inServiceStrategy'][ 'launchConfig'] = rancher.get_launch_config() rancher.do_upgrade(upgrade) self.assertEqual(True, rancher.finish_upgrade())