Skip to content

Commit 0c2be8d

Browse files
hanwen-clusterhanwen-pcluste
authored andcommitted
[Integ-tests] Fix DCV test failed because of change for external Slurm dbd
Signed-off-by: Hanwen <hanwenli@amazon.com>
1 parent c2275c9 commit 0c2be8d

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/integration-tests/tests/cloudwatch_logging/test_cloudwatch_logging.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@
3333
HEAD_NODE_ROLE_NAME = "HeadNode"
3434
COMPUTE_NODE_ROLE_NAME = "ComputeFleet"
3535
LOGIN_NODE_ROLE_NAME = "LoginNode"
36-
NODE_ROLE_NAMES = {HEAD_NODE_ROLE_NAME, COMPUTE_NODE_ROLE_NAME, LOGIN_NODE_ROLE_NAME}
36+
EXTERNAL_SLURM_DBD_ROLE_NAME = "ExternalSlurmDbd"
37+
NODE_ROLE_NAMES = {HEAD_NODE_ROLE_NAME, COMPUTE_NODE_ROLE_NAME, LOGIN_NODE_ROLE_NAME, EXTERNAL_SLURM_DBD_ROLE_NAME}
3738

3839

3940
def _get_log_group_name_for_cluster(cluster_name):
@@ -227,12 +228,19 @@ def _read_login_node_config(self):
227228
# TODO add logic to execute remote commands on LoginNodes
228229
return login_node_config
229230

231+
def _read_external_dbd_node_config(self):
232+
"""Read the node configuration JSON file at NODE_CONFIG_PATH on a external dbd node."""
233+
external_dbd_node_config = {}
234+
# TODO add logic to execute remote commands on ExternalDbd node
235+
return external_dbd_node_config
236+
230237
def _read_node_configs(self):
231238
"""Return a dict mapping node role names to the config at NODE_CONFIG_PATH."""
232239
return {
233240
HEAD_NODE_ROLE_NAME: self._read_head_node_config(),
234241
COMPUTE_NODE_ROLE_NAME: self._read_compute_node_config(),
235242
LOGIN_NODE_ROLE_NAME: self._read_login_node_config(),
243+
EXTERNAL_SLURM_DBD_ROLE_NAME: self._read_external_dbd_node_config(),
236244
}
237245

238246
@staticmethod

0 commit comments

Comments
 (0)