def testServiceList(self): """ Test to make sure right services get returned """ config_file = get_test_config("slurm/check_ok.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception as e: self.fail("Received exception while parsing configuration: %s" % e) services = settings.enabled_services() expected_services = set(['condor-ce', 'globus-gridftp-server']) self.assertEqual( services, expected_services, "List of enabled services incorrect, " + "got %s but expected %s" % (services, expected_services)) config_file = get_test_config("slurm/slurm_disabled.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception as e: self.fail("Received exception while parsing configuration: %s" % e) services = settings.enabled_services() expected_services = set() self.assertEqual( services, expected_services, "List of enabled services incorrect, " + "got %s but expected %s" % (services, expected_services)) config_file = get_test_config("slurm/ignored.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception as e: self.fail("Received exception while parsing configuration: %s" % e) services = settings.enabled_services() expected_services = set() self.assertEqual( services, expected_services, "List of enabled services incorrect, " + "got %s but expected %s" % (services, expected_services))
def testParsing(self): """ Test configuration parsing """ config_file = get_test_config("slurm/slurm1.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception as e: self.fail("Received exception while parsing configuration: %s" % e) attributes = settings.get_attributes() options = { 'OSG_JOB_MANAGER_HOME': '/opt/slurm', 'OSG_PBS_LOCATION': '/opt/slurm', 'OSG_JOB_MANAGER': 'SLURM' } for option in options: value = options[option] self.assertTrue(attributes.has_key(option), "Attribute %s missing" % option) err_msg = "Wrong value obtained for %s, " \ "got %s instead of %s" % (option, attributes[option], value) self.assertEqual(attributes[option], value, err_msg)
def testValidSettings2(self): """ Test the check_attributes function to see if it works on valid settings """ config_file = get_test_config("slurm/check_ok2.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e)
def testMissingPBSLocation(self): """ Test the check_attributes function to see if it catches missing pbs location """ config_file = get_test_config("slurm/missing_location.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e)
def testParsingIgnored(self): """ Test PBS section parsing when set to Ignore """ config_file = get_test_config("slurm/ignored.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e)
def testServiceList(self): """ Test to make sure right services get returned """ config_file = get_test_config("slurm/check_ok.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e)
def testInvalidUtilityContact(self): """ Test the check_attributes function to see if it catches invalid utility contacts """ config_file = get_test_config("slurm/invalid_utility_contact.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e)
def testValidSettings2(self): """ Test the check_attributes function to see if it works on valid settings """ config_file = get_test_config("slurm/check_ok2.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception as e: self.fail("Received exception while parsing configuration: %s" % e) attributes = settings.get_attributes() self.assertTrue(settings.check_attributes(attributes), "Correct settings incorrectly flagged as invalid")
def testParsingIgnored(self): """ Test PBS section parsing when set to Ignore """ config_file = get_test_config("slurm/ignored.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception as e: self.fail("Received exception while parsing configuration: %s" % e) attributes = settings.get_attributes() self.assertEqual(len(attributes), 0, "Ignored configuration should have no attributes")
try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e) services = settings.enabled_services() expected_services = set(['condor-ce', 'globus-gridftp-server']) self.assertEqual( services, expected_services, "List of enabled services incorrect, " + "got %s but expected %s" % (services, expected_services)) config_file = get_test_config("slurm/seg_enabled.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file) settings = slurm.SlurmConfiguration(logger=global_logger) try: settings.parse_configuration(configuration) except Exception, e: self.fail("Received exception while parsing configuration: %s" % e) services = settings.enabled_services() expected_services = set(['condor-ce', 'globus-gridftp-server']) self.assertEqual( services, expected_services, "List of enabled services incorrect, " + "got %s but expected %s" % (services, expected_services)) config_file = get_test_config("slurm/slurm_disabled.ini") configuration = ConfigParser.SafeConfigParser() configuration.read(config_file)