Fix up tests

airbnb · Aug 22, 2019 · 8ab098e · 8ab098e
1 parent 4f32211
commit 8ab098e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/stream_alert_cli/terraform/generate.py b/stream_alert_cli/terraform/generate.py
@@ -23,7 +23,6 @@
     DEFAULT_SNS_MONITORING_TOPIC,
     InvalidClusterName,
     infinitedict,
-    MisconfigurationError,
 )
 from stream_alert_cli.terraform.alert_merger import generate_alert_merger
 from stream_alert_cli.terraform.alert_processor import generate_alert_processor

diff --git a/tests/unit/stream_alert_cli/terraform/test_generate.py b/tests/unit/stream_alert_cli/terraform/test_generate.py
@@ -14,7 +14,7 @@
 limitations under the License.
 """
 from mock import ANY, patch
-from nose.tools import assert_equal, assert_false, assert_raises, assert_true
+from nose.tools import assert_equal, assert_false, assert_true
 
 from stream_alert_cli.config import CLIConfig
 from stream_alert_cli.terraform import (
@@ -24,7 +24,6 @@
     flow_logs,
     generate
 )
-from stream_alert_cli.terraform.common import MisconfigurationError
 
 
 class TestTerraformGenerate(object):
@@ -495,12 +494,10 @@ def test_generate_main_with_sqs_url_unspecified(self):
         """CLI - Terraform Generate Main with unspecified classifier_sqs.use_prefix"""
         del self.config['global']['infrastructure']['classifier_sqs']['use_prefix']
 
-        assert_raises(
-            MisconfigurationError,
-            generate.generate_main,
-            config=self.config,
-            init=False
-        )
+        result = generate.generate_main(config=self.config, init=False)
+
+        assert_equal(result['module']['globals']['source'], 'modules/tf_stream_alert_globals')
+        assert_true(result['module']['globals']['sqs_use_prefix'])
 
     def test_generate_main_with_sqs_url_true(self):
         """CLI - Terraform Generate Main with classifier_sqs.use_prefix = True"""