def test_generate_stream_alert_test(self):
"""CLI - Terraform Generate stream_alert Module (test cluster)"""
terraform_generate.generate_stream_alert('test', self.cluster_dict,
self.config)
expected_test_cluster = {
'module': {
'stream_alert_test': {
'source': 'modules/tf_stream_alert',
'account_id': '12345678910',
'region': 'us-west-1',
'prefix': 'unit-testing',
'cluster': 'test',
'kms_key_arn': '${aws_kms_key.stream_alert_secrets.arn}',
'rule_processor_memory': 128,
'rule_processor_timeout': 25,
'rule_processor_version': '$LATEST',
'rule_processor_config': '${var.rule_processor_config}',
'alert_processor_memory': 128,
'alert_processor_timeout': 25,
'alert_processor_version': '$LATEST',
'alert_processor_config': '${var.alert_processor_config}',
}
}
}
assert_equal(self.cluster_dict['module']['stream_alert_test'],
expected_test_cluster['module']['stream_alert_test'])
def test_generate_stream_alert_advanced(self):
"""CLI - Terraform Generate StreamAlert - Advanced Cluster)"""
terraform_generate.generate_stream_alert('advanced', self.cluster_dict,
self.config)
expected_advanced_cluster = {
'module': {
'stream_alert_advanced': {
'source': 'modules/tf_stream_alert',
'account_id': '12345678910',
'region': 'us-west-1',
'prefix': 'unit-testing',
'cluster': 'advanced',
'kms_key_arn': '${aws_kms_key.stream_alert_secrets.arn}',
'rule_processor_enable_metrics': False,
'rule_processor_log_level': 'info',
'rule_processor_memory': 128,
'rule_processor_timeout': 25,
'rule_processor_version': '$LATEST',
'rule_processor_config': '${var.rule_processor_config}',
'alert_processor_enable_metrics': False,
'alert_processor_log_level': 'info',
'alert_processor_memory': 128,
'alert_processor_timeout': 25,
'alert_processor_version': '$LATEST',
'alert_processor_config': '${var.alert_processor_config}',
'output_lambda_functions':
['my-lambda-function:production'],
'output_s3_buckets': ['my-s3-bucket.with.data'],
'input_sns_topics': ['my-sns-topic-name'],
'alert_processor_vpc_enabled': True,
'alert_processor_vpc_subnet_ids': ['subnet-id-1'],
'alert_processor_vpc_security_group_ids': ['sg-id-1']
}
}
}
assert_equal(
self.cluster_dict['module']['stream_alert_advanced'],
expected_advanced_cluster['module']['stream_alert_advanced'])