Tests: Reorganize common and global tests

tests/common.py

@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+# Copyright (C) 2016 Adrien Vergé
+#
+# 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 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import unittest
+
+import yaml
+
+from yamllint.config import YamlLintConfig
+from yamllint import linter
+
+
+class RuleTestCase(unittest.TestCase):
+    def build_fake_config(self, conf):
+        if conf is None:
+            conf = {}
+        else:
+            conf = yaml.safe_load(conf)
+        conf = {'extends': 'default',
+                'rules': conf}
+        return YamlLintConfig(yaml.safe_dump(conf))
+
+    def check(self, source, conf, **kwargs):
+        expected_problems = []
+        for key in kwargs:
+            assert key.startswith('problem')
+            if len(kwargs[key]) > 2:
+                if kwargs[key][2] == 'syntax':
+                    rule_id = None
+                else:
+                    rule_id = kwargs[key][2]
+            else:
+                rule_id = self.rule_id
+            expected_problems.append(linter.LintProblem(
+                kwargs[key][0], kwargs[key][1], rule=rule_id))
+        expected_problems.sort()
+
+        real_problems = list(linter.run(source, self.build_fake_config(conf)))
+        self.assertEqual(real_problems, expected_problems)