unit tests

dmwm · Oct 11, 2021 · 401ccb0 · 401ccb0
1 parent 86bb82b
commit 401ccb0
Showing 1 changed file with 24 additions and 1 deletion.
diff --git 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()