Merge "Minimum diskGB varialbe should not be enforced, should assume …

…that the mimimum requirement for 256GB, if the requirement is not met, enforce setting the variable in testbed.py. Closes-Bug: 1460757"

fabfile/tasks/helpers.py

@@ -16,6 +16,7 @@
 import datetime
 from fabfile.tasks.esxi_defaults import apply_esxi_defaults
 from fabfile.utils.cluster import get_ntp_server
+from fabfile.utils.analytics import get_analytics_data_dir, get_minimum_diskGB
 
 @task
 @parallel
@@ -1143,15 +1144,31 @@ def delete_cassandra_db_files():
     sudo('rm -rf %s/saved_caches' %(db_path))
 
 
+@task
+@EXECUTE_TASK
+@roles('database')
+def check_disk_space():
+    data_dir = get_analytics_data_dir()
+    if not exists(data_dir, use_sudo=True):
+        return True
+    disk_cmd = "df -Pk " + data_dir + " | grep % | awk '{print $2}'"
+    total_disk = sudo(disk_cmd)
+    if (int(total_disk)/(1024*1024) < int(get_minimum_diskGB())):
+        return False
+    return True
+
 @task
 @roles('build')
 def pre_check():
-    db = getattr(testbed, 'minimum_diskGB', None)
-    if not db:
-        print "\nERROR: Minimum disk space for analytics db is not set in testbed.py"
+    result = execute('check_disk_space')
+    nodes_without_minimum_space = filter(lambda node: node != None,
+        map(lambda (node, met): node if not met else None, result.items()))
+    if nodes_without_minimum_space:
+        print "\nERROR: Minimum disk space(256GB) for analytics db is not met in nodes: %s"\
+            % nodes_without_minimum_space
         print "\tPlease set 'minimum_diskGB' in testbed.py and continue."
         print "\tSpecifiy the avalilable disk space of database node in GB"
-        print "\tRecommended to use a database node with 256GB disk."
+        print "\tHowever minimum disk space for database node is 256GB."
         exit(1)
     database_nodes = deepcopy(env.roledefs['database'])
     if (len(database_nodes) % 2) != 1:

fabfile/utils/analytics.py

@@ -24,7 +24,8 @@ def get_database_dir():
     return getattr(testbed, 'database_dir', None)
 
 def get_analytics_data_dir():
-    return getattr(testbed, 'analytics_data_dir', None)
+    return getattr(testbed, 'analytics_data_dir',
+                   '/var/lib/cassandra/data')
 
 def get_ssd_data_dir():
     return getattr(testbed, 'ssd_data_dir', None)
@@ -46,7 +47,7 @@ def get_analytics_flow_ttl():
 #end get_analytics_flow_ttl
 
 def get_minimum_diskGB():
-    return getattr(testbed, 'minimum_diskGB', None)
+    return getattr(testbed, 'minimum_diskGB', '256')
 #end get_analytics_flow_ttl
 
 def get_kafka_enabled():