# This file is part of cloud-init. See LICENSE file for license information.

"""Run functions."""

import os

from tests.cloud_tests import bddeb, collect, util, verify


def tree_collect(args):
    """Collect data using deb build from current tree.

    @param args: cmdline args
    @return_value: fail count
    """
    failed = 0
    tmpdir = util.TempDir(tmpdir=args.data_dir, preserve=args.preserve_data)

    with tmpdir as data_dir:
        args.data_dir = data_dir
        args.deb = os.path.join(tmpdir.tmpdir, 'cloud-init_all.deb')
        try:
            failed += bddeb.bddeb(args)
            failed += collect.collect(args)
        except Exception:
            failed += 1
            raise

    return failed


def tree_run(args):
    """Run test suite using deb build from current tree.

    @param args: cmdline args
    @return_value: fail count
    """
    failed = 0
    tmpdir = util.TempDir(tmpdir=args.data_dir, preserve=args.preserve_data)

    with tmpdir as data_dir:
        args.data_dir = data_dir
        args.deb = os.path.join(tmpdir.tmpdir, 'cloud-init_all.deb')
        try:
            failed += bddeb.bddeb(args)
            failed += collect.collect(args)
            failed += verify.verify(args)
        except Exception:
            failed += 1
            raise

    return failed


def run(args):
    """Run test suite.

    @param args: cmdline args
    @return_value: fail count
    """
    failed = 0
    tmpdir = util.TempDir(tmpdir=args.data_dir, preserve=args.preserve_data)

    with tmpdir as data_dir:
        args.data_dir = data_dir
        try:
            failed += collect.collect(args)
            failed += verify.verify(args)
        except Exception:
            failed += 1
            raise

    return failed

# vi: ts=4 expandtab