"""
Tests for pyDia
"""

import unittest
import target
import pykd

class DiaTest( unittest.TestCase ):

    def testCtor(self):
        """ DiaSymbol can not be created direct """
        try: pykd.DiaSymbol()
        except RuntimeError: pass

    def testFind(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertNotEqual(0, len(gScope))
            symFunction = gScope.find("FuncWithName0")
            self.assertTrue(1 == len( symFunction ))
            symFunction = gScope.findEx(pykd.SymTagNull,
                                           "FuNc*Name?",
                                           pykd.nsCaseInRegularExpression)
            self.assertTrue(len(symFunction) > 1)
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testSize(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertEqual(1, gScope["g_ucharValue"].type().size())
            self.assertEqual(2, gScope["g_ushortValue"].type().size())
            self.assertEqual(4, gScope["g_ulongValue"].type().size())
            self.assertEqual(8, gScope["g_ulonglongValue"].type().size())
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testValue(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertEqual(0x5555, gScope["g_constNumValue"].value())
            self.assertEqual(True, gScope["g_constBoolValue"].value())
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testName(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertEqual( "g_constNumValue", 
                              gScope["g_constNumValue"].name() )
            self.assertEqual( "FuncWithName0",
                              gScope["FuncWithName0"].name() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testRva(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            _rva = gScope["FuncWithName0"].rva()
            self.assertNotEqual(0, _rva)
            modLen = target.module.end() - target.module.begin()
            self.assertTrue( _rva < modLen )
            _rva = gScope["g_string"].rva()
            self.assertNotEqual(0, _rva)
            self.assertTrue( _rva < modLen )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testSymTag(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertEqual( pykd.SymTagFunction, 
                              gScope["FuncWithName0"].symTag() )
            self.assertEqual( pykd.SymTagData,
                              gScope["g_string"].symTag() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testLocType(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertEqual( pykd.LocIsConstant,
                              gScope["g_constNumValue"].locType() )
            self.assertEqual( pykd.LocIsStatic,
                              gScope["FuncWithName1"].locType() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testBasicType(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertFalse(gScope["g_string"].type().isBasic())
            self.assertEqual( pykd.btBool, 
                              gScope["g_constBoolValue"].type().baseType() )
            self.assertEqual( pykd.btULong,
                              gScope["g_ulongValue"].type().baseType() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testBits(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            structWithBits = gScope["structWithBits"]
            bitField = structWithBits["m_bit0_4"]
            self.assertEqual(pykd.LocIsBitField, bitField.locType())
            self.assertEqual(0, bitField.bitPos())
            self.assertEqual(5, bitField.size())
            bitField = structWithBits["m_bit5"]
            self.assertEqual(pykd.LocIsBitField, bitField.locType())
            self.assertEqual(5, bitField.bitPos())
            self.assertEqual(1, bitField.size())
            bitField = structWithBits["m_bit6_7"]
            self.assertEqual(pykd.LocIsBitField, bitField.locType())
            self.assertEqual(6, bitField.bitPos())
            self.assertEqual(2, bitField.size())
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testIndexId(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertNotEqual( gScope["classChild"].indexId(), 
                                 gScope["classBase"].indexId() )
            self.assertNotEqual( gScope["FuncWithName0"].indexId(), 
                                 gScope["FuncWithName1"].indexId() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testUdtKind(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            self.assertEqual(pykd.UdtStruct, gScope["structTest"].udtKind())
            self.assertEqual(pykd.UdtUnion, gScope["unionTest"].udtKind())
            self.assertEqual(pykd.UdtClass, gScope["classBase"].udtKind())
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testOffset(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            structTest = gScope["structTest"]
            self.assertEqual( 0, structTest["m_field0"].offset() )
            self.assertTrue( structTest["m_field0"].offset() < 
                             structTest["m_field1"].offset() )
            self.assertTrue( structTest["m_field1"].offset() < 
                             structTest["m_field2"].offset() )
            self.assertTrue( structTest["m_field2"].offset() < 
                             structTest["m_field3"].offset() )
            self.assertTrue( structTest["m_field3"].offset() < 
                             structTest.size() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testMachine(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            machine = gScope.machineType()
            self.assertTrue( (machine == pykd.IMAGE_FILE_MACHINE_I386) or
                             (machine == pykd.IMAGE_FILE_MACHINE_AMD64) )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testFindByRva(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            func = gScope["FuncWithName0"]
            tplSymOffset = gScope.findByRva(func.rva(), pykd.SymTagFunction)
            self.assertEqual(tplSymOffset[0].indexId(), func.indexId())
            self.assertEqual(tplSymOffset[1], 0)
            tplSymOffset = gScope.findByRva(func.rva()+2, pykd.SymTagFunction)
            self.assertEqual(tplSymOffset[0].indexId(), func.indexId())
            self.assertEqual(tplSymOffset[1], 2)
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testSymbolById(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            func = gScope["FuncWithName0"]
            self.assertEqual( gScope.symbolById(func.indexId()).indexId(),
                              func.indexId())
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)

    def testCount(self):
        try:
            gScope = pykd.diaLoadPdb( str(target.module.pdb()) )
            var = gScope["FuncWithName1"]["_unionTest"]
            self.assertEqual( 2, var.type().count() )
        except pykd.DiaException as diaExcept:
            print diaExcept
            self.assertTrue(False)