diff options
Diffstat (limited to 'python/samba/tests/__init__.py')
| -rw-r--r-- | python/samba/tests/__init__.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/python/samba/tests/__init__.py b/python/samba/tests/__init__.py index c4d24fa913f..b6293dff757 100644 --- a/python/samba/tests/__init__.py +++ b/python/samba/tests/__init__.py @@ -31,9 +31,10 @@ import subprocess import sys import tempfile import unittest +import re import samba.auth import samba.dcerpc.base -from samba.compat import PY3 +from samba.compat import PY3, text_type if not PY3: # Py2 only from samba.samdb import SamDB @@ -161,6 +162,14 @@ class TestCase(unittest.TestCase): self._cleanups = getattr(self, "_cleanups", []) + [ (fn, args, kwargs)] + def assertRegexpMatches(self, text, regex, msg=None): + # PY3 note: Python 3 will never see this, but we use + # text_type for the benefit of linters. + if isinstance(regex, (str, text_type)): + regex = re.compile(regex) + if not regex.search(text): + self.fail(msg) + def _addSkip(self, result, reason): addSkip = getattr(result, 'addSkip', None) if addSkip is not None: |