Test ghe-restore restoring settings with -c or on unconfigured vms

Author: rtomayko

test/test-ghe-restore.sh

@@ -45,7 +45,7 @@ echo "fake ghe-export-settings data" > "$GHE_DATA_DIR/current/settings.json"
 echo "fake license data" > "$GHE_DATA_DIR/current/enterprise.ghl"
 echo "rsync" > "$GHE_DATA_DIR/current/strategy"
 
-begin_test "ghe-restore"
+begin_test "ghe-restore into unconfigured vm"
 (
     set -e
     rm -rf "$GHE_REMOTE_DATA_DIR"
@@ -79,6 +79,85 @@ begin_test "ghe-restore"
 )
 end_test
 
+begin_test "ghe-restore into configured vm"
+(
+    set -e
+    rm -rf "$GHE_REMOTE_DATA_DIR"
+    setup_remote_metadata
+
+    # create settings file -- used to determine if instance has been configured.
+    touch "$GHE_REMOTE_DATA_DIR/enterprise/dna.json"
+
+    # set restore host environ var
+    GHE_RESTORE_HOST=127.0.0.1
+    export GHE_RESTORE_HOST
+
+    # run ghe-restore and write output to file for asserting against
+    ghe-restore -v > "$TRASHDIR/restore-out"
+    cat "$TRASHDIR/restore-out"
+
+    # verify connect to right host
+    grep -q "Connect 127.0.0.1 OK" "$TRASHDIR/restore-out"
+
+    # verify all import scripts were run
+    grep -q "alice/index.html" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-mysql data" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-redis data" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-authorized-keys data" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-ssh-host-keys data" "$TRASHDIR/restore-out"
+    grep -q "ghe-import-es-indices" "$TRASHDIR/restore-out"
+
+    # verify settings import was *not* run due to instance already being
+    # configured.
+    ! grep -q "fake ghe-export-settings data" "$TRASHDIR/restore-out"
+
+    # verify all repository data was transferred to the restore location
+    diff -ru "$GHE_DATA_DIR/current/repositories" "$GHE_REMOTE_DATA_USER_DIR/repositories"
+
+    # verify all pages data was transferred to the restore location
+    diff -ru "$GHE_DATA_DIR/current/pages" "$GHE_REMOTE_DATA_USER_DIR/pages"
+)
+end_test
+
+begin_test "ghe-restore -c into configured vm"
+(
+    set -e
+    rm -rf "$GHE_REMOTE_DATA_DIR"
+    setup_remote_metadata
+
+    # create settings file -- used to determine if instance has been configured.
+    touch "$GHE_REMOTE_DATA_DIR/enterprise/dna.json"
+
+    # set restore host environ var
+    GHE_RESTORE_HOST=127.0.0.1
+    export GHE_RESTORE_HOST
+
+    # run ghe-restore and write output to file for asserting against
+    ghe-restore -v -c > "$TRASHDIR/restore-out"
+    cat "$TRASHDIR/restore-out"
+
+    # verify connect to right host
+    grep -q "Connect 127.0.0.1 OK" "$TRASHDIR/restore-out"
+
+    # verify all import scripts were run
+    grep -q "alice/index.html" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-mysql data" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-redis data" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-authorized-keys data" "$TRASHDIR/restore-out"
+    grep -q "fake ghe-export-ssh-host-keys data" "$TRASHDIR/restore-out"
+    grep -q "ghe-import-es-indices" "$TRASHDIR/restore-out"
+
+    # verify settings were imported
+    grep -q "fake ghe-export-settings data" "$TRASHDIR/restore-out"
+
+    # verify all repository data was transferred to the restore location
+    diff -ru "$GHE_DATA_DIR/current/repositories" "$GHE_REMOTE_DATA_USER_DIR/repositories"
+
+    # verify all pages data was transferred to the restore location
+    diff -ru "$GHE_DATA_DIR/current/pages" "$GHE_REMOTE_DATA_USER_DIR/pages"
+)
+end_test
+
 begin_test "ghe-restore with host arg"
 (
     set -e