From 401ccb0b304d925faa59c02e02d4701dbfbf9c1c Mon Sep 17 00:00:00 2001 From: Alan Malta Rodrigues Date: Tue, 5 Oct 2021 15:30:57 +0200 Subject: [PATCH] unit tests --- .../python/WMCore_t/BossAir_t/BasePlugin_t.py | 25 ++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) diff --git a/test/python/WMCore_t/BossAir_t/BasePlugin_t.py b/test/python/WMCore_t/BossAir_t/BasePlugin_t.py index 1de73a1de6..4a90cadf19 100644 --- a/test/python/WMCore_t/BossAir_t/BasePlugin_t.py +++ b/test/python/WMCore_t/BossAir_t/BasePlugin_t.py @@ -12,7 +12,7 @@ from WMCore_t.BossAir_t.BossAir_t import BossAirTest -from WMCore.BossAir.Plugins.BasePlugin import BasePlugin +from WMCore.BossAir.Plugins.BasePlugin import BasePlugin, BossAirPluginException class BasePluginTest(BossAirTest): @@ -40,6 +40,29 @@ def testScramArchToOS(self): return + def testScramArchtoRequiredArch(self): + """ + Test mapping of ScramArch to a given architecture + """ + bp = BasePlugin(config=None) + + self.assertEqual(bp.scramArchtoRequiredArch('slc5_amd64_gcc481'), 'X86_64') + self.assertEqual(bp.scramArchtoRequiredArch('slc6_amd64_gcc630'), 'X86_64') + self.assertEqual(bp.scramArchtoRequiredArch('slc7_amd64_gcc10'), 'X86_64') + self.assertEqual(bp.scramArchtoRequiredArch('slc7_aarch64_gcc700'), 'aarch64') + self.assertEqual(bp.scramArchtoRequiredArch('slc7_ppc64le_gcc9'), 'ppc64le') + self.assertIsNone(bp.scramArchtoRequiredArch(None)) + self.assertIsNone(bp.scramArchtoRequiredArch(None)) + with self.assertRaises(BossAirPluginException): + bp.scramArchtoRequiredArch("slc7_BLAH_gcc700") + + self.assertEqual(bp.scramArchtoRequiredArch(['slc5_amd64_gcc481', 'slc6_amd64_gcc630']), 'X86_64') + self.assertEqual(bp.scramArchtoRequiredArch(['slc7_amd64_gcc10', 'slc7_aarch64_gcc700']), 'X86_64') + self.assertEqual(bp.scramArchtoRequiredArch( + ['slc7_amd64_gcc10', 'slc7_aarch64_gcc700', 'slc7_ppc64le_gcc9']), 'X86_64') + self.assertEqual(bp.scramArchtoRequiredArch(['slc7_aarch64_gcc700', 'slc7_ppc64le_gcc9']), 'ppc64le') + + return if __name__ == '__main__': unittest.main()