def test_cloudwatch_alarm_creation(cloudwatch_client_stub):
# create test cluster node IDs and an associated cloudwatch helper
node_ids = ["i-abc", "i-def"]
cloudwatch_helper = helpers.get_cloudwatch_helper(node_ids)
# given a directive to create cluster CloudWatch alarms...
# expect to make a call to create alarms for each node in the cluster
stubs.put_cluster_alarms_success(cloudwatch_client_stub, cloudwatch_helper)
# given our mocks and the example cloudwatch alarm config as input...
# expect cluster alarms to be created successfully
cloudwatch_helper.put_cloudwatch_alarm()
# expect no pending responses left in the CloudWatch client stub queue
cloudwatch_client_stub.assert_no_pending_responses()
def test_cloudwatch_alarm_update_worker_node(ec2_client_stub, ssm_client_stub,
cloudwatch_client_stub):
# create test cluster worker node ID and an associated cloudwatch helper
node_id = "i-abc"
is_head_node = False
cloudwatch_helper = helpers.get_cloudwatch_helper(node_id)
# given a directive to check for the Unified CloudWatch Agent status...
# expect CloudWatch Agent is installed
stubs.get_ec2_cwa_installed_tag_true(ec2_client_stub, node_id)
# given a directive to update a cluster CloudWatch Alarm Config with new
# changes...
# expect the stored the CloudWatch Alarm Config is different from local
# config
cw_ssm_param_name = helpers.get_ssm_param_name(
cloudwatch_helper.cluster_name, "alarm")
# given a directive to update a cluster CloudWatch Alarm Config with new
# changes...
# expect the stored the CloudWatch Alarm Config is different from local
# config
stubs.get_head_node_config_hash_different(ec2_client_stub, "alarm",
cloudwatch_helper, node_id)
stubs.get_cur_node_config_hash_different(ec2_client_stub, "alarm", node_id)
# given an updated CloudWatch Alarm Config file...
# expect to update the node tag equal to updated config file sha1 hash
# to reflect the changes in config file
stubs.update_hash_tag_success(ec2_client_stub, node_id, "alarm",
cloudwatch_helper)
stubs.get_param_ssm_same(ssm_client_stub, cw_ssm_param_name,
cloudwatch_helper, "alarm")
# given a directive to create cluster CloudWatch Alarms...
# expect to make a call to create alarms for each node in the cluster
stubs.put_cluster_alarms_success(cloudwatch_client_stub, cloudwatch_helper)
# given our mocks and the example CloudWatch Alarm config as input...
# expect CloudWatch Alarm configured to use updated file on each cluster
# node successfully
cloudwatch_helper._update_cloudwatch_config("alarm", is_head_node)
# expect no pending responses left in client stub queues
ec2_client_stub.assert_no_pending_responses()
ssm_client_stub.assert_no_pending_responses()
def test_cloudwatch_alarm_creation(cloudwatch_client_stub, ssm_client_stub):
# create test cluster node IDs and an associated cloudwatch helper
node_id = "i-abc"
cloudwatch_helper = helpers.get_cloudwatch_helper(node_id)
# given a directive to update a cluster CloudWatch Alarm Config without any
# change...
# expect the stored the CloudWatch Alarm Config is same as local config
cw_ssm_param_name = helpers.get_ssm_param_name(
cloudwatch_helper.cluster_name, "alarm")
stubs.get_param_ssm_same(ssm_client_stub, cw_ssm_param_name,
cloudwatch_helper, "alarm")
# given a directive to create cluster CloudWatch alarms...
# expect to make a call to create alarms for each node in the cluster
stubs.put_cluster_alarms_success(cloudwatch_client_stub, cloudwatch_helper)
# given our mocks and the example CloudWatch Alarm config as input...
# expect cluster alarms to be created successfully
cloudwatch_helper._put_cloudwatch_alarm()
# expect no pending responses left in the CloudWatch client stub queue
cloudwatch_client_stub.assert_no_pending_responses()