From e8b17a79132582a7af277b2f6ae4a5f45a449abc Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Wed, 30 Oct 2024 11:57:55 -0700
Subject: [PATCH] print df

---
 e2e_tests/common_scripts.sh | 8 ++++++--
 e2e_tests/run_e2e_tests.sh  | 2 ++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/e2e_tests/common_scripts.sh b/e2e_tests/common_scripts.sh
index 154d6cc1..7af88e89 100644
--- a/e2e_tests/common_scripts.sh
+++ b/e2e_tests/common_scripts.sh
@@ -6,6 +6,11 @@ log() {
   echo "$timestamp: $message"
 }
 
+print_disk_usage() {
+    echo free disk space
+    df -h | grep "/$"
+}
+
 wait_for_healthz() {
   local healthz_url="$1"
   local timeout_seconds="${2:-30}"  # Default timeout of 30 seconds
@@ -28,8 +33,7 @@ wait_for_healthz() {
       return 1
     fi
 
-    echo print free disk space
-    df -h
+    print_disk_usage
 
     sleep $sleep_between
   done
diff --git a/e2e_tests/run_e2e_tests.sh b/e2e_tests/run_e2e_tests.sh
index a4a652a7..ff8eeaa1 100644
--- a/e2e_tests/run_e2e_tests.sh
+++ b/e2e_tests/run_e2e_tests.sh
@@ -4,6 +4,8 @@ set -e
 
 . ./common_scripts.sh
 
+print_disk_usage
+
 print_debug() {
   log "Received signal to stop"
   log "Printing debug logs for model_server"