add bin/create_brancher and bin/destroy_brancher

Author: vdloo

bin/create_brancher

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

bin/destroy_brancher

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

hypernode_api_python/client.py

@@ -840,7 +840,6 @@ def get_active_branchers(self, app_name):
             "GET", HYPERNODE_API_BRANCHER_APP_ENDPOINT.format(app_name)
         )
 
-    # TODO: add entrypoint for this method in bin/ and commands.py
     def create_brancher(self, app_name, data):
         """
         Create a new branch (server replica) of your Hypernode.
@@ -854,7 +853,6 @@ def create_brancher(self, app_name, data):
             "POST", HYPERNODE_API_BRANCHER_APP_ENDPOINT.format(app_name), data=data
         )
 
-    # TODO: add entrypoint for this method in bin/ and commands.py
     def destroy_brancher(self, brancher_name):
         """
         Destroy an existing brancher node of your Hypernode.

hypernode_api_python/commands.py

@@ -641,3 +641,63 @@ def get_active_branchers(args=None):
     client = get_client()
     app_name = get_app_name()
     print_response(client.get_active_branchers(app_name))
+
+
+def create_brancher(args=None):
+    parser = ArgumentParser(
+        description="""
+Create a Brancher Hypernode from the specified app.
+Outputs the app_info of the brancher to be created..
+
+Example:
+$ ./bin/create_brancher
+{
+  "name": "yourappname-ephoj82yb",
+  "parent": "yourappname",
+  "type": "brancher",
+  "product": "FALCON_M_202203",
+  "domainname": "yourappname-ephoj82yb.hypernode.io",
+  ...
+}
+""",
+        formatter_class=RawTextHelpFormatter,
+    )
+    parser.parse_args(args=args)
+    client = get_client()
+    app_name = get_app_name()
+    data = {}
+    print_response(client.create_brancher(app_name, data=data))
+
+
+def destroy_brancher(args=None):
+    parser = ArgumentParser(
+        description="""
+Destroy a Brancher Hypernode.
+
+Examples:
+$ ./bin/destroy_brancher yourbrancherappname-eph12345
+A job has been posted to cancel the 'yourbrancherappname-eph12345' brancher app.
+""",
+        formatter_class=RawTextHelpFormatter,
+    )
+    parser.add_argument(
+        "brancher_app_name",
+        help="The name of the brancher to destroy. See ./bin/get_active_branchers",
+    )
+    args = parser.parse_args(args=args)
+    client = get_client()
+    try:
+        client.destroy_brancher(args.brancher_app_name)
+        print(
+            "A job has been posted to cancel the '{}' brancher app.".format(
+                args.brancher_app_name
+            )
+        )
+        exit(EX_OK)
+    except Exception as e:
+        print(
+            "Brancher app '{}' failed to be cancelled: {}".format(
+                args.brancher_app_name, e
+            )
+        )
+        exit(EX_UNAVAILABLE)

tests/commands/test_create_brancher.py

@@ -0,0 +1,40 @@
+from hypernode_api_python.commands import create_brancher
+from tests.testcase import TestCase
+
+
+class TestCreateBrancher(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_create_brancher_gets_client(self):
+        create_brancher([])
+
+        self.get_client.assert_called_once_with()
+
+    def test_create_brancher_gets_app_name(self):
+        create_brancher([])
+
+        self.get_app_name.assert_called_once_with()
+
+    def test_create_brancher_creates_brancher(self):
+        create_brancher([])
+
+        expected_data = {}
+        self.client.create_brancher.assert_called_once_with(
+            "myappname", data=expected_data
+        )
+
+    def test_create_brancher_prints_response(self):
+        create_brancher([])
+
+        self.print_response.assert_called_once_with(
+            self.client.create_brancher.return_value
+        )

tests/commands/test_destroy_brancher.py

@@ -0,0 +1,53 @@
+from os import EX_UNAVAILABLE, EX_OK
+
+from hypernode_api_python.commands import destroy_brancher
+from tests.testcase import TestCase
+
+
+class TestDestroyBrancher(TestCase):
+    def setUp(self):
+        self.exit = self.set_up_patch("hypernode_api_python.commands.exit")
+        self.print = self.set_up_patch("hypernode_api_python.commands.print")
+        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
+
+    def test_destroy_brancher_gets_client(self):
+        destroy_brancher(["myappname-eph1234"])
+
+        self.get_client.assert_called_once_with()
+
+    def test_destroy_brancher_destroys_brancher(self):
+        destroy_brancher(["myappname-eph1234"])
+
+        self.client.destroy_brancher.assert_called_once_with("myappname-eph1234")
+
+    def test_destroy_brancher_prints_app_name_is_valid(self):
+        destroy_brancher(["myappname-eph1234"])
+
+        self.print.assert_called_once_with(
+            "A job has been posted to cancel the 'myappname-eph1234' brancher app."
+        )
+
+    def test_destroy_brancher_prints_app_name_is_invalid(self):
+        self.client.destroy_brancher.side_effect = MemoryError("Killed")
+
+        destroy_brancher(["myappname-eph1234"])
+
+        self.print.assert_called_once_with(
+            "Brancher app 'myappname-eph1234' failed to be cancelled: Killed"
+        )
+
+    def test_destroy_brancher_exits_zero_when_cancel_job_posted(self):
+        destroy_brancher(["myappname-eph1234"])
+
+        self.exit.assert_called_once_with(EX_OK)
+
+    def test_destroy_brancher_exits_nonzero_when_unexpected_exception_occurs(self):
+        self.client.destroy_brancher.side_effect = RuntimeError
+
+        destroy_brancher(["myappname-eph1234"])
+
+        self.exit.assert_called_once_with(EX_UNAVAILABLE)