functest/cli/commands/cli_env.py

   1 #!/usr/bin/env python
   2 #
   3 # jose.lausuch@ericsson.com
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9
  10 import os
  11
  12 import click
  13 import git
  14
  15 import functest.utils.functest_utils as ft_utils
  16
  17 ENV_FILE = "/home/opnfv/functest/conf/env_active"
  18 FUNCTEST_REPO = ft_utils.FUNCTEST_REPO
  19
  20
  21 class CliEnv:
  22
  23     def __init__(self):
  24         pass
  25
  26     def prepare(self):
  27         if self.status(verbose=False) == 0:
  28             answer = raw_input("It seems that the environment has been "
  29                                "already prepared. Do you want to do "
  30                                "it again? [y|n]\n")
  31             while True:
  32                 if answer.lower() in ["y", "yes"]:
  33                     os.remove(ENV_FILE)
  34                     break
  35                 elif answer.lower() in ["n", "no"]:
  36                     return
  37                 else:
  38                     answer = raw_input("Invalid answer. Please type [y|n]\n")
  39
  40         cmd = ("python %s/functest/ci/prepare_env.py start" % FUNCTEST_REPO)
  41         ft_utils.execute_command(cmd)
  42
  43     def show(self):
  44         CI_INSTALLER_TYPE = os.getenv('INSTALLER_TYPE')
  45         if CI_INSTALLER_TYPE is None:
  46             CI_INSTALLER_TYPE = "Unknown"
  47         CI_INSTALLER_IP = os.getenv('INSTALLER_IP')
  48         if CI_INSTALLER_IP is None:
  49             CI_INSTALLER_IP = "Unknown"
  50         CI_INSTALLER = ("%s, %s" % (CI_INSTALLER_TYPE, CI_INSTALLER_IP))
  51
  52         CI_SCENARIO = os.getenv('DEPLOY_SCENARIO')
  53         if CI_SCENARIO is None:
  54             CI_SCENARIO = "Unknown"
  55
  56         CI_NODE = os.getenv('NODE_NAME')
  57         if CI_NODE is None:
  58             CI_NODE = "Unknown"
  59
  60         repo = git.Repo(FUNCTEST_REPO)
  61         branch = repo.head.reference
  62         GIT_BRANCH = branch.name
  63         GIT_HASH = branch.commit.hexsha
  64
  65         CI_BUILD_TAG = os.getenv('BUILD_TAG')
  66         if CI_BUILD_TAG is not None:
  67             CI_BUILD_TAG = CI_BUILD_TAG.lstrip(
  68                 "jenkins-").lstrip("functest").lstrip("-")
  69
  70         CI_DEBUG = os.getenv('CI_DEBUG')
  71         if CI_DEBUG is None:
  72             CI_DEBUG = "false"
  73
  74         STATUS = "not ready"
  75         if self.status(verbose=False) == 0:
  76             STATUS = "ready"
  77
  78         click.echo("+======================================================+")
  79         click.echo("| Functest Environment info                            |")
  80         click.echo("+======================================================+")
  81         click.echo("|  INSTALLER: %s|" % CI_INSTALLER.ljust(41))
  82         click.echo("|   SCENARIO: %s|" % CI_SCENARIO.ljust(41))
  83         click.echo("|        POD: %s|" % CI_NODE.ljust(41))
  84         click.echo("| GIT BRACNH: %s|" % GIT_BRANCH.ljust(41))
  85         click.echo("|   GIT HASH: %s|" % GIT_HASH.ljust(41))
  86         if CI_BUILD_TAG:
  87             click.echo("|  BUILD TAG: %s|" % CI_BUILD_TAG.ljust(41))
  88         click.echo("| DEBUG FLAG: %s|" % CI_DEBUG.ljust(41))
  89         click.echo("+------------------------------------------------------+")
  90         click.echo("|     STATUS: %s|" % STATUS.ljust(41))
  91         click.echo("+------------------------------------------------------+")
  92         click.echo("")
  93
  94     def status(self, verbose=True):
  95         ret_val = 0
  96         if not os.path.isfile(ENV_FILE):
  97             if verbose:
  98                 click.echo("Functest environment is not installed.\n")
  99             ret_val = 1
 100         elif verbose:
 101             click.echo("Functest environment ready to run tests.\n")
 102
 103         return ret_val