11import time
22
33from node import Node
4- from metrics .metrics import get_node_metrics , check_longhorn_metric
4+ from metrics .metrics import get_node_metrics , check_longhorn_metric , find_longhorn_metric_samples
55from metrics .metrics import get_longhorn_components_memory_cpu_usage
66from metrics .metrics import check_longhorn_components_memory_cpu_usage
77from utility .utility import get_retry_count_and_interval
@@ -12,7 +12,7 @@ class metrics_keywords:
1212
1313 def __init__ (self ):
1414 self .node = Node ()
15- retry_count , retry_interval = get_retry_count_and_interval ()
15+ self . retry_count , self . retry_interval = get_retry_count_and_interval ()
1616
1717 def get_node_total_memory_in_mi (self , node_name ):
1818 total_memory = self .node .get_node_total_memory (node_name )
@@ -52,6 +52,14 @@ def check_if_node_under_memory_pressure(self, node_name):
5252 def check_longhorn_metric (self , metric_name , node_name = None , metric_label = None , expected_value = None ):
5353 check_longhorn_metric (metric_name , node_name , metric_label , expected_value )
5454
55+ def wait_for_longhorn_metric_absent (self , metric_name , node_name = None ):
56+ for i in range (self .retry_count ):
57+ logging (f"Waiting for longhorn metric { metric_name } absent on node { node_name } ... ({ i } )" )
58+ if not find_longhorn_metric_samples (metric_name , node_name ):
59+ logging (f"Longhorn metric { metric_name } is absent on node { node_name } " )
60+ return
61+ time .sleep (self .retry_interval )
62+
5563 def get_longhorn_components_memory_cpu_usage (self ):
5664 get_longhorn_components_memory_cpu_usage ()
5765
0 commit comments