From 38612bcddbad68dc28eafc70bb7b86112d752cc2 Mon Sep 17 00:00:00 2001
From: Jorrit Schaap <schaap@astron.nl>
Date: Fri, 22 Feb 2019 14:40:00 +0000
Subject: [PATCH] COB-49: check for head node

---
 SubSystems/Online_Cobalt/validation/validate          |  3 +++
 .../Online_Cobalt/validation/validation_utils.sh      | 11 +++++++++++
 2 files changed, 14 insertions(+)

diff --git a/SubSystems/Online_Cobalt/validation/validate b/SubSystems/Online_Cobalt/validation/validate
index 1bd9e0409ea..50c77d3ec8c 100755
--- a/SubSystems/Online_Cobalt/validation/validate
+++ b/SubSystems/Online_Cobalt/validation/validate
@@ -105,6 +105,9 @@ do
     $NO_GPU_EXIT_CODE) echo "NO GPU ON THIS NODE"
        NRNOTAPPLICABLE=$((NRNOTAPPLICABLE + 1))
        ;;
+    $NOT_ON_HEAD_EXIT_CODE) echo "CAN ONLY RUN ON HEADNODE"
+       NRNOTAPPLICABLE=$((NRNOTAPPLICABLE + 1))
+       ;;
     *) echo "ERROR (see $LOGFILE)"
        NRFAILURE=$((NRFAILURE + 1))
        NRAPPLICABLE=$((NRAPPLICABLE + 1))
diff --git a/SubSystems/Online_Cobalt/validation/validation_utils.sh b/SubSystems/Online_Cobalt/validation/validation_utils.sh
index 8298e59f0ae..aa3ac6c6a95 100644
--- a/SubSystems/Online_Cobalt/validation/validation_utils.sh
+++ b/SubSystems/Online_Cobalt/validation/validation_utils.sh
@@ -10,6 +10,7 @@ NEED_NON_ROOT_EXIT_CODE=43
 INVALID_CLUSTER_EXIT_CODE=44
 INVALID_NODE_EXIT_CODE=45
 NO_GPU_EXIT_CODE=46
+NOT_ON_HEAD_EXIT_CODE=47
 
 check_root_privileges()
 {
@@ -41,6 +42,16 @@ check_running_on_cobalt2()
     fi
 }
 
+check_running_on_cobalt2_head()
+{
+    # early exit if not on cobalt2 cluster
+    check_running_on_cobalt2
+
+    if [[ `hostname -f` != "cbm299."* ]]; then
+        exit $NOT_ON_HEAD_EXIT_CODE
+    fi
+}
+
 check_has_nvidia_gpu()
 {
     lspci | grep -i nvidia
-- 
GitLab