diff options
Diffstat (limited to 'qtip/cli/commands/cmd_perftest.py')
-rw-r--r-- | qtip/cli/commands/cmd_perftest.py | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/qtip/cli/commands/cmd_perftest.py b/qtip/cli/commands/cmd_perftest.py new file mode 100644 index 00000000..5cfe4110 --- /dev/null +++ b/qtip/cli/commands/cmd_perftest.py @@ -0,0 +1,57 @@ +############################################################################## +# Copyright (c) 2016 ZTE Corp and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +from prettytable import PrettyTable +import yaml +import click +import os +from qtip.cli import helper + + +class PerfTest: + + def __init__(self): + self.path = os.path.join(helper.fetch_root(), 'perftest/summary') + + def list(self): + table = PrettyTable(["Name", "Description"]) + table.align = 'l' + with open(self.path) as tests: + line = tests.read() + data = yaml.safe_load(line)['test_cases'] + for i in range(0, len(data)): + points = data[i] + table.add_row([points['name'], points['description']]) + click.echo(table) + + def run(self): + click.echo("Run a perftest") + + +@click.group() +def cli(): + pass + + +@cli.group() +@click.pass_context +def perftest(ctx): + pass + +_perftest = PerfTest() + + +@perftest.command("list", help="Lists all perftest benchmarks.") +def list(): + _perftest.list() + + +@perftest.command("run", help="Executes a single perftest benchmark.") +def execute(): + _perftest.run() |