mirror of
https://github.com/vivisect/vivisect
synced 2026-06-08 18:04:23 +00:00
rakuyo
34 lines
825 B
Python
34 lines
825 B
Python
import os
|
|
|
|
import vtrace
|
|
import vtrace.tests as vt_tests
|
|
|
|
class ThreadNotifier(vtrace.Notifier):
|
|
|
|
def __init__(self):
|
|
vtrace.Notifier.__init__(self)
|
|
self.threadexit = False
|
|
self.threadcreate = False
|
|
|
|
def notify(self, event, trace):
|
|
if event == vtrace.NOTIFY_CREATE_THREAD:
|
|
self.threadcreate = True
|
|
return
|
|
|
|
if event == vtrace.NOTIFY_EXIT_THREAD:
|
|
self.threadexit = True
|
|
return
|
|
|
|
class VtraceThreadTest(vt_tests.VtraceProcessTest):
|
|
|
|
pypath = os.path.join('vtrace','tests','mains','mainthreads.py')
|
|
|
|
def test_vtrace_threads(self):
|
|
n = ThreadNotifier()
|
|
|
|
self.trace.registerNotifier(vtrace.NOTIFY_ALL, n)
|
|
self.runUntilExit()
|
|
|
|
self.assertTrue(n.threadcreate)
|
|
self.assertTrue(n.threadexit)
|