From b9c9d92988bc3a40778e53e9894e64fefd7a301c Mon Sep 17 00:00:00 2001
From: Tyler Gillson <tyler.gillson@gmail.com>
Date: Thu, 26 Oct 2023 16:46:02 -0400
Subject: [PATCH] add livenessSeconds & VIP config

Signed-off-by: Tyler Gillson <tyler.gillson@gmail.com>
---
 test/templates/two-node-create.json.tmpl | 2 +-
 test/test-two-node.sh                    | 9 ++++++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/test/templates/two-node-create.json.tmpl b/test/templates/two-node-create.json.tmpl
index 1a05b14..3b9a6f9 100644
--- a/test/templates/two-node-create.json.tmpl
+++ b/test/templates/two-node-create.json.tmpl
@@ -7,7 +7,7 @@
         "spec": {
                 "cloudConfig": {
                         "controlPlaneEndpoint": {
-                                "host": "10.10.142.174",
+                                "host": "_____place_holder_____",
                                 "type": "IP"
                         },
                         "sshKeys": [
diff --git a/test/test-two-node.sh b/test/test-two-node.sh
index ac56f12..4be3ada 100755
--- a/test/test-two-node.sh
+++ b/test/test-two-node.sh
@@ -52,6 +52,7 @@ export DOMAIN=dev.spectrocloud.com
 export PUBLIC_PACK_REPO_UID=<YOUR_PUBLIC_PACK_REPO_UID> # this varies per Palette tenant
 export CLUSTER_NAME=two-node
 export CLUSTER_PROFILE_UID= # if left blank, a cluster profile will be created
+export CLUSTER_VIP= # choose an unassigned VIP
 
 # images
 export OCI_REGISTRY=ttl.sh
@@ -93,6 +94,7 @@ stylus:
   debug: true
   twoNode:
     enabled: true
+    livenessSeconds: 15
 install:
   poweroff: true
 users:
@@ -257,8 +259,13 @@ function prepare_cluster() {
         echo STYLUS_HASH is unset. Please execute build_all and retry.
         exit 1
     fi
+    if nslookup $CLUSTER_VIP >/dev/null; then
+        echo CLUSTER_VIP: $CLUSTER_VIP is allocated. Please retry with an unallocated VIP.
+        exit 1
+    fi
     jq '
       .metadata.name = env.CLUSTER_NAME |
+      .spec.cloudConfig.controlPlaneEndpoint.host = env.CLUSTER_VIP |
       .spec.machinePoolConfig[0].cloudConfig.edgeHosts[0].hostUid = env.HOST_1 |
       .spec.machinePoolConfig[1].cloudConfig.edgeHosts[0].hostUid = env.HOST_2 |
       .spec.profiles[0].uid = env.CLUSTER_PROFILE_UID |
@@ -401,7 +408,7 @@ function main() {
 if [[ $sourced == 1 ]]; then
     set +e
     echo "You can now use any of these functions:"
-    echo
+    echo ""
     grep ^function  ${BASH_SOURCE[0]} | grep -v main | awk '{gsub(/function /,""); gsub(/\(\) \{/,""); print;}'
     echo
 else