Merge 68972ec into 837a0a9

git_json_tree.py

@@ -126,6 +126,7 @@ def decode(repo, tree_id):
                 return []
             elif b'.object' in tree:
                 return {}
+
             raise TypeError('Unknown tree type.')
 
         if all((isinstance(key[0], key_types) for key in items)):

test_git_json_tree.py

@@ -1,13 +1,15 @@
 """Test Git-JSON-Tree library."""
 
+import json
 from string import printable
 
 import hypothesis.strategies as st
 import pytest
+from click.testing import CliRunner
 from dulwich.repo import Repo
 from hypothesis import assume, given, settings
 
-from git_json_tree import decode, encode
+from git_json_tree import cli, decode, encode
 
 
 @pytest.fixture()
@@ -16,6 +18,12 @@ def repo(tmpdir):
     yield Repo.init_bare(str(tmpdir))
 
 
+@pytest.fixture()
+def runner(repo):
+    """Define a click runner."""
+    return CliRunner()
+
+
 def json_data():
     """Generate JSON data."""
     return st.recursive(
@@ -43,3 +51,23 @@ def test_encode_decode(data, repo):
     """Test (d)encoding."""
     assume(isinstance(data, (dict, list)))
     assert decode(repo, encode(repo, data)) == data
+
+
+@given(data=json_data())
+@settings(max_examples=100, deadline=10000)
+def test_cli_encoder(data, runner):
+    """Test cli encoder."""
+    assume(isinstance(data, (dict, list)))
+    encoded = runner.invoke(cli, ['encode'], input=json.dumps(data))
+    decoded = runner.invoke(cli, ['decode', encoded.output.strip()])
+    assert json.loads(decoded.output) == data
+
+
+@given(data=json_data())
+@settings(max_examples=100, deadline=10000)
+def test_smudge_clean(data, runner):
+    """Test Git integration."""
+    assume(isinstance(data, (dict, list)))
+    cleaned = runner.invoke(cli, ['clean'], input=json.dumps(data))
+    smudged = runner.invoke(cli, ['smudge'], input=cleaned.output)
+    assert json.loads(smudged.output) == data