chore(tests): add integration tests for deployments command (#374)

Co-authored-by: Chris Wetherill <[email protected]>

Author: Roger Lam

gradient/api_sdk/repositories/gradient_deployments.py

@@ -5,7 +5,7 @@
 def create_deployment(name, project_id, spec, cluster_id=None, api_key=None):
     client = graphql_client(api_key)
     query = gql(
-    """
+        """
         mutation createDeployment($input: CreateDeploymentInput!) {
             createDeployment(input: $input) {
                 deployment {
@@ -29,7 +29,7 @@ def create_deployment(name, project_id, spec, cluster_id=None, api_key=None):
 def update_deployment(id, name=None, project_id=None, spec=None, cluster_id=None, api_key=None):
     client = graphql_client(api_key)
     query = gql(
-    """
+        """
         mutation updateDeployment($input: UpdateDeploymentInput!) {
             updateDeployment(input: $input) {
                 deployment {
@@ -53,15 +53,15 @@ def update_deployment(id, name=None, project_id=None, spec=None, cluster_id=None
         input["spec"] = spec
 
     params = {
-        "input": input	
+        "input": input
     }
     return client.execute(query, variable_values=params)['updateDeployment']['deployment']
 
 
 def get_deployment(id, first=100, api_key=None):
     client = graphql_client(api_key)
     query = gql(
-    """
+        """
         query getDeployment($id: UUID!, $first: Int!) {
             deployment(id: $id) {
                 id
@@ -133,7 +133,7 @@ def get_deployment(id, first=100, api_key=None):
 def list_deployments(first=100, api_key=None):
     client = graphql_client(api_key)
     query = gql(
-    """
+        """
         query getDeployments($first: Int!) {
             deployments(first: $first) {
                 nodes {
@@ -187,7 +187,7 @@ def list_deployments(first=100, api_key=None):
 def delete_deployment(id, api_key=None):
     client = graphql_client(api_key)
     query = gql(
-    """
+        """
         mutation deleteDeployment($input: DeleteDeploymentInput!) {
             deleteDeployment(input: $input) {
                 deployment {

gradient/cli/__init__.py

@@ -18,14 +18,11 @@
 import gradient.cli.tensorboards
 import gradient.cli.workflows
 from gradient.api_sdk.config import config
-
-
 if config.USE_LEGACY_DEPLOYMENTS:
     import gradient.cli.deployments
 else:
     import gradient.cli.gradient_deployments
 
-
 def show(self, file=None):
     if file is None:
         file = get_text_stderr()

gradient/cli/cli.py

@@ -31,11 +31,6 @@ def cli():
     pass
 
 
-# TODO: delete experiment - not implemented in the api
-# TODO: modify experiment - not implemented in the api
-# TODO: create experiment template?? What is the difference between experiment and experiment template?
-
-
 @cli.command("version", help="Show the version and exit")
 def get_version():
     command = login_commands.ShowVersionCommand()

gradient/cli/gradient_deployments.py

@@ -6,10 +6,10 @@
 from gql import gql
 from gql.transport.exceptions import TransportQueryError
 
-from gradient.api_sdk import create_deployment, list_deployments, delete_deployment, get_deployment, update_deployment
+from gradient.api_sdk.repositories import gradient_deployments
 from gradient.cli import common
 from gradient.cli.cli import cli
-from gradient.cli.common import  api_key_option, ClickGroup
+from gradient.cli.common import api_key_option, ClickGroup
 from gradient.commands.helpers import print_table, formatted_graphql
 from gradient.exceptions import ApplicationError
 
@@ -52,7 +52,6 @@ def deployments():
     help="Cluster ID",
     cls=common.GradientOption,
 )
-
 @click.option(
     "--spec",
     "spec_path",
@@ -70,7 +69,7 @@ def create_deployment_command(ctx, api_key, name, project_id, cluster_id, spec_p
         return
 
     try:
-        deployment = create_deployment(name, project_id, spec, cluster_id, api_key=api_key)
+        deployment = gradient_deployments.create_deployment(name, project_id, spec, cluster_id, api_key=api_key)
         print(f'Created deployment: {deployment["id"]}')
     except TransportQueryError as error:
         logger.error(error.errors[0]['message'])
@@ -126,20 +125,21 @@ def update_deployment_command(ctx, api_key, id, name, project_id, spec_path, clu
             return
 
     try:
-        deployment = update_deployment(id, name, project_id, spec, cluster_id, api_key=api_key)
+        deployment = gradient_deployments.update_deployment(id, name, project_id, spec, cluster_id, api_key=api_key)
         print(f'Updated deployment: {deployment["id"]}')
     except TransportQueryError as error:
         logger.error(error.errors[0]['message'])
     except Exception as error:
         print(error)
         logger.error(f'There was an error, please try again')
 
+
 @deployments.command("list", help="List deployments")
 @api_key_option
 @click.pass_context
 def list_deployments_command(ctx, api_key):
     try:
-        deployments = list_deployments(api_key=api_key)
+        deployments = gradient_deployments.list_deployments(api_key=api_key)
         if len(deployments) == 0:
             print('No deployments found')
             return
@@ -167,11 +167,12 @@ def list_deployments_command(ctx, api_key):
 @click.pass_context
 def get_deployment_command(ctx, api_key, id):
     try:
-        deployment = get_deployment(id, api_key=api_key)
+        deployment = gradient_deployments.get_deployment(id, api_key=api_key)
         if deployment['deployment'] is None:
             print('Deployment not found')
         else:
-            print(json.dumps(formatted_graphql(deployment['deployment']), indent=4))
+            print(json.dumps(formatted_graphql(
+                deployment['deployment']), indent=4))
     except TransportQueryError as error:
         logger.error(error.errors[0]['message'])
     except Exception as error:
@@ -191,8 +192,7 @@ def get_deployment_command(ctx, api_key, id):
 @click.pass_context
 def delete_deployment_command(ctx, api_key, id):
     try:
-        deployment = delete_deployment(id, api_key=api_key)
-
+        deployment = gradient_deployments.delete_deployment(id, api_key=api_key)
         if deployment is None:
             print('Deployment not found')
         else: