From adf046cf7917db3603b154ae364edfd5ffde5579 Mon Sep 17 00:00:00 2001
From: Oliver Sanders <oliver.sanders@metoffice.gov.uk>
Date: Tue, 27 Mar 2018 12:29:55 +0100
Subject: [PATCH] Add test from #2520.

---
 tests/job-poll/03-execution-time-limit.t      | 85 +++++++++++++++++++
 .../03-execution-time-limit/reference.log     |  3 +
 .../job-poll/03-execution-time-limit/suite.rc | 20 +++++
 3 files changed, 108 insertions(+)
 create mode 100644 tests/job-poll/03-execution-time-limit.t
 create mode 100644 tests/job-poll/03-execution-time-limit/reference.log
 create mode 100644 tests/job-poll/03-execution-time-limit/suite.rc

diff --git a/tests/job-poll/03-execution-time-limit.t b/tests/job-poll/03-execution-time-limit.t
new file mode 100644
index 00000000000..09360fa5f26
--- /dev/null
+++ b/tests/job-poll/03-execution-time-limit.t
@@ -0,0 +1,85 @@
+#!/bin/bash
+# THIS FILE IS PART OF THE CYLC SUITE ENGINE.
+# Copyright (C) 2008-2018 NIWA
+#
+# 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 execution time limit polling.
+. "$(dirname "${0}")/test_header"
+#-------------------------------------------------------------------------------
+set_test_number 4
+create_test_globalrc '
+[hosts]
+   [[localhost]]
+        task communication method = poll
+        submission polling intervals = PT2S
+        execution polling intervals = PT1M
+        [[[batch systems]]]
+            [[[[background]]]]
+                execution time limit polling intervals = PT5S'
+install_suite "${TEST_NAME_BASE}" "${TEST_NAME_BASE}"
+#-------------------------------------------------------------------------------
+run_ok "${TEST_NAME_BASE}-validate" cylc validate "${SUITE_NAME}"
+suite_run_ok "${TEST_NAME_BASE}-run" \
+    cylc run --reference-test --no-detach "${SUITE_NAME}"
+#-------------------------------------------------------------------------------
+cmp_times () {
+    # Test if the times $1 and $2 are within $3 seconds of eachother.
+    python -c '
+import sys
+from isodatetime.parsers import TimePointParser
+parser = TimePointParser()
+time_1 = parser.parse(sys.argv[1])
+time_2 = parser.parse(sys.argv[2])
+diff = (time_1 - time_2).get_seconds()
+if abs(diff) <= int(sys.argv[3]):
+    sys.exit(0)
+else:
+    sys.exit(1)
+    ' $1 $2 $3
+}
+time_offset () {
+    # Add an ISO8601 duration to an ISO8601 date-time.
+    python -c '
+import sys
+from isodatetime.parsers import TimePointParser, DurationParser
+print TimePointParser().parse(sys.argv[1]) + DurationParser().parse(sys.argv[2])
+    ' $1 $2
+}
+#-------------------------------------------------------------------------------
+LOG_FILE="${SUITE_RUN_DIR}/log/suite/log"
+# Test logging of the "next job poll" message when task starts.
+TEST_NAME="${TEST_NAME_BASE}-log-entry"
+LINE=$(grep -A 1 '[foo.1].*current\:submitted.*started' "${LOG_FILE}" | tail -1)
+run_ok "${TEST_NAME}" grep -q 'health check settings: execution timeout=PT10S' \
+    <<< "${LINE}"
+# Determine poll times.
+PREDICTED_POLL_TIME=$(time_offset \
+    "$(cut -d ' ' -f 1 <<< "${LINE}")" \
+    "$(sed 's/.*execution timeout=\([^,]\+\).*/\1/' <<< "${LINE}")")
+ACTUALL_POLL_TIME=$(sed -n \
+    's/\(.*\) INFO - \[foo.1\] -(current:running) failed (polled).*/\1/p' \
+    "${LOG_FILE}")
+# Test execution timeout polling.
+TEST_NAME="${TEST_NAME_BASE}-poll-time"
+if cmp_times ${PREDICTED_POLL_TIME} ${ACTUALL_POLL_TIME} 1; then
+    ok "${TEST_NAME}"
+else
+    echo "Poll time differs from log entry '${PREDICTED_POLL_TIME} != " \
+        "${ACTUALL_POLL_TIME}'." >&2
+    fail "${TEST_NAME}"
+fi
+#-------------------------------------------------------------------------------
+purge_suite "${SUITE_NAME}"
+exit
diff --git a/tests/job-poll/03-execution-time-limit/reference.log b/tests/job-poll/03-execution-time-limit/reference.log
new file mode 100644
index 00000000000..2613e1abf97
--- /dev/null
+++ b/tests/job-poll/03-execution-time-limit/reference.log
@@ -0,0 +1,3 @@
+2017-12-21T11:22:59Z INFO - Initial point: 1
+2017-12-21T11:22:59Z INFO - Final point: 1
+2017-12-21T11:22:59Z INFO - [foo.1] -triggered off []
diff --git a/tests/job-poll/03-execution-time-limit/suite.rc b/tests/job-poll/03-execution-time-limit/suite.rc
new file mode 100644
index 00000000000..7c79f80cdc4
--- /dev/null
+++ b/tests/job-poll/03-execution-time-limit/suite.rc
@@ -0,0 +1,20 @@
+[cylc]
+    [[events]]
+        timeout = PT1M
+    [[reference test]]
+        expected task failures = foo.1
+[scheduling]
+    [[dependencies]]
+        graph = """
+            foo
+            foo:fail => ! foo
+        """
+[runtime]
+    [[foo]]
+        script = """
+            #rm "$CYLC_TASK_LOG_ROOT.status"  # Disable polling.
+            export PATH=/bin  # Disable task message.
+            sleep 30  # Job must exceed exeuction time limit.
+        """
+        [[[job]]]
+            execution time limit = PT5S