@@ -58,7 +58,9 @@ def test_cw_dashboard_builder(mocker, test_datadir, set_env, config_file_name, r
5858 if cluster_config .is_cw_dashboard_enabled :
5959 assert_that (output_yaml ).contains ("CloudwatchDashboard" )
6060 assert_that (output_yaml ).contains ("Head Node EC2 Metrics" )
61- _verify_head_node_instance_metrics_graphs (output_yaml , cluster_config .scheduling .scheduler )
61+ _verify_head_node_instance_metrics_graphs (
62+ output_yaml , cluster_config .scheduling .scheduler , cluster_config .is_cw_logging_enabled
63+ )
6264
6365 if cluster_config .are_alarms_enabled :
6466 assert_that (output_yaml ).contains ("Cluster Alarms" )
@@ -136,7 +138,7 @@ def _verify_metric_filter_dimensions(metric_filters):
136138 )
137139
138140
139- def _verify_head_node_instance_metrics_graphs (output_yaml , scheduler ):
141+ def _verify_head_node_instance_metrics_graphs (output_yaml , scheduler , is_cw_logging_enabled ):
140142 """Verify CloudWatch graphs within the Head Node Instance Metrics section."""
141143 assert_that (output_yaml ).contains ("Head Node Instance Metrics" )
142144 assert_that (output_yaml ).contains ("CPU Utilization" )
@@ -146,8 +148,8 @@ def _verify_head_node_instance_metrics_graphs(output_yaml, scheduler):
146148 assert_that (output_yaml ).contains ("Disk Read/Write Ops" )
147149 assert_that (output_yaml ).contains ("Disk Used Percent" )
148150 assert_that (output_yaml ).contains ("Memory Used Percent" )
149- # Daemons Heartbeats widget is only created for Slurm scheduler
150- if scheduler == "slurm" :
151+ # Daemons Heartbeats widget is only created for Slurm scheduler with logging enabled
152+ if scheduler == "slurm" and is_cw_logging_enabled :
151153 assert_that (output_yaml ).contains ("Daemons Heartbeats" )
152154 assert_that (output_yaml ).contains ("ClustermgtdHeartbeat" )
153155 else :
0 commit comments