pykd

1
import unittest
2
import pykd
3
import sys
4
import os
5

6
class TaggedTest(unittest.TestCase):
7
    def setUp(self):
8
        dump_file = os.path.join( os.path.dirname(sys.argv[0]),
9
                                  r"..\..\kdlibcpp\kdlib\tests\dumps\win8_x64_mem\win8_x64_mem.cab" )
10
        self.dump_id = pykd.loadDump( dump_file )
11

12
        self._existing_id = "D03DC06F-D88E-44C5-BA2A-FAE035172D19"
13
        self._non_existing_id = "88597A32-1493-41CA-BF87-2A950DF4CEE0"
14

15
    def tearDown(self):
16
        pykd.closeDump( self.dump_id )
17

18
    def testEnum(self):
19
        ids = pykd.enumTagged()
20

21
        self.assertTrue( self._existing_id.lower() in [i.lower() for i in ids] )
22

23
        self.assertFalse( self._non_existing_id.lower() in [i.lower() for i in ids] )
24

25

26
    def testLoadBuffer(self):
27
        buff = pykd.loadTaggedBuffer( self._existing_id )
28
        self.assertEqual( len(buff), 0x410 )
29

30
        self.assertRaises( pykd.DbgException, pykd.loadTaggedBuffer, self._non_existing_id )
31

32