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
|
# THIS FILE IS PART OF THE CYLC WORKFLOW ENGINE.
# 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/>.
"""Test the "cylc dump" command."""
import pytest
from cylc.flow.option_parsers import (
Options,
)
from cylc.flow.scripts.dump import (
dump,
get_option_parser,
)
DumpOptions = Options(get_option_parser())
async def test_dump_tasks(flow, scheduler, start):
"""It should show n=0 tasks.
See: https://github.com/cylc/cylc-flow/pull/5600
"""
id_ = flow({
'scheduler': {
'allow implicit tasks': 'true',
},
'scheduling': {
'graph': {
'R1': 'a => b => c',
},
},
})
schd = scheduler(id_)
async with start(schd):
# schd.release_tasks_to_run()
await schd.update_data_structure()
ret = []
await dump(
id_,
DumpOptions(disp_form='tasks', legacy_format=True),
write=ret.append
)
assert ret == ['a, 1, waiting, not-held, queued, not-runahead']
@pytest.mark.parametrize(
'attributes_bool, flow_nums, dump_str',
[
pytest.param(
True,
[1, 2],
'1/a:waiting (held,queued,runahead) flows=[1,2]',
id='1'
),
pytest.param(
False,
[1, 2],
'1/a:waiting',
id='2'
)
]
)
async def test_dump_format(
flow, scheduler, start, attributes_bool, flow_nums, dump_str
):
"""Check the new "cylc dump" output format, i.e. task IDs.
See: https://github.com/cylc/cylc-flow/pull/6440
"""
id_ = flow({
'scheduler': {
'allow implicit tasks': 'true',
},
'scheduling': {
'graph': {
'R1': 'a',
},
},
})
schd = scheduler(id_)
async with start(schd):
[itask] = schd.pool.get_tasks()
itask.state_reset(
is_held=attributes_bool,
is_runahead=attributes_bool,
is_queued=attributes_bool
)
itask.flow_nums = set(flow_nums)
schd.pool.data_store_mgr.delta_task_held(
itask.tdef.name, itask.point, itask.state.is_held)
schd.pool.data_store_mgr.delta_task_state(itask)
schd.pool.data_store_mgr.delta_task_flow_nums(itask)
await schd.update_data_structure()
ret = []
await dump(
id_,
DumpOptions(disp_form='tasks', show_flows=attributes_bool),
write=ret.append
)
assert ret == [dump_str]
|
