Merge pull request #21 from unmade/fix-coverage

Fix coverage

.isort.cfg

@@ -0,0 +1,6 @@
+[settings]
+line_length=88
+multi_line_output=3
+include_trailing_comma=true
+known_first_party=thriftpyi
+default_section=THIRDPARTY

pyproject.toml

@@ -34,11 +34,6 @@ pre-commit = "^1.16"
 mypy = "^0.701.0"
 pylint = "^2.3"
 
-[tool.isort]
-line_length = 88
-multi_line_output = 3
-include_trailing_comma = true
-
 [build-system]
 requires = ["poetry>=0.12"]
 build-backend = "poetry.masonry.api"

src/thriftpyi/cli.py

@@ -1,4 +1,5 @@
 import click
+
 from thriftpyi.main import thriftpyi
 
 

src/thriftpyi/proxies.py

@@ -3,6 +3,7 @@
 
 import thriftpy2
 from thriftpy2.thrift import TPayloadMeta
+
 from thriftpyi.utils import get_python_type
 
 

src/thriftpyi/renderers.py

@@ -2,6 +2,7 @@
 from importlib import resources
 
 from jinja2 import Template
+
 from thriftpyi.entities import Content
 
 

tests/test_cli.py

@@ -3,6 +3,7 @@
 
 import pytest
 from click.testing import CliRunner
+
 from thriftpyi.cli import main
 
 

tests/test_utils.py

@@ -0,0 +1,16 @@
+from unittest import mock
+
+from thriftpyi.utils import _get_binary as get_binary
+
+# This should be redo with 'skipif' and different
+# thriftpy2 versions (old one and the one with binary support)
+
+
+def test_get_binary_returns_str():
+    with mock.patch("thriftpy2.thrift.TType.BINARY", 11):
+        assert get_binary([]) == "str"
+
+
+def test_get_binary_returns_bytes():
+    with mock.patch("thriftpy2.thrift.TType.BINARY", 18):
+        assert get_binary([]) == "bytes"