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)
