1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#! /usr/bin/python
# -*- coding: utf-8 -*-
import unittest
import sys
import rpm
from subprocess import Popen, PIPE
import os
import gobject
PROGNAME = "abrt_test"
sys.path.append("../src/gui")
import CCDBusBackend
def info_msg(fmt, *args):
sys.stdout.write("%s: %s\n" % (PROGNAME, fmt % args))
def log1(message):
pass
CRASH = "crash"
TIMEOUT = "timeout"
# prevent creating daemon twice
# otherwise we get can't create signal..
daemon = None
def get_daemon():
global daemon
if not daemon:
daemon = CCDBusBackend.DBusManager()
return daemon
class TestCrash(unittest.TestCase):
def setUp(self):
#Test.__init__(self)
#print "setUp TestCrash"
self.daemon = get_daemon()
self.package_nvr = None
self.timeout = None
self.result = None
self.timeout = None
self.loop = gobject.MainLoop()
self.crash_signal = self.daemon.connect("crash", self.on_crash)
def get_package(self, filename):
ts = rpm.TransactionSet()
mi = ts.dbMatch(rpm.RPMTAG_BASENAMES, filename)
for h in mi:
package = h['nvr']
log1("Killing %s from package %s" % (filename, package))
return package
def on_timeout(self, test, critical):
#info_msg("Test has timed out")
if critical:
#info_msg("[ FAILED ]")
self.result = TIMEOUT
self.loop.quit()
def on_crash(self, daemon, crashed_package_nvr, uuid, uid):
if self.timeout:
#print "removing source id: %i" % self.timeout
gobject.source_remove(self.timeout)
log1("got crash signal")
self.loop.quit()
self.assertEqual(self.package_nvr, crashed_package_nvr,
"expected: %s, got: %s" % (crashed_package_nvr, self.package_nvr))
self.result = CRASH
class TestCompiled(TestCrash):
#def __init__(self):
#TestCrash.__init__(self)
def setUp(self):
#info_msg("C/C++ crash detection")
TestCrash.setUp(self)
def kill_sleep(self):
# kill runs sleep 100 and sends a SEGV to it"
self.timeout = gobject.timeout_add(5000, self.on_timeout, "c/c++", True)
app_to_kill = "/bin/sleep"
self.package_nvr = self.get_package(app_to_kill)
pid = Popen([app_to_kill,"100"], stdout=PIPE, bufsize=-1).pid
os.kill(pid, 11)
return False
def test_c_cpp_crash(self):
gobject.timeout_add(5, self.kill_sleep)
self.loop.run()
self.assertEqual(self.result, CRASH, "expected: %s, got: %s" % (CRASH, self.result))
def test_timeout_on_repeat(self):
#info_msg("testing repeated crash - should timedout")
gobject.timeout_add(5, self.kill_sleep)
self.loop.run()
self.assertEqual(self.result, TIMEOUT, "expected: %s, got: %s" % (TIMEOUT, self.result))
#def runTest(self):
# info_msg("Testing C/C++")
def tearDown(self):
self.daemon.disconnect(self.crash_signal)
class TestPython(TestCrash):
def test_catch_exception(self):
pass
def test_pyhook(self):
# TODO: how to test the hook with default config?
# we need broken signed package...
pass
if __name__ == "__main__":
try:
# finds every instance of class TestCompiled and run every method
# called test_*
suite = unittest.TestLoader().loadTestsFromTestCase(TestCompiled)
unittest.TextTestRunner(verbosity=2).run(suite)
suite = unittest.TestLoader().loadTestsFromTestCase(TestPython)
unittest.TextTestRunner(verbosity=2).run(suite)
except KeyboardInterrupt:
sys.exit()
|