monitorstack/tests/__init__.py

65 lines
1.7 KiB
Python

#!/usr/bin/env python
# Copyright 2017, Major Hayden <major@mhtx.net>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""This an __init__.py."""
import json
import os
from click.testing import CliRunner
from monitorstack import utils
from monitorstack.cli import cli
def read_config():
"""Load the test config file.
:returns: dict
"""
os_config_file = os.path.expanduser(
os.path.abspath(
os.path.dirname(__file__) + '/files/test-openstack.ini'
)
)
return utils.read_config(os_config_file)
def runner(module, extra_args=None):
"""Run click CLI tests.
:param module: Name of module to run.
:type module: str
:param extra_args: List of extra arguments to pass to the CLI.
:type extra_args: list
:returns: dict || object
"""
_runner = CliRunner()
args = [
'-f', 'json',
module
]
if extra_args:
args.extend(extra_args)
result = _runner.invoke(cli, args)
try:
return json.loads(result.output)
except ValueError:
return result
def fake_version_info(major, minor, serial):
"""Return tuple for fake python version info."""
return major, minor, serial