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
|
#!/usr/bin/python3
# This file is part of libmodulemd
# Copyright (C) 2018 Red Hat, Inc.
#
# Fedora-License-Identifier: MIT
# SPDX-2.0-License-Identifier: MIT
# SPDX-3.0-License-Identifier: MIT
#
# This program is free software.
# For more information on the license, see COPYING.
# For more information on free software, see
# <https://www.gnu.org/philosophy/free-sw.en.html>.
from contextlib import contextmanager
import signal
import os
import unittest
class TestBase(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestBase, self).__init__(*args, **kwargs)
self._caught_signal = False
@property
def source_root(self):
return os.getenv("MESON_SOURCE_ROOT")
@property
def test_data_path(self):
return os.getenv("TEST_DATA_PATH")
def _catch_signal(self, *sigargs):
if self._caught_signal:
raise AssertionError("Multiple signals were caught")
self._caught_signal = True
@contextmanager
def expect_signal(
self, expected_signal=signal.SIGTRAP, only_on_fatal_warnings=False
):
expect_signal = (not only_on_fatal_warnings) or self.warnings_fatal
self._caught_signal = False
saved_signal = signal.signal(expected_signal, self._catch_signal)
yield None
signal.signal(expected_signal, saved_signal)
if not self._caught_signal and expect_signal:
raise AssertionError("No signal got caught")
elif self._caught_signal and not expect_signal:
raise AssertionError("Signal caught in non-warning state")
@property
def warnings_fatal(self):
gdebug = os.getenv("G_DEBUG", "").split(",")
return "fatal-warnings" in gdebug
def assertRaisesRegex(self, *args, **kwargs):
"""Asserts that the message in a raised exception matches a regex.
Args:
The same as unittest.TestCase.assertRaisesRegex().
"""
try:
return super(TestBase, self).assertRaisesRegex(*args, **kwargs)
except AttributeError:
return self.assertRaisesRegexp(*args, **kwargs)
|
