pykd

1
import unittest
2
import pykd
3
import target
4
import time
5

6
class ProcessTest(unittest.TestCase):
7

8
    def setUp(self):
9
        pykd.startProcess( target.appPath )
10
        pykd.startProcess( target.appPath )
11
        pykd.startProcess( target.appPath )
12

13
    def tearDown(self):
14
        pykd.killAllProcesses()
15

16
    def testGetCurrentProcess(self):
17
        proc = pykd.targetProcess.getCurrent()
18
        self.assertNotEqual(0, proc.systemID )
19
        self.assertNotEqual(0, proc.peb )
20

21
    def testEnumThreads(self):
22
        proc = pykd.targetProcess.getCurrent()
23
        threadNumber = proc.getNumberThreads()
24
        self.assertLess(0, threadNumber)
25
        for i in range(threadNumber):
26
            thread = proc.getThread(i)
27
            self.assertNotEqual(0, thread.systemID )
28
            self.assertNotEqual(0, thread.teb )
29

30
    def testEnumProcesses(self):
31
        processNumber = pykd.targetProcess.getNumber()
32
        for i in range(processNumber):
33
            proc = pykd.targetProcess(i)
34
            self.assertNotEqual(0, proc.systemID)
35
            self.assertNotEqual(0, proc.peb)
36

37
    def testSetCurrentThread(self):
38
        proc = pykd.targetProcess.getCurrent()
39
        threadNumber = proc.getNumberThreads()
40
        self.assertLess(0, threadNumber)
41
        for i in range(threadNumber):
42
            thread = proc.getThread(i)
43
            thread.setCurrent()
44

45
    def testGetBreakpoint(self):
46
        proc = pykd.targetProcess.getCurrent()
47
        self.assertEqual(0, proc.getNumberBreakpoints())
48

49

50

51

52

53

54