1 files changed, 49 insertions, 0 deletions
diff --git a/tests/testsuites/omprog-restart-terminated-bin.sh b/tests/testsuites/omprog-restart-terminated-bin.sh
new file mode 100755
index 0000000..2c2b2cc
--- /dev/null
+++ b/tests/testsuites/omprog-restart-terminated-bin.sh
@@ -0,0 +1,49 @@
+#!/bin/bash
+
+outfile=$RSYSLOG_OUT_LOG
+terminate=false
+
+function handle_sigusr1 {
+    echo "Received SIGUSR1, will terminate after the next message" >> $outfile
+    >&2 echo "[stderr] Received SIGUSR1, will terminate after the next message"
+    terminate=true
+}
+trap "handle_sigusr1" SIGUSR1
+
+function handle_sigterm {
+    echo "Received SIGTERM, terminating" >> $outfile
+    >&2 echo "[stderr] Received SIGTERM, terminating"
+    exit 1
+}
+trap "handle_sigterm" SIGTERM
+
+echo "Starting" >> $outfile
+
+# Write also to stderr (useful for testing the 'output' setting)
+>&2 echo "[stderr] Starting"
+
+# Tell rsyslog we are ready to start processing messages
+echo "OK"
+
+read log_line
+while [[ -n "$log_line" ]]; do
+    echo "Received $log_line" >> $outfile
+    >&2 echo "[stderr] Received $log_line"
+
+    if [[ $terminate == true ]]; then
+        # Terminate prematurely by closing pipe, without confirming the message
+        echo "Terminating without confirming the last message" >> $outfile
+        >&2 echo "[stderr] Terminating without confirming the last message"
+        exit 0
+    fi
+
+    # Tell rsyslog we are ready to process the next message
+    echo "OK"
+
+    read log_line
+done
+
+echo "Terminating normally" >> $outfile
+>&2 echo "[stderr] Terminating normally"
+
+exit 0