From d7e0479ee742073fc1093ef9f2177153e93ae4a2 Mon Sep 17 00:00:00 2001
From: barredterra <14891507+barredterra@users.noreply.github.com>
Date: Mon, 5 Jul 2021 19:53:02 +0200
Subject: [PATCH] test: add test for `bench version`

---
 frappe/tests/test_commands.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/frappe/tests/test_commands.py b/frappe/tests/test_commands.py
index 07bdf8791e..8f9bbb1afb 100644
--- a/frappe/tests/test_commands.py
+++ b/frappe/tests/test_commands.py
@@ -426,3 +426,13 @@ class TestCommands(BaseTestCommands):
 		self.assertEqual(self.returncode, 0)
 		self.assertIn("pong", self.stdout)
 
+	def test_version(self):
+		self.execute("bench version")
+		self.assertEqual(self.returncode, 0)
+
+		for output in ["legacy", "plain", "table", "json"]:
+			self.execute(f"bench version -f {output}")
+			self.assertEqual(self.returncode, 0)
+		
+		self.execute("bench version -f invalid")
+		self.assertEqual(self.returncode, 1)