summaryrefslogtreecommitdiffstats
path: root/ipatests/pytest_plugins/integration.py
blob: 5a0e46845d5b997002411faa126c25d4ba5b0e29 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# Authors:
#   Petr Viktorin <pviktori@redhat.com>
#
# Copyright (C) 2011  Red Hat
# see file 'COPYING' for use and warranty information
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""Pytest plugin for IPA Integration tests"""

import pytest

from ipatests.test_integration.config import get_global_config


@pytest.yield_fixture(scope='class')
def integration_config(request):
    cls = request.cls

    def get_resources(resource_container, resource_str, num_needed):
        if len(resource_container) < num_needed:
            raise pytest.skip(
                'Not enough %s available (have %s, need %s)' %
                (resource_str, len(resource_container), num_needed))
        return resource_container[:num_needed]

    config = get_global_config()
    if not config.domains:
        raise pytest.skip('Integration testing not configured')

    cls.logs_to_collect = {}

    cls.domain = config.domains[0]

    # Check that we have enough resources available
    cls.master = cls.domain.master
    cls.replicas = get_resources(cls.domain.replicas, 'replicas',
                                    cls.num_replicas)
    cls.clients = get_resources(cls.domain.clients, 'clients',
                                cls.num_clients)
    cls.ad_domains = get_resources(config.ad_domains, 'AD domains',
                                    cls.num_ad_domains)

    # Check that we have all required extra hosts at our disposal
    available_extra_roles = [role for domain in cls.get_domains()
                                    for role in domain.extra_roles]
    missing_extra_roles = list(set(cls.required_extra_roles) -
                                    set(available_extra_roles))

    if missing_extra_roles:
        raise pytest.skip("Not all required extra hosts available, "
                          "missing: %s, available: %s"
                          % (missing_extra_roles,
                             available_extra_roles))

    for host in cls.get_all_hosts():
        host.add_log_collector(cls.collect_log)
        cls.prepare_host(host)

    try:
        cls.install()
    except:
        cls.uninstall()
        raise

    yield config

    for host in cls.get_all_hosts():
        host.remove_log_collector(collect_log)

    try:
        cls.uninstall()
    finally:
        del cls.master
        del cls.replicas
        del cls.clients
        del cls.ad_domains
        del cls.domain