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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#!/usr/bin/env python3
#
# Copyright (C) 2018 Codethink Limited
# Copyright (C) 2019 Bloomberg Finance LLP
#
# This program 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 of the License, or (at your option) any later version.
#
# This library 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 this library. If not, see <http://www.gnu.org/licenses/>.
#
# Authors:
# Tristan Maat <tristan.maat@codethink.co.uk>
#
import os
import pytest
from buildstream.testing import register_repo_kind, sourcetests_collection_hook
from buildstream.testing.integration import integration_cache # pylint: disable=unused-import
from tests.testutils.repo.git import Git
from tests.testutils.repo.bzr import Bzr
from tests.testutils.repo.tar import Tar
from tests.testutils.repo.zip import Zip
#
# This file is loaded by pytest, we use it to add a custom
# `--integration` option to our test suite, and to install
# a session scope fixture.
#
#################################################
# Implement pytest option #
#################################################
def pytest_addoption(parser):
parser.addoption('--integration', action='store_true', default=False,
help='Run integration tests')
parser.addoption('--remote-execution', action='store_true', default=False,
help='Run remote-execution tests only')
def pytest_runtest_setup(item):
# Without --integration: skip tests not marked with 'integration'
if not item.config.getvalue('integration'):
if item.get_closest_marker('integration'):
pytest.skip('skipping integration test')
# With --remote-execution: only run tests marked with 'remoteexecution'
if item.config.getvalue('remote_execution'):
if not item.get_closest_marker('remoteexecution'):
pytest.skip('skipping non remote-execution test')
# Without --remote-execution: skip tests marked with 'remoteexecution'
else:
if item.get_closest_marker('remoteexecution'):
pytest.skip('skipping remote-execution test')
#################################################
# remote_services fixture #
#################################################
#
# This is returned by the `remote_services` fixture
#
class RemoteServices():
def __init__(self, **kwargs):
self.action_service = kwargs.get('action_service')
self.artifact_service = kwargs.get('artifact_service')
self.exec_service = kwargs.get('exec_service')
self.source_service = kwargs.get('source_service')
self.storage_service = kwargs.get('storage_service')
@pytest.fixture(scope='session')
def remote_services(request):
kwargs = {}
# Look for remote services configuration in environment.
if 'ARTIFACT_CACHE_SERVICE' in os.environ:
kwargs['artifact_service'] = os.environ.get('ARTIFACT_CACHE_SERVICE')
if 'REMOTE_EXECUTION_SERVICE' in os.environ:
kwargs['action_service'] = os.environ.get('REMOTE_EXECUTION_SERVICE')
kwargs['exec_service'] = os.environ.get('REMOTE_EXECUTION_SERVICE')
kwargs['storage_service'] = os.environ.get('REMOTE_EXECUTION_SERVICE')
if 'SOURCE_CACHE_SERVICE' in os.environ:
kwargs['source_service'] = os.environ.get('SOURCE_CACHE_SERVICE')
return RemoteServices(**kwargs)
#################################################
# Setup for templated source tests #
#################################################
register_repo_kind('git', Git)
register_repo_kind('bzr', Bzr)
register_repo_kind('tar', Tar)
register_repo_kind('zip', Zip)
# This hook enables pytest to collect the templated source tests from
# buildstream.testing
def pytest_sessionstart(session):
sourcetests_collection_hook(session)
|
