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
|
'''Test recoverable_problem'''
# Copyright (C) 2012 Canonical Ltd.
# Author: Evan Dandrea <ev@ubuntu.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2 of the License, or (at your
# option) any later version. See http://www.gnu.org/copyleft/gpl.html for
# the full text of the license.
import unittest
import sys
import os
import subprocess
import tempfile
import time
import shutil
import apport.report
class T(unittest.TestCase):
def setUp(self):
self.report_dir = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, self.report_dir)
os.environ['APPORT_REPORT_DIR'] = self.report_dir
self.datadir = os.environ.get('APPORT_DATA_DIR', '/usr/share/apport')
def wait_for_report(self):
base = os.path.abspath(sys.argv[0]).replace('/', '_')
path = os.path.join(self.report_dir,
'%s.%d.crash' % (base, os.getuid()))
seconds = 0
while not os.path.exists(path):
time.sleep(1)
seconds += 1
self.assertTrue(seconds < 10, 'timeout while waiting for %s to be created.' % path)
return path
def call_recoverable_problem(self, data):
cmd = ['%s/recoverable_problem' % self.datadir]
proc = subprocess.Popen(cmd, stdin=subprocess.PIPE,
stderr=subprocess.PIPE)
err = proc.communicate(data.encode('UTF-8'))[1]
if proc.returncode != 0:
self.assertNotEqual(err, b'') # we expect some error message
raise subprocess.CalledProcessError(proc.returncode, cmd[0])
self.assertEqual(err, b'')
def test_recoverable_problem(self):
'''recoverable_problem with valid data'''
self.call_recoverable_problem('hello\0there')
path = self.wait_for_report()
with open(path, 'rb') as report_path:
report = apport.report.Report()
report.load(report_path)
self.assertEqual(report['hello'], 'there')
self.assertTrue('Pid:\t%d' % os.getpid() in report['ProcStatus'])
def test_recoverable_problem_dupe_sig(self):
'''recoverable_problem duplicate signature includes ExecutablePath'''
self.call_recoverable_problem('Package\0test\0DuplicateSignature\0ds')
path = self.wait_for_report()
with open(path, 'rb') as report_path:
report = apport.report.Report()
report.load(report_path)
exec_path = report.get('ExecutablePath')
self.assertEqual(report['DuplicateSignature'], '%s:ds' %
(exec_path))
self.assertTrue('Pid:\t%d' % os.getpid() in report['ProcStatus'])
def test_invalid_data(self):
'''recoverable_problem with invalid data'''
self.assertRaises(subprocess.CalledProcessError,
self.call_recoverable_problem, 'hello')
self.assertRaises(subprocess.CalledProcessError,
self.call_recoverable_problem,
'hello\0there\0extraneous')
self.assertRaises(subprocess.CalledProcessError,
self.call_recoverable_problem,
'hello\0\0there')
unittest.main()
|