From aa5916e3b6e44c370104d2e3b11d3355c9e1f185 Mon Sep 17 00:00:00 2001
From: Claudia Beresford <claudiaberesford@gmail.com>
Date: Fri, 12 Nov 2021 14:26:20 +0000
Subject: [PATCH] Format run.py

---
 test/tools/run.py | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/test/tools/run.py b/test/tools/run.py
index 183f781d..443efeee 100755
--- a/test/tools/run.py
+++ b/test/tools/run.py
@@ -10,6 +10,7 @@
 import string
 from os.path import dirname, abspath
 
+
 @click.group()
 def cli():
     """
@@ -17,6 +18,7 @@ def cli():
     """
     pass
 
+
 @cli.command()
 @click.option('-c', '--config-file', type=str, help='Path to configuration file')
 @click.option('-o', '--org-id', type=str, help='Equinix organisation id (required)')
@@ -38,7 +40,8 @@ def run_e2e(config_file, org_id, project_name, ssh_key_name, device_name, skip_t
             click.echo(str(e))
             sys.exit()
 
-    cfg.set_run_flag_config(org_id, project_name, ssh_key_name, device_name, skip_teardown)
+    cfg.set_run_flag_config(org_id, project_name,
+                            ssh_key_name, device_name, skip_teardown)
     try:
         cfg.validate_run()
     except ValueError as e:
@@ -46,12 +49,15 @@ def run_e2e(config_file, org_id, project_name, ssh_key_name, device_name, skip_t
         sys.exit()
 
     if cfg['device']['id'] == None:
-        click.echo(f"Running e2e tests. Will create project `{cfg['project_name']}`, ssh_key `{cfg['device']['ssh_key_name']}` and device `{cfg['device']['name']}`")
-        click.echo("Note: this will create and bootstrap a new device in Equinix and may take some time")
+        click.echo(
+            f"Running e2e tests. Will create project `{cfg['project_name']}`, ssh_key `{cfg['device']['ssh_key_name']}` and device `{cfg['device']['name']}`")
+        click.echo(
+            "Note: this will create and bootstrap a new device in Equinix and may take some time")
     else:
         tool_dir = dirname(abspath(__file__))
         if os.path.exists(tool_dir+"/private.key") != True:
-            click.echo(f"`private.key` file must be saved at `{tool_dir}` when `--existing-device-id` flag set")
+            click.echo(
+                f"`private.key` file must be saved at `{tool_dir}` when `--existing-device-id` flag set")
             sys.exit()
         click.echo("running e2e tests using device `{cfg['device']['id']}`")
 
@@ -64,6 +70,7 @@ def run_e2e(config_file, org_id, project_name, ssh_key_name, device_name, skip_t
         dev_id, dev_ip = runner.device_details()
         click.echo(f"Device `{dev_id}` left alive for debugging. Use with `--config-file` setting `device.id` to re-run tests. SSH command `ssh -i test/tools/private.key root@{dev_ip}`. Teardown device with `teardown-device` command.")
 
+
 @cli.command()
 @click.option('-c', '--config-file', type=str, help='Path to configuration file')
 @click.option('-o', '--org-id', type=str, help='Equinix organisation id (required)')
@@ -92,12 +99,15 @@ def create_device(config_file, org_id, project_id, ssh_key_name, device_name, us
         click.echo(str(e))
         sys.exit()
 
-    click.echo(f"Creating device {cfg['device']['name']} with config {cfg['device']}")
+    click.echo(
+        f"Creating device {cfg['device']['name']} with config {cfg['device']}")
 
     welder = Welder(token, cfg)
     ip = welder.create_all()
 
-    click.echo(f"Device {cfg['device']['name']} created. SSH command `ssh -i test/tools/private.key root@{ip}`. Run tests with `run-e2e`. Teardown with `delete-device`.")
+    click.echo(
+        f"Device {cfg['device']['name']} created. SSH command `ssh -i test/tools/private.key root@{ip}`. Run tests with `run-e2e`. Teardown with `delete-device`.")
+
 
 @cli.command()
 @click.option('-o', '--org-id', type=str, help='Equinix organisation id (required)')
@@ -119,5 +129,6 @@ def delete_device(org_id, device_id):
     welder = Welder(token, org_id)
     welder.delete_device(device_id)
 
+
 if __name__ == "__main__":
     cli()