6
|
1 import os
|
|
2 import unittest
|
|
3 from commons.core.sql.TableJobAdaptatorFactory import TableJobAdaptatorFactory
|
|
4 from commons.core.sql.DbFactory import DbFactory
|
|
5
|
|
6 class Test_TableJobAdaptatorFactory(unittest.TestCase):
|
|
7
|
|
8 def test_createInstance_SGE(self):
|
|
9 REPET_JOB_MANAGER_Initial_Value = os.environ["REPET_JOB_MANAGER"]
|
|
10 os.environ["REPET_JOB_MANAGER"] = "SGE"
|
|
11 instance = TableJobAdaptatorFactory.createInstance(DbFactory.createInstance(), "dummyJobTable")
|
|
12 obsClassName = instance.__class__.__name__
|
|
13 expClassName = "TableJobAdaptatorSGE"
|
|
14 os.environ["REPET_JOB_MANAGER"] = REPET_JOB_MANAGER_Initial_Value
|
|
15 self.assertEquals(expClassName, obsClassName)
|
|
16
|
|
17 def test_createInstance_Torque(self):
|
|
18 REPET_JOB_MANAGER_Initial_Value = os.environ["REPET_JOB_MANAGER"]
|
|
19 os.environ["REPET_JOB_MANAGER"] = "Torque"
|
|
20 instance = TableJobAdaptatorFactory.createInstance(DbFactory.createInstance(), "dummyJobTable")
|
|
21 obsClassName = instance.__class__.__name__
|
|
22 expClassName = "TableJobAdaptatorTorque"
|
|
23 os.environ["REPET_JOB_MANAGER"] = REPET_JOB_MANAGER_Initial_Value
|
|
24 self.assertEquals(expClassName, obsClassName)
|
|
25
|
|
26 if __name__ == "__main__":
|
|
27 unittest.main() |