Added unit tests for CLI

jsonschema/cli.py

@@ -2,6 +2,7 @@
 
 import argparse
 import json
+import sys
 
 from . import (
     validate, Draft4Validator, Draft3Validator,
@@ -9,12 +10,12 @@ from . import (
 )
 from .validators import validator_for
 
-def main():
+def main(args):
     parser = argparse.ArgumentParser(description='JSON Schema validator')
     parser.add_argument('schema', help='filename of the JSON Schema')
     parser.add_argument('document', help='filename of the JSON document to validate')
     parser.add_argument('--format', help='validate value format', action='store_true')
-    args = parser.parse_args()
+    args = parser.parse_args(args)
 
     schema = json.load(open(args.schema, 'r'))
     document = json.load(open(args.document, 'r'))
@@ -37,4 +38,4 @@ def main():
     print("√ JSON document is valid.")
 
 if __name__ == '__main__':
-    main()
+    main(sys.argv)

jsonschema/tests/test_cli.py

@@ -0,0 +1,78 @@
+import StringIO
+
+import pytest
+
+from .compat import mock, unittest
+from .. import (
+    cli, Draft4Validator, Draft3Validator,
+    draft3_format_checker, draft4_format_checker,
+)
+
+MOCK_SCHEMAS = {
+    'draft3': {"$schema": "http://json-schema.org/draft-03/schema#"},
+    'draft4': {"$schema": "http://json-schema.org/draft-04/schema#"},
+}
+
+
+class TestCLI(unittest.TestCase):
+    def test_missing_arguments(self):
+        with pytest.raises(SystemExit) as e:
+            cli.main([])
+
+    @mock.patch('__builtin__.open')
+    @mock.patch('jsonschema.cli.validate')
+    def test_filename_argument_order(self, validate, open_):
+        def mock_file(filename, mode):
+            return StringIO.StringIO('{"filename": "%s"}' % filename)
+        open_.side_effect = mock_file
+
+        cli.main(['document.json', 'schema.json'])
+
+        open_.assert_has_calls([mock.call('document.json', 'r'),
+                                mock.call('schema.json', 'r')],
+                               any_order=True)
+        self.assertEqual(open_.call_count, 2)
+
+        validate.assert_called_once_with({'filename': 'schema.json'},
+                                         {'filename': 'document.json'},
+                                         Draft4Validator,
+                                         format_checker=None)
+
+    @mock.patch('__builtin__.open')
+    @mock.patch('jsonschema.cli.json.load')
+    @mock.patch('jsonschema.cli.validate')
+    def test_raise_exception(self, validate, json_load, open_):
+        validate.side_effect = Exception('Did not validate correctly')
+        with pytest.raises(Exception) as e:
+            cli.main([None, None])
+        self.assertEqual(e.exconly(), "Exception: Did not validate correctly")
+
+    @mock.patch('__builtin__.open')
+    @mock.patch('jsonschema.cli.json.load')
+    @mock.patch('jsonschema.cli.validate')
+    def test_format(self, validate, json_load, open_):
+        schema = {"$schema": "http://json-schema.org/draft-04/schema#"}
+        json_load.return_value = schema
+
+        cli.main([None, None])
+        validate.assert_called_once_with(schema, schema, Draft4Validator,
+                                         format_checker=None)
+        validate.reset_mock()
+        cli.main([None, None, '--format'])
+        validate.assert_called_once_with(schema, schema, Draft4Validator,
+                                         format_checker=draft4_format_checker)
+
+    @mock.patch('__builtin__.open')
+    @mock.patch('jsonschema.cli.json.load')
+    @mock.patch('jsonschema.cli.validate')
+    def test_draft3(self, validate, json_load, open_):
+        schema = {"$schema": "http://json-schema.org/draft-03/schema#"}
+        json_load.return_value = schema
+
+        cli.main([None, None])
+        validate.assert_called_once_with(schema, schema, Draft3Validator,
+                                         format_checker=None)
+        validate.reset_mock()
+        cli.main([None, None, '--format'])
+        validate.assert_called_once_with(schema, schema, Draft3Validator,
+                                         format_checker=draft3_format_checker)