mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-20 03:23:23 +08:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
#
|
|
#
|
|
#
|
|
|
|
import sys
|
|
import os
|
|
import unittest
|
|
|
|
# Dynamically append current pykd.pyd path to PYTHONPATH
|
|
sys.path.append(os.path.dirname(sys.argv[1]))
|
|
import pykd
|
|
|
|
import target
|
|
import basetest
|
|
import typeinfo
|
|
import regtest
|
|
import moduletest
|
|
|
|
def getTestSuite( singleName = "" ):
|
|
if singleName == "":
|
|
return unittest.TestSuite(
|
|
[ unittest.TestLoader().loadTestsFromTestCase( basetest.BaseTest ),
|
|
unittest.TestLoader().loadTestsFromTestCase( moduletest.ModuleTest ),
|
|
# unittest.TestLoader().loadTestsFromTestCase( typeinfo.TypeInfoTest ),
|
|
# unittest.TestLoader().loadTestsFromTestCase( regtest.CpuRegTest )
|
|
] )
|
|
else:
|
|
return unittest.TestSuite( unittest.TestLoader().loadTestsFromName( singleName ) )
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
targetAppPath = sys.argv[1]
|
|
|
|
target.moduleName = os.path.splitext(os.path.basename(targetAppPath))[0]
|
|
print "\nTest module: %s" % targetAppPath
|
|
|
|
dbg = pykd.dbgClient()
|
|
|
|
dbg.startProcess( targetAppPath )
|
|
# pykd.go()
|
|
|
|
target.module = dbg.loadModule( target.moduleName )
|
|
|
|
suite = getTestSuite()
|
|
|
|
unittest.TextTestRunner(stream=sys.stdout, verbosity=2).run( suite )
|