From f94e421f4ec9bb87723ea876876b28598c053461 Mon Sep 17 00:00:00 2001
From: William Wernert <william.wernert@securityonionsolutions.com>
Date: Mon, 4 Jan 2021 14:46:48 -0500
Subject: [PATCH] [fix] Fix automation compatibility

---
 setup/so-setup    | 6 ++----
 setup/so-whiptail | 7 +++++++
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/setup/so-setup b/setup/so-setup
index bede7990d..8b8f99b01 100755
--- a/setup/so-setup
+++ b/setup/so-setup
@@ -175,7 +175,7 @@ if ! [[ -f $install_opt_file ]]; then
 		echo "User cancelled setup." | tee -a "$setup_log"
 		whiptail_cancel
 	fi
-	if [[ $setup_type == 'iso' ]]; then
+	if [[ $setup_type == 'iso' ]] && [ "$automated" == no ]; then
 		whiptail_first_menu_iso
 		if [[ $option == "Configure Network" ]]; then
 			network_init_whiptail
@@ -184,9 +184,7 @@ if ! [[ -f $install_opt_file ]]; then
 			printf '%s\n' \
 				"MNIC=$MNIC" \
 				"HOSTNAME=$HOSTNAME" > "$net_init_file"
-			whiptail --title "Security Onion Setup" \
-				--msgbox "Successfully set up networking, setup will now exit." 7 75
-			exit 0
+			whiptail_net_setup_complete
 		else
 			whiptail_install_type
 		fi
diff --git a/setup/so-whiptail b/setup/so-whiptail
index b034ab679..6af5b701f 100755
--- a/setup/so-whiptail
+++ b/setup/so-whiptail
@@ -794,6 +794,13 @@ whiptail_management_interface_setup() {
 	whiptail_check_exitstatus $exitstatus
 }
 
+whiptail_net_setup_complete() {
+	[ -n "$TESTING" ] && return
+
+	whiptail --title "Security Onion Setup" \
+		--msgbox "Successfully set up networking, setup will now exit." 7 75
+	exit 0
+}
 
 
 whiptail_management_server() {