Update test and add test for forced behaviour

Author: lildude

test/test-ghe-restore.sh

@@ -83,6 +83,7 @@ echo "fake ghe-export-ssl-ca-certificates data" > "$GHE_DATA_DIR/current/ssl-ca-
 echo "fake license data" > "$GHE_DATA_DIR/current/enterprise.ghl"
 echo "fake manage password hash data" > "$GHE_DATA_DIR/current/manage-password"
 echo "rsync" > "$GHE_DATA_DIR/current/strategy"
+echo "$GHE_REMOTE_VERSION" >  "$GHE_DATA_DIR/current/version"
 if [ "$GHE_VERSION_MAJOR" -eq 2 ]; then
   touch "$GHE_DATA_DIR/current/es-scan-complete"
 fi
@@ -668,12 +669,38 @@ begin_test "ghe-restore fails when restore 2.9/2.10 snapshot without audit log m
   echo "v2.9.10" > "$GHE_DATA_DIR/current/version"
   rm "$GHE_DATA_DIR/current/es-scan-complete"
 
-  ! output=$(ghe-restore -v -f localhost 2>&1)
+  ! output=$(ghe-restore -v localhost 2>&1)
 
-  echo $output | grep -q "Error: Snapshot must be from GitHub Enterprise v2.9 or v.2.10 after running the"
+  echo $output | grep -q "Error: Snapshot must be from GitHub Enterprise v2.9 or v2.10 after running the"
 
   echo "v2.10.5" > "$GHE_DATA_DIR/current/version"
-  ! output=$(ghe-restore -v -f localhost 2>&1)
+  ! output=$(ghe-restore -v localhost 2>&1)
 
-  echo $output | grep -q "Error: Snapshot must be from GitHub Enterprise v2.9 or v.2.10 after running the"
+  echo $output | grep -q "Error: Snapshot must be from GitHub Enterprise v2.9 or v2.10 after running the"
 )
+end_test
+
+begin_test "ghe-restore force restore of 2.9/2.10 snapshot without audit log migration sentinel file to 2.11"
+(
+  set -e
+
+  # noop if not testing against 2.11
+  if [ "$GHE_VERSION_MAJOR" -le 1 ] || [ "$GHE_VERSION_MINOR" -ne 11 ]; then
+    exit 0
+  fi
+
+  rm -rf "$GHE_REMOTE_ROOT_DIR"
+  setup_remote_metadata
+
+  echo "rsync" > "$GHE_DATA_DIR/current/strategy"
+  echo "v2.9.10" > "$GHE_DATA_DIR/current/version"
+
+  # Create fake remote repositories dir
+  mkdir -p "$GHE_REMOTE_DATA_USER_DIR/repositories"
+
+  ghe-restore -v -f localhost
+
+  echo "v2.10.5" > "$GHE_DATA_DIR/current/version"
+  ghe-restore -v -f localhost
+)
+end_test