Added: [pdftool] --version option

mypaceshun · Dec 16, 2021 · 35f78d9 · 35f78d9
1 parent 3e24ca1
commit 35f78d9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/shuncommands/pdftool.py b/shuncommands/pdftool.py
@@ -7,6 +7,8 @@
 
 
 @click.group()
+@click.version_option(None, '-v', '--version')
+@click.help_option('-h', '--help')
 def ctx():
     '''
     \b
@@ -15,6 +17,8 @@ def ctx():
 
 
 @ctx.command()
+@click.version_option(None, '-v', '--version')
+@click.help_option('-h', '--help')
 @click.option('password', '-p',
               help='decrypt password',
               prompt=True,

diff --git a/tests/test_pdftool.py b/tests/test_pdftool.py
@@ -4,12 +4,26 @@
 
 
 class TestPdfTool():
-    def test_run(self):
+    def test_run_help(self):
         runner = CliRunner()
         res = runner.invoke(pdftool.ctx, ['--help'])
         assert res.exit_code == 0
 
+        res = runner.invoke(pdftool.ctx, ['-h'])
+        assert res.exit_code == 0
+
+    def test_run_version(self):
+        runner = CliRunner()
+        res = runner.invoke(pdftool.ctx, ['--version'])
+        assert res.exit_code == 0
+
+        res = runner.invoke(pdftool.ctx, ['-v'])
+        assert res.exit_code == 0
+
     def test_run_unlock(self):
         runner = CliRunner()
         res = runner.invoke(pdftool.ctx, ['unlock', '--help'])
         assert res.exit_code == 0
+
+        res = runner.invoke(pdftool.ctx, ['unlock', '-h'])
+        assert res.exit_code == 0