azure/aria/aria-extension-cloudify/src/aria/tests/cli/runner.py

   1 # Licensed to the Apache Software Foundation (ASF) under one or more
   2 # contributor license agreements.  See the NOTICE file distributed with
   3 # this work for additional information regarding copyright ownership.
   4 # The ASF licenses this file to You under the Apache License, Version 2.0
   5 # (the "License"); you may not use this file except in compliance with
   6 # the License.  You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 import click.testing
  17
  18 import aria.cli.commands as commands
  19
  20
  21 def invoke(command_string):
  22     command_list = command_string.split()
  23     command, sub, args = command_list[0], command_list[1], command_list[2:]
  24     runner = click.testing.CliRunner()
  25     outcome = runner.invoke(getattr(
  26         getattr(commands, command), sub), args)
  27     return outcome