def testScramArchParsing(self):
"""
Test the various modes of parsing for the scram arch
"""
try:
os.chdir(self.testDir)
with tempfile.NamedTemporaryFile() as tf:
tf.write('GLIDEIN_REQUIRED_OS = "rhel6" \n')
tf.write('Memory = 2048\n')
tf.flush()
with tmpEnv(_CONDOR_MACHINE_AD=tf.name):
self.assertEqual(getSingleScramArch('slc6_blah_blah'), 'slc6_blah_blah')
self.assertEqual(getSingleScramArch('slc5_blah_blah'), 'slc5_blah_blah')
self.assertEqual(getSingleScramArch(['slc6_blah_blah', 'slc7_blah_blah']),
'slc6_blah_blah')
self.assertEqual(getSingleScramArch(['slc6_blah_blah', 'slc5_blah_blah']),
'slc6_blah_blah')
self.assertEqual(getSingleScramArch(['slc7_blah_blah', 'slc8_blah_blah']), None)
with tempfile.NamedTemporaryFile() as tf:
tf.write('GLIDEIN_REQUIRED_OS = "rhel7" \n')
tf.write('Memory = 2048\n')
tf.flush()
with tmpEnv(_CONDOR_MACHINE_AD=tf.name):
self.assertEqual(getSingleScramArch('slc6_blah_blah'), 'slc6_blah_blah')
self.assertEqual(getSingleScramArch('slc7_blah_blah'), 'slc7_blah_blah')
self.assertEqual(getSingleScramArch(['slc6_blah_blah', 'slc7_blah_blah']),
'slc7_blah_blah')
self.assertEqual(getSingleScramArch(['slc6_blah_blah', 'slc5_blah_blah']), None)
self.assertEqual(getSingleScramArch(['slc7_blah_blah', 'slc8_blah_blah']),
'slc7_blah_blah')
except Exception:
raise
finally:
os.chdir(self.oldCwd)
return
def testSetEnv(self):
"""
Test the grouper function (returns chunk of an iterable)
"""
with tmpEnv(NEW_ENV_VAR=u'blah'):
self.assertTrue('NEW_ENV_VAR' in os.environ)
self.assertEquals(os.environ['NEW_ENV_VAR'], u'blah')
self.assertFalse('NEW_ENV_VAR' in os.environ)
def testSetEnv(self):
"""
Test the grouper function (returns chunk of an iterable)
"""
with tmpEnv(NEW_ENV_VAR=u'blah'):
self.assertTrue('NEW_ENV_VAR' in os.environ)
self.assertEqual(os.environ['NEW_ENV_VAR'], u'blah')
self.assertFalse('NEW_ENV_VAR' in os.environ)