From a362b4a5f4bdb0eb848925deec8f176b9feee200 Mon Sep 17 00:00:00 2001
From: ChrisERo <cer95@cornell.edu>
Date: Sat, 8 Jan 2022 16:23:13 -0500
Subject: [PATCH] feat: added tests for this PR that test exechook paired with
 --one-time flag

---
 test_e2e.sh                              | 77 ++++++++++++++++++++++++
 test_exechook_command_fail_with_sleep.sh | 22 +++++++
 test_exechook_command_with_sleep.sh      | 22 +++++++
 3 files changed, 121 insertions(+)
 create mode 100755 test_exechook_command_fail_with_sleep.sh
 create mode 100755 test_exechook_command_with_sleep.sh

diff --git a/test_e2e.sh b/test_e2e.sh
index c69f2f6..a42a77a 100755
--- a/test_e2e.sh
+++ b/test_e2e.sh
@@ -143,6 +143,8 @@ SLOW_GIT_FETCH=/slow_git_fetch.sh
 ASKPASS_GIT=/askpass_git.sh
 EXECHOOK_COMMAND=/test_exechook_command.sh
 EXECHOOK_COMMAND_FAIL=/test_exechook_command_fail.sh
+EXECHOOK_COMMAND_SLEEPY=/test_exechook_command_with_sleep.sh
+EXECHOOK_COMMAND_FAIL_SLEEPY=/test_exechook_command_fail_with_sleep.sh
 RUNLOG="$DIR/runlog.exechook-fail-retry"
 rm -f $RUNLOG
 touch $RUNLOG
@@ -1072,6 +1074,81 @@ function e2e::exechook_fail_retry() {
     fi
 }
 
+##############################################
+# Test exechook-success with GIT_SYNC_ONE_TIME
+##############################################
+function exechook_success_once_helper() {
+    # First sync
+    echo "$FUNCNAME 2" > "$REPO"/file
+    git -C "$REPO" commit -qam "$FUNCNAME 2"
+
+    GIT_SYNC \
+        --period=${2}ms \
+        --one-time \
+        --repo="file://$REPO" \
+        --branch="$MAIN_BRANCH" \
+        --root="$ROOT" \
+        --link="link" \
+        --exechook-command="$EXECHOOK_COMMAND_SLEEPY" \
+        >> "$1" 2>&1 &
+    sleep 7
+    assert_link_exists "$ROOT"/link
+    assert_file_exists "$ROOT"/link/file
+    assert_file_exists "$ROOT"/link/exechook
+    assert_file_exists "$ROOT"/link/link-exechook
+    assert_file_eq "$ROOT"/link/file "$FUNCNAME 2"
+    assert_file_eq "$ROOT"/link/exechook "$FUNCNAME 2"
+    assert_file_eq "$ROOT"/link/link-exechook "$FUNCNAME 2"
+}
+
+function e2e::exechook_success_once() {
+  for i in $(seq 0 50 200); do
+    for j in $(seq 2); do
+      clean_root
+      init_repo
+      exechook_success_once_helper "$1" "$i"
+    done
+  done
+}
+
+##############################################
+# Test exechook-fail with GIT_SYNC_ONE_TIME
+##############################################
+function exechook_fail_once_helper() {
+    cat /dev/null > "$RUNLOG"
+        # First sync - return a failure to ensure that we try again
+        echo "$FUNCNAME 2" > "$REPO"/file
+        git -C "$REPO" commit -qam "$FUNCNAME 2"
+
+        GIT_SYNC \
+            --period="$2"ms \
+            --one-time \
+            --repo="file://$REPO" \
+            --branch="$MAIN_BRANCH" \
+            --root="$ROOT" \
+            --link="link" \
+            --exechook-command="$EXECHOOK_COMMAND_FAIL_SLEEPY" \
+            --exechook-backoff=1s \
+            >> "$1" 2>&1 &
+
+        # Check that exechook was called
+        sleep 10
+        RUNS=$(cat "$RUNLOG" | wc -l)
+        if [[ "$RUNS" < 2 ]]; then
+            fail "exechook called $RUNS times, it should be at least 2"
+        fi
+}
+
+function e2e::exechook_fail_once() {
+  for i in $(seq 0 50 200); do
+    for j in $(seq 2); do
+      clean_root
+      init_repo
+      exechook_fail_once_helper "$1" "$i"
+    done
+  done
+}
+
 ##############################################
 # Test webhook success
 ##############################################
diff --git a/test_exechook_command_fail_with_sleep.sh b/test_exechook_command_fail_with_sleep.sh
new file mode 100755
index 0000000..492e4a1
--- /dev/null
+++ b/test_exechook_command_fail_with_sleep.sh
@@ -0,0 +1,22 @@
+#!/bin/sh
+#
+# Copyright 2021 The Kubernetes Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Use for e2e test of --exechook-command.
+# This option takes no command arguments, so requires a wrapper script.
+
+sleep 4
+date >> /var/log/runs
+exit 1
diff --git a/test_exechook_command_with_sleep.sh b/test_exechook_command_with_sleep.sh
new file mode 100755
index 0000000..c494308
--- /dev/null
+++ b/test_exechook_command_with_sleep.sh
@@ -0,0 +1,22 @@
+#!/bin/sh
+#
+# Copyright 2020 The Kubernetes Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Use for e2e test of --exechook-command.
+# This option takes no command arguments, so requires a wrapper script.
+
+sleep 4
+cat file > exechook
+cat ../link/file > link-exechook