summaryrefslogtreecommitdiff
path: root/test/test_utils.py
diff options
context:
space:
mode:
authorTorsten Marek <tmarek@google.com>2013-06-19 14:34:47 +0200
committerTorsten Marek <tmarek@google.com>2013-06-19 14:34:47 +0200
commit2ceab9e47e77ffd2cf5284877abcbb9d66048b9e (patch)
tree65f39310a7d6cc1ad30a88df37b075348c22eb6f /test/test_utils.py
parent8e98049bd624c91c80ec1cf2e85ab7c60144ce8e (diff)
downloadpylint-2ceab9e47e77ffd2cf5284877abcbb9d66048b9e.tar.gz
Also use the proper way of checking for enabled messages for leave_* methods.
Diffstat (limited to 'test/test_utils.py')
-rw-r--r--test/test_utils.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/test/test_utils.py b/test/test_utils.py
index f78aaae..282826b 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -41,6 +41,10 @@ class PyLintASTWalkerTest(TestCase):
def visit_assname(self, module):
self.called.add('assname')
+ @check_messages('second-message')
+ def leave_assname(self, module):
+ raise NotImplementedError
+
def testCheckMessages(self):
linter = self.MockLinter({'first-message': True,
'second-message': False,