pykd

1

2
import unittest
3
import target
4
import pykd
5

6
class OutHandler( pykd.eventHandler ):
7

8
    def __init__(self):
9
        pykd.eventHandler.__init__(self)
10
        self.out_counter=0
11
        self.verbose_counter = 0
12
        
13
    def onDebugOutput(self, text, mask):
14
        if mask == pykd.outputFlag.Normal:
15
            self.out_counter += 1
16
        elif mask == pykd.outputFlag.Verbose:
17
            self.verbose_counter += 1    
18

19
class OutputHandlerTest( unittest.TestCase ):
20

21
    def setUp(self):
22
        self.oldMask = pykd.getOutputMask()
23
        self.handler = OutHandler()     
24

25
    def tearDown(self):
26
        pykd.killAllProcesses()
27
        pykd.setOutputMask(self.oldMask)
28

29
    def testDebugOutput(self):
30
        pykd.startProcess( target.appPath )
31
        self.assertTrue( 0 < self.handler.out_counter)
32
        self.assertFalse( 0 < self.handler.verbose_counter)
33

34
    def testDebugOutputMask(self):
35
        pykd.setOutputMask(pykd.outputFlag.Normal | pykd.outputFlag.Verbose)
36
        pykd.startProcess( target.appPath )
37
        self.assertTrue( 0 < self.handler.out_counter)
38
        self.assertTrue( 0 < self.handler.verbose_counter)