#! /usr/bin/env python
import unittest
from PIPRes.wrap.idl import *
import  PIPRes.corba.api1.CipresIDL_api1
import  PIPRes.corba.api2.CipresIDL_api2


class IDLTest(unittest.TestCase):
    def testGetClassFromRepoID(self):
        for cn in ['Advisor', 'AsyncTreeInfer', 'AsyncTreeIterator', 
                   'ChDBFactory', 'CharacterDB', 'CipresProperties', 
                   'ConsensusTreeCalculator', 'DelegatingTreeImprove', 
                   'LifeCycle', 'MatrixAlign', 'ReadNexus', 
                   'RecursiveScriptable', 'Registry',
                   'Rid3TreeImprove', 'Scriptable', 'TreeDB', 'TreeDBFactory', 
                   'TreeDecompose', 'TreeEvaluate', 'TreeImprove', 'TreeInfer', 
                   'TreeIterator', 'TreeMerge', 'TreePrune', 'TreeRefine', 
                   'TreeSupplier',]:
            r = "IDL:CipresIDL_api1/%s:1.0" % cn
            c = getClassFromRepoID(r)
            self.assertEqual(eval("PIPRes.corba.api1.CipresIDL_api1.%s" % cn), c)
        #now test api2
        for cn in ['Iterator', 'ModelSupplier','TreeInput', 'TreeListInput', 
                   'TreeListOutput', 'TreeOutput', 'TreeScoreInput', 
                   'TreeScoreOutput']:
            r = "IDL:CipresIDL_api2/%s:1.0" % cn
            c = getClassFromRepoID(r)
            self.assertEqual(eval("PIPRes.corba.api2.CipresIDL_api2.%s" % cn), c)

if __name__ == '__main__':
    unittest.main()