Merge pull request #170 from moggers87/169-version-command

Set package name and message string for version option

salmon/commands.py

@@ -78,7 +78,7 @@ def inner(fn):
 
 
 @click.group(epilog=copyright_notice)
-@click.version_option()
+@click.version_option(package_name="salmon-mail", message="%(version)s")
 def main():
     """Python mail server"""
     pass

tests/test_command.py

@@ -6,7 +6,7 @@
 
 from click import testing
 
-from salmon import queue, commands, encoding, mail, routing, utils
+from salmon import queue, commands, encoding, mail, routing, utils, __version__ as salmon_version
 
 from .setup_env import SalmonTestCase
 
@@ -108,6 +108,12 @@ def test_sendmail_command(self, client_mock):
         runner.invoke(commands.main, ("sendmail", "--host", "127.0.0.1", "--port", "8899", "test@localhost"))
         self.assertEqual(client_mock.return_value.sendmail.call_count, 1)
 
+    def test_version_option(self):
+        runner = CliRunner()
+        result = runner.invoke(commands.main, ("--version"))
+        self.assertEqual(result.exit_code, 0)
+        self.assertEqual(result.output, f"{salmon_version}\n")
+
 
 class StartCommandTestCase(SalmonTestCase):
     @patch('salmon.utils.daemonize')