def test_generate_cluster_test(self):
"""CLI - Terraform Generate 'Test' Cluster"""
tf_cluster = terraform_generate.generate_cluster(config=self.config,
cluster_name='test')
cluster_keys = {'module', 'output'}
test_modules = {
'stream_alert_test', 'cloudwatch_monitoring_test', 'kinesis_test',
'kinesis_events_test'
}
assert_equal(set(tf_cluster['module'].keys()), test_modules)
assert_equal(set(tf_cluster.keys()), cluster_keys)
def test_generate_cluster_advanced(self):
"""CLI - Terraform Generate 'Advanced' Cluster"""
tf_cluster = terraform_generate.generate_cluster(
config=self.config, cluster_name='advanced')
cluster_keys = {'module', 'output'}
advanced_modules = {
'stream_alert_advanced', 'cloudwatch_monitoring_advanced',
'kinesis_advanced', 'kinesis_events_advanced',
'flow_logs_advanced', 'cloudtrail_advanced'
}
assert_equal(set(tf_cluster['module'].keys()), advanced_modules)
assert_equal(set(tf_cluster.keys()), cluster_keys)