Add tests for validators and utils

src/reddel_server/validators.py

@@ -17,7 +17,7 @@ class ValidatorInterface(six.with_metaclass(abc.ABCMeta, object)):
     Override :meth:`ValidatorInterface.__call__`.
     """
     @abc.abstractmethod
-    def __call__(self, red):
+    def __call__(self, red):  # pragma: no cover
         """Validate the given redbaron source
 
         :param red: the source
@@ -26,7 +26,7 @@ def __call__(self, red):
         """
         pass
 
-    def transform(self, red):
+    def transform(self, red):  # pragma: no cover
         """Transform the given red baron
 
         :param red: a red baron or other nodes

tests/test_utils.py

@@ -0,0 +1,20 @@
+"""Tests for the utils module"""
+import pytest
+
+from reddel_server import utils
+
+
+def test_get_attr_from_dotted_path(mocker):
+    """Test the function call importlib correctly"""
+    class MyClass:
+        pass
+
+    importmock = mocker.Mock()
+    modmock = mocker.Mock()
+    modmock.MyClass = MyClass
+    importmock.return_value = modmock
+
+    mockedimport = mocker.patch("importlib.import_module", importmock)
+    result = utils.get_attr_from_dotted_path("test.this.path.MyClass")
+    assert result is MyClass
+    mockedimport.assert_called_once_with('test.this.path')

tests/test_validators.py

@@ -0,0 +1,42 @@
+"""Tests for the validators module"""
+import pytest
+import redbaron
+
+import reddel_server
+
+DEF_SRC = """def foo(bar):
+    spam = bar + 1
+    return spam
+"""
+
+DEF_SRC2 = DEF_SRC + """
+def foo2(bar):
+	pass
+"""
+
+@pytest.mark.parametrize('identifiers,single,src,valid', [
+    [['def', 'defnode'], True, DEF_SRC, True],
+    [['def', 'defnode'], True, DEF_SRC2, False],
+    [['name'], True, DEF_SRC, False],
+    [['def'], False, DEF_SRC2, True]
+])
+def test_barontype_call(identifiers, single, src, valid):
+    val = reddel_server.BaronTypeValidator(identifiers, single=single)
+    red = redbaron.RedBaron(src)
+    if valid:
+        val(red)
+    else:
+        with pytest.raises(reddel_server.ValidationError):
+            val(red)
+
+def test_barontype_transform_single():
+    val = reddel_server.BaronTypeValidator((), single=True)
+    red = redbaron.RedBaron(DEF_SRC)
+    transformed = val.transform(red)
+    assert transformed is red[0]
+
+def test_barontype_transform_singleOff():
+    val = reddel_server.BaronTypeValidator((), single=False)
+    red = redbaron.RedBaron(DEF_SRC)
+    transformed = val.transform(red)
+    assert transformed is red

tox.ini

@@ -25,6 +25,7 @@ deps =
     pytest
     pytest-travis-fold
     pytest-cov
+    pytest-mock
 commands =
     {posargs:py.test --cov --cov-report=term-missing -vv tests}