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
117
118
119
|
# Copyright (C) NIWA & British Crown (Met Office) & Contributors.
#
# 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/>.
#
# Tests for the get_host_from_platform lookup.
import pytest
from cylc.flow.exceptions import NoHostsError, NoPlatformsError
from cylc.flow.platforms import get_host_from_platform, get_platform_from_group
from cylc.flow.exceptions import CylcError
TEST_PLATFORM = {
'name': 'Elephant',
'hosts': ['nellie', 'dumbo', 'jumbo'],
'selection': {'method': 'definition order'}
}
TEST_GROUP = {
'platforms': ['one', 'two', 'three'],
'selection': {'method': 'definition order'}
}
@pytest.mark.parametrize(
'badhosts, expect',
[
pytest.param(None, 'nellie'),
pytest.param({'nellie', 'dumbo'}, 'jumbo')
]
)
def test_get_host_from_platform(badhosts, expect):
platform = TEST_PLATFORM
assert get_host_from_platform(platform, badhosts) == expect
def test_get_host_from_platform_fails_no_goodhosts():
platform = TEST_PLATFORM
with pytest.raises(NoHostsError) as err:
get_host_from_platform(platform, {'nellie', 'dumbo', 'jumbo'})
assert err.exconly() == (
'cylc.flow.exceptions.NoHostsError: '
'Unable to find valid host for Elephant'
)
def test_get_host_from_platform_fails_bad_method():
platform = TEST_PLATFORM.copy()
platform['selection']['method'] = 'roulette'
with pytest.raises(CylcError) as err:
get_host_from_platform(platform, {'Elephant'})
assert err.exconly() == (
'cylc.flow.exceptions.CylcError: method "roulette" is not a '
'supported host selection method.'
)
@pytest.mark.parametrize(
'expect, bad_hosts',
[
pytest.param(
'one', None, id='No bad_hosts get platform one'),
pytest.param(
'three', {'foo', 'bar'},
id='All platforms bad except platform three')
]
)
def test_get_platform_from_group(monkeypatch, expect, bad_hosts):
def get_plat(name):
if name == 'three':
return {'hosts': {'foo', 'bar', 'baz'}}
else:
return {'hosts': {'foo', 'bar'}}
monkeypatch.setattr('cylc.flow.platforms.platform_from_name', get_plat)
platform = get_platform_from_group(TEST_GROUP, 'mygroup_name', bad_hosts)
assert platform == expect
def test_get_platform_from_group_fails_no_goodhosts(monkeypatch):
monkeypatch.setattr(
'cylc.flow.platforms.platform_from_name',
lambda _: {'hosts': {'foo', 'bar', 'baz'}},
)
with pytest.raises(NoPlatformsError) as err:
get_platform_from_group(
TEST_GROUP, 'mygroup_name', {'foo', 'bar', 'baz'}
)
assert err.exconly() == (
'cylc.flow.exceptions.NoPlatformsError: '
'Unable to find a platform from group mygroup_name.'
)
def test_get_platform_from_group_fails_bad_method(monkeypatch):
monkeypatch.setattr(
'cylc.flow.platforms.platform_from_name',
lambda _: {'hosts': {'foo', 'bar', 'baz'}},
)
group = TEST_GROUP.copy()
group['selection']['method'] = 'roulette'
with pytest.raises(CylcError) as err:
get_platform_from_group(group, {'foo'})
assert err.exconly() == (
'cylc.flow.exceptions.CylcError: "roulette" is not a '
'supported platform selection method.'
)
|
