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
120
121
122
123
124
125
126
127
128
129
130
|
#
# Copyright 2009- ECMWF.
#
# This software is licensed under the terms of the Apache Licence version 2.0
# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
# In applying this licence, ECMWF does not waive the privileges and immunities
# granted to it by virtue of its status as an intergovernmental organisation
# nor does it submit to any jurisdiction.
#
# This test ensures that the simulator performs as expected.
import os, fnmatch
import ecflow
import ecflow_test_util as Test
def simulate_defs_with_time():
# suite test_time_series
# family family
# task t1
# time 00:00 23:00 04:00 # should run 6 times 00:00 04:00 08:00 12:00 16:00 20:00
# endfamily
# endsuite
print("simulate_defs_with_time()")
defs = ecflow.Defs()
suite = defs.add_suite("test_time_series")
clock = ecflow.Clock(1, 1, 2011, False) # day,month, year, hybrid make test start at midnight, otherwise current time used
suite.add_clock(clock)
family = suite.add_family("family")
task = family.add_task("t1")
ts = ecflow.TimeSeries(ecflow.TimeSlot(0, 0), ecflow.TimeSlot(23, 0), ecflow.TimeSlot(4, 0), True)
task.add_time(ecflow.Time(ts))
task.add_verify(ecflow.Verify(ecflow.State.complete, 6)) # expect task to complete 6 times
theResult = defs.simulate()
assert len(theResult) == 0, "Expected simulation to return without any errors, but found:\n" + theResult
print(" simple check for state change time")
print(" iso_extended:", task.get_state_change_time())
print(" iso_extended:", task.get_state_change_time("is0_extended"))
print(" iso :", task.get_state_change_time("iso"))
print(" simple :", task.get_state_change_time("simple"))
print(" rubbish :", task.get_state_change_time("rubbish"))
os.remove("test_time_series.def.log")
def simulate_deadlock():
# This simulation is expected to fail, since we have a deadlock/ race condition
print("simulate_deadlock")
# suite dead_lock
# family family
# task t1
# trigger t2 == complete
# task t2
# trigger t1 == complete
# endfamily
# endsuite
defs = ecflow.Defs()
suite = defs.add_suite("dead_lock")
fam = suite.add_family("family")
fam.add_task("t1").add_trigger("t2 == complete")
fam.add_task("t2").add_trigger("t1 == complete")
theResult = defs.simulate()
assert len(theResult) != 0, "Expected simulation to return errors, but found none"
print(theResult)
os.remove("dead_lock.def.log")
os.remove("defs.depth")
os.remove("defs.flat")
def test_time_series():
# suite suite
# clock real <sunday>
# family family
# task t1
# time 00:30 23:59 04:00 # should run 6 times 00:30 4:30 8:30 12:30 16:30 20:30
# endfamily
# endsuite
print("Simulator:: ...test_time_series")
# Initialise real clock on a Monday, such that task should _only_ run
# on Monday since we are using a hybrid clock
clock = ecflow.Clock(12, 10, 2009, False) # day,month, year, hybrid 12 October 2009 was a Monday
theDefs = ecflow.Defs()
suite = theDefs.add_suite("test_time_series")
suite.add_clock(clock)
fam = suite.add_family("family")
task = fam.add_task("t")
task.add_time("00:30 23:59 04:00");
task.add_verify(ecflow.Verify(ecflow.State.complete, 6)) # expect task to complete 6 times
theResult = theDefs.simulate()
assert len(theResult) == 0, "Expected simulation to return without any errors, but found:\n" + theResult
os.remove("test_time_series.def.log")
if __name__ == "__main__":
Test.print_test_start(os.path.basename(__file__))
simulate_defs_with_time()
simulate_deadlock()
test_time_series()
# traverse the CSim test data, make sure python simulation matches c++
workspace_dir = ecflow.File.source_dir()
csim_test_data = workspace_dir + "/CSim/test/data/good_defs"
print(csim_test_data)
for path in Test.all_files(csim_test_data, '*.def'):
print(path)
theDefs = ecflow.Defs(path)
theResult = theDefs.simulate()
assert len(theResult) == 0, "Expected simulation to return without any errors, but found:\n" + theResult
# remove generated .log files
for path in Test.all_files(".", '*.log'):
os.remove(path)
print("All Tests pass")
|
