add cluster up and cluster down functionality (#5)

Author: MichaelClifford

src/codeflare_sdk/cluster/cluster.py

@@ -10,16 +10,15 @@ def __init__(self, config: ClusterConfiguration):
         self.config = config
 
     # creates a new cluser with the provided or default spec
-    def up(self):
-        pass
-
-    def down(self, name):
-        # FIXME on what the exact details should be
-        # 1. delete the appwrapper and that should delete the cluster
-        # FIXME on what the exact details should be
-        # 1. delete the appwrapper and that should delete the cluster
-        pass
-
+    def up(self, namespace='default'):
+        with oc.project(namespace):
+            oc.invoke("apply", ["-f", 
+            "https://raw.githubusercontent.com/IBM/multi-cluster-app-dispatcher/quota-management/doc/usage/examples/kuberay/config/aw-raycluster.yaml"])
+
+    def down(self, name, namespace='default'):
+        with oc.project(namespace):
+            oc.invoke("delete",["AppWrapper", self.config.name])
+        
     def status(self, print_to_console=True):
         cluster = _ray_cluster_status(self.config.name)
         if cluster:

tests/test_clusters.py

@@ -1,7 +1,12 @@
 from codeflare_sdk.cluster.cluster import list_all_clusters, _app_wrapper_status
 from codeflare_sdk.cluster.cluster import Cluster, ClusterConfiguration
+import time
+
+def test_cluster_up():
+     cluster = Cluster(ClusterConfiguration(name='raycluster-autoscaler'))
+     cluster.up()
+     time.sleep(15)
 
-#for now these tests assume that the cluster was already created
 def test_list_clusters():
     clusters = list_all_clusters()
 
@@ -11,4 +16,8 @@ def test_cluster_status():
 
 def test_app_wrapper_status():
     print(_app_wrapper_status('raycluster-autoscaler'))
-    
+
+def test_cluster_down():
+    cluster = Cluster(ClusterConfiguration(name='raycluster-autoscaler'))
+    cluster.down(name='raycluster-autoscaler')
+