def test_read_conductor_config_with_optional_configs(self): from ots.worker.conductor import conductor optional_value = "ps aux" optional_username = "******" conf_file = os.path.join(os.path.dirname(__file__), "conductor.conf") conf = conductor._read_configuration_files(conf_file, 0) self.assertTrue(type(conf) == type(dict())) self.assertTrue(conf["target_username"] != "") self.assertTrue(conf["device_packaging"] != "") self.assertTrue(conf["pre_test_info_commands_debian"] != "") self.assertTrue(conf["pre_test_info_commands_rpm"] != "") self.assertTrue(conf["pre_test_info_commands"] != "") self.assertTrue(conf["files_fetched_after_testing"] != "") self.assertTrue(conf["tmp_path"] != "") # Check that we do not have value in pre_test_info_commands that we will # insert from optional configuration file self.assertFalse(optional_value in conf["pre_test_info_commands"]) temp_folder = tempfile.mkdtemp("_optional_confs") temp_config = tempfile.mktemp(suffix=".conf", dir=temp_folder) fp = open(temp_config, "w") fp.write( "[conductor]\n" 'pre_test_info_commands: "%s"\n' "target_username: %s" % (optional_value, optional_username) ) fp.close() conf["custom_config_folder"] = temp_folder conf = conductor._read_optional_config_files(temp_folder, conf) self.assertTrue(optional_value in conf["pre_test_info_commands"]) self.assertTrue(conf["target_username"] == optional_username) os.unlink(temp_config) os.rmdir(temp_folder)
def test_read_conductor_config(self): from ots.worker.conductor import conductor conf_file = os.path.join(os.path.dirname(__file__), "conductor.conf") conf = conductor._read_configuration_files(conf_file, 0) self.assertTrue(type(conf) == type(dict())) self.assertTrue(conf['target_username'] != "") self.assertTrue(conf['device_packaging'] != "") self.assertTrue(conf['pre_test_info_commands_debian'] != "") self.assertTrue(conf['pre_test_info_commands_rpm'] != "") self.assertTrue(conf['pre_test_info_commands'] != "") self.assertTrue(conf['files_fetched_after_testing'] != "") self.assertTrue(conf['tmp_path'] != "")
def test_read_conductor_config(self): from ots.worker.conductor import conductor conf_file = os.path.join(os.path.dirname(__file__), "conductor.conf") conf = conductor._read_configuration_files(conf_file, 0) self.assertTrue(type(conf) == type(dict())) self.assertTrue(conf["target_username"] != "") self.assertTrue(conf["device_packaging"] != "") self.assertTrue(conf["pre_test_info_commands_debian"] != "") self.assertTrue(conf["pre_test_info_commands_rpm"] != "") self.assertTrue(conf["pre_test_info_commands"] != "") self.assertTrue(conf["files_fetched_after_testing"] != "") self.assertTrue(conf["tmp_path"] != "")
def test_read_conductor_config_with_optional_configs(self): from ots.worker.conductor import conductor optional_value = "ps aux" optional_username = "******" conf_file = os.path.join(os.path.dirname(__file__), "conductor.conf") conf = conductor._read_configuration_files(conf_file, 0) self.assertTrue(type(conf) == type(dict())) self.assertTrue(conf['target_username'] != "") self.assertTrue(conf['device_packaging'] != "") self.assertTrue(conf['pre_test_info_commands_debian'] != "") self.assertTrue(conf['pre_test_info_commands_rpm'] != "") self.assertTrue(conf['pre_test_info_commands'] != "") self.assertTrue(conf['files_fetched_after_testing'] != "") self.assertTrue(conf['tmp_path'] != "") # Check that we do not have value in pre_test_info_commands that we will # insert from optional configuration file self.assertFalse(optional_value in conf['pre_test_info_commands']) temp_folder = tempfile.mkdtemp("_optional_confs") temp_config = tempfile.mktemp(suffix='.conf', dir=temp_folder) fp = open(temp_config, 'w') fp.write('[conductor]\n'\ 'pre_test_info_commands: "%s"\n'\ 'target_username: %s' % ( optional_value, optional_username)) fp.close() conf['custom_config_folder'] = temp_folder conf = conductor._read_optional_config_files(temp_folder, conf) self.assertTrue(optional_value in conf['pre_test_info_commands']) self.assertTrue(conf['target_username'] == optional_username) os.unlink(temp_config) os.rmdir(temp_folder)