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)