test-bots

#!/usr/bin/env python3

# This file is part of Cockpit.
#
# Copyright (C) 2017 Red Hat, Inc.
#
# Cockpit is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation; either version 2.1 of the License, or
# (at your option) any later version.
#
# Cockpit is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Cockpit; If not, see <http://www.gnu.org/licenses/>.

import glob
import os
import sys
import unittest

from importlib.machinery import SourceFileLoader

sys.dont_write_bytecode = True

BOTS_DIR = os.path.dirname(__file__)


def main():
    loader = unittest.TestLoader()
    suite = unittest.TestSuite()
    for filename in glob.glob(os.path.join(BOTS_DIR, "task", "test-*")):
        name = os.path.basename(filename)
        sys.path[0] = os.path.dirname(filename)
        module = SourceFileLoader(name.replace("-", "_"), filename).load_module()
        suite.addTest(loader.loadTestsFromModule(module))

    runner = unittest.TextTestRunner()
    result = runner.run(suite)
    if result.wasSuccessful():
        return 0
    return 1


if __name__ == '__main__':
    sys.exit(main())