add more test

Author: alingse

jsoncsv/main.py

@@ -10,9 +10,13 @@
 
 def separator_type(sep):
     if len(sep) != 1:
-        raise click.BadOptionUsage('separator can only be a char')
+        raise click.BadOptionUsage(
+            option_name='separator',
+            message='separator can only be a char')
     if sep == unit_char:
-        raise click.BadOptionUsage('separator can not be `\\` ')
+        raise click.BadOptionUsage(
+            option_name='separator',
+            message='separator can not be `\\` ')
     return sep
 
 

tests/test_jsoncsv.py

@@ -10,24 +10,29 @@
 
 
 class Testjsoncsv(unittest.TestCase):
-
     def test_jsoncsv_expand(self):
         runner = CliRunner()
-        args = ['-e', 'fixture/files/raw.0.json', 'fixture/files/tmp.expand.0.json']
+        args = ['-e', 'fixture/files/raw.0.json',
+                'fixture/files/tmp.expand.0.json']
         result = runner.invoke(jsoncsv, args=args)
         assert result.exit_code == 0
 
     def test_jsoncsv_expand_with_json_array(self):
         runner = CliRunner()
-        args = ['-e', 'fixture/files/raw.1.json', 'fixture/files/tmp.expand.1.json', '-A']
+        args = ['-e', 'fixture/files/raw.1.json',
+                'fixture/files/tmp.expand.1.json', '-A']
         result = runner.invoke(jsoncsv, args=args)
         assert result.exit_code == 0
 
     def test_jsoncsv_expand_restore(self):
         runner = CliRunner(echo_stdin=True)
-        result = runner.invoke(jsoncsv, args=['-e', 'fixture/files/raw.2.json', 'fixture/files/tmp.expand.2.json'])
+        result = runner.invoke(jsoncsv,
+                               args=['-e', 'fixture/files/raw.2.json',
+                                     'fixture/files/tmp.expand.2.json'])
         assert result.exit_code == 0
-        result = runner.invoke(jsoncsv, args=['-r', 'fixture/files/tmp.expand.2.json', 'fixture/files/tmp.restore.2.json'])
+        result = runner.invoke(jsoncsv,
+                               args=['-r', 'fixture/files/tmp.expand.2.json',
+                                     'fixture/files/tmp.restore.2.json'])
         assert result.exit_code == 0
 
         with io.open('fixture/files/raw.2.json', 'r') as f:
@@ -37,3 +42,17 @@ def test_jsoncsv_expand_restore(self):
             resotre_data = [json.loads(line) for line in f]
 
         self.assertEqual(input_data, resotre_data)
+
+    def test_jsoncsv_with_error_args(self):
+        runner = CliRunner()
+        args = ['-s', 'aa', '-e', 'fixture/files/raw.0.json',
+                'fixture/files/tmp.expand.0.json']
+        result = runner.invoke(jsoncsv, args=args)
+        assert result.exit_code != 0
+
+    def test_jsoncsv_with_error_args_expand_and_restore(self):
+        runner = CliRunner()
+        args = ['-r', '-e', 'fixture/files/raw.0.json',
+                'fixture/files/tmp.expand.0.json']
+        result = runner.invoke(jsoncsv, args=args)
+        assert result.exit_code != 0