gistfile1.py

# coding: utf-8
"""
    pythonic.tests.test_checkers
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    :copyright: 2013 by Daniel Neuhäuser
    :license: BSD, see LICENSE.rst for more details
"""
import os

import pytest

from pythonic.checkers import ASTChecker


class CheckerTestBaseMeta(type):
    def __new__(cls, name, bases, attributes):
        @pytest.fixture(params=attributes["warnings"])
        def filename(self, request):
            return request.param
        attributes["filename"] = filename
        return type.__new__(cls, name, bases, attributes)


class CheckerTestBase(object):
    __metaclass__ = CheckerTestBaseMeta
    checker_cls = None
    warnings = {}

    @pytest.fixture
    def messages(self, filename):
        return self.warnings[filename]

    def get_bad_path(self, filename):
        return os.path.join(
            os.path.abspath(os.path.dirname(__file__)),
            "bad",
            filename
        )

    def test_warnings(self, filename, messages, capsys):
        bad_file = self.get_bad_path(filename)
        self.checker_cls(bad_file).run()
        out, err = capsys.readouterr()
        assert not out
        for observed, expected in zip(err.splitlines(), messages):
            assert observed == bad_file + expected



class TestASTChecker(CheckerTestBase):
    checker_cls = ASTChecker

    warnings = {
        "star_import.py": [
            ":1:0: Should not use *-import"
        ],
        "multi_import.py": [
            ":1:0: Should not import multiple modules with one statement"
        ]
    }