implement bin/check_xgrade

Author: vdloo

bin/check_xgrade

@@ -0,0 +1 @@
+command

hypernode_api_python/commands.py

@@ -553,3 +553,30 @@ def get_active_products(args=None):
     parser.parse_args(args=args)
     client = get_client()
     print_response(client.get_active_products())
+
+
+def check_xgrade(args=None):
+    parser = ArgumentParser(
+        description="""
+Verify that the specified app can be upgraded to the specified product.
+This checks if there is enough disk space available. The output will also
+show whether or not there will be an IP change and if a volume swap xgrade
+would be performed instead of an rsync xgrade.
+
+Example:
+$ ./bin/check_xgrade FALCON_L_202203
+{
+  "has_valid_vat_number": true,
+  "has_valid_payment_method": true,
+  "will_change_ip": false,
+  "will_do_volswap": false,
+  "will_disk_fit": true
+}
+""",
+        formatter_class=RawTextHelpFormatter,
+    )
+    parser.add_argument("product_code", help="The code of the product to check")
+    args = parser.parse_args(args=args)
+    client = get_client()
+    app_name = get_app_name()
+    print_response(client.check_xgrade(app_name, args.product_code))

tests/commands/test_check_xgrade.py

@@ -0,0 +1,37 @@
+from hypernode_api_python.commands import check_xgrade
+from tests.testcase import TestCase
+
+
+class TestCheckXgrade(TestCase):
+    def setUp(self):
+        self.print_response = self.set_up_patch(
+            "hypernode_api_python.commands.print_response"
+        )
+        self.get_client = self.set_up_patch("hypernode_api_python.commands.get_client")
+        self.client = self.get_client.return_value
+        self.get_app_name = self.set_up_patch(
+            "hypernode_api_python.commands.get_app_name"
+        )
+        self.get_app_name.return_value = "myappname"
+
+    def test_check_xgrade_gets_client(self):
+        check_xgrade(["FALCON_S_202203"])
+
+        self.get_client.assert_called_once_with()
+
+    def test_check_xgrade_gets_app_name(self):
+        check_xgrade(["FALCON_S_202203"])
+
+        self.get_app_name.assert_called_once_with()
+
+    def test_check_xgrade_gets_product_info(self):
+        check_xgrade(["FALCON_S_202203"])
+
+        self.client.check_xgrade.assert_called_once_with("myappname", "FALCON_S_202203")
+
+    def test_check_xgrade_prints_product(self):
+        check_xgrade(["FALCON_S_202203"])
+
+        self.print_response.assert_called_once_with(
+            self.client.check_xgrade.return_value
+        )