work around for capacity metrics

Signed-off-by: hwassman <[email protected]>

Author: Helene

source/queryHandler/Query.py

@@ -49,6 +49,10 @@ class Query(object):
                   "operation", "protocol", "waiters_time_threshold", "export",
                   "nodegroup", "account", "filesystem", "tct_csap", "tct_operation", "cloud_nodeclass"])
 
+    DISK_CAP_METRICS = set(["gpfs_disk_disksize", "gpfs_disk_free_fullkb", "gpfs_disk_free_fragkb",
+                            "gpfs_pool_disksize", "gpfs_pool_free_fragkb", "gpfs_pool_free_fullkb", 
+                            "gpfs_fs_inode_used", "gpfs_fs_inode_free", "gpfs_fs_inode_alloc", "gpfs_fs_inode_max"])
+
     def __init__(self, metrics=None, bucketsize=1, filters=None, groupby=None, includeDiskData=False):
         '''
         Constructor, filters and groupby must be preformmated
@@ -184,7 +188,11 @@ def addMeasurement(self, meassure):
     def __str__(self):
         # dd = '-a' if self.includeDiskData else ''
         # Workaround for RTC Defect 280368: Zimon capacity query does not return all results (seen on CNSA)
-        if (self.metrics and any('gpfs_disk_' in metric for metric in self.metrics)) or (self.sensor and self.sensor == "GPFSDiskCap"):
+        if (self.metrics and 
+            any(str(metric) in self.DISK_CAP_METRICS for metric in self.metrics)
+            ) or (self.sensor and 
+                  self.sensor in ("GPFSDiskCap","GPFSPoolCap", "GPFSInodeCap")
+                  ):
             dd = '-ar'
         elif self.includeDiskData:
             dd = '-a'

tests/test_query.py

@@ -0,0 +1,21 @@
+from source.queryHandler.Query import Query
+from source.__version__ import __version__ as version
+from nose2.tools.decorators import with_setup
+
+
+def my_setup():
+    global metrics, metrics1
+    metrics = ['cpu_user']
+    metrics1 = ['gpfs_fs_inode_used']
+
+@with_setup(my_setup)
+def test_case01():
+    query = Query(metrics)
+    assert len(str(query)) > 0
+    assert "cpu_user" in str(query)
+    assert "-ar" not in str(query)
+
+@with_setup(my_setup)
+def test_case02():
+    query = Query(metrics1)
+    assert "-ar" in str(query)