Include TERRATEST_REGION in check

Author: yorinasub17

hooks/check_skip_env.py

@@ -11,7 +11,7 @@
 def has_setenv_skip(fpath):
     with open(fpath) as f:
         for line in f:
-            if re.match(r'^\s+os.Setenv\(\"SKIP_', line):
+            if re.match(r'^\s+os.Setenv\(\"(SKIP_|TERRATEST_REGION)', line):
                 return True
     return False
 

test/check_skip_env_test.py

@@ -34,6 +34,10 @@ def test_skip_uncommented(self):
         test_file_path = os.path.join(self.fixture_dir, 'skip_uncommented_test.go')
         self._check_failure([test_file_path], [test_file_path])
 
+    def test_terratest_region_uncommented(self):
+        test_file_path = os.path.join(self.fixture_dir, 'terratest_region_uncommented_test.go')
+        self._check_failure([test_file_path], [test_file_path])
+
     def test_multiple_skip_uncommented(self):
         test_file_path = os.path.join(self.fixture_dir, 'multiple_skip_uncommented_test.go')
         self._check_failure([test_file_path], [test_file_path])
@@ -48,6 +52,7 @@ def test_everything(self):
             os.path.join(self.fixture_dir, 'skip_uncommented_test.go'),
             os.path.join(self.fixture_dir, 'multiple_skip_uncommented_test.go'),
             os.path.join(self.fixture_dir, 'nested_uncommented_test.go'),
+            os.path.join(self.fixture_dir, 'terratest_region_uncommented_test.go'),
         ]
         self._check_failure(all_test_files, failed_files)
 

test/fixtures/non_skip_uncommented_test.go

@@ -12,7 +12,7 @@ func TestWithStages(t *testing.T) {
 	t.Parallel()
 
 	// Uncomment the items below to skip certain parts of the test
-	os.Setenv("TERRATEST_REGION", "eu-west-1")
+	os.Setenv("THE_ANSWER", "42")
 	//os.Setenv("SKIP_setup", "true")
 	//os.Setenv("SKIP_deploy", "true")
 	//os.Setenv("SKIP_validate", "true")

test/fixtures/terratest_region_uncommented_test.go

@@ -0,0 +1,36 @@
+package basecase
+
+import (
+	"os"
+	"testing"
+
+	"github.com/gruntwork-io/terratest/modules/logger"
+	test_structure "github.com/gruntwork-io/terratest/modules/test-structure"
+)
+
+func TestWithStages(t *testing.T) {
+	t.Parallel()
+
+	// Uncomment the items below to skip certain parts of the test
+	os.Setenv("TERRATEST_REGION", "eu-west-1")
+	//os.Setenv("SKIP_setup", "true")
+	//os.Setenv("SKIP_deploy", "true")
+	//os.Setenv("SKIP_validate", "true")
+	//os.Setenv("SKIP_cleanup", "true")
+
+	test_structure.RunTestStage(t, "setup", func() {
+		logger.Logf(t, "setup")
+	})
+
+	defer test_structure.RunTestStage(t, "cleanup", func() {
+		logger.Logf(t, "cleanup")
+	})
+
+	test_structure.RunTestStage(t, "deploy", func() {
+		logger.Logf(t, "deploy")
+	})
+
+	test_structure.RunTestStage(t, "validate", func() {
+		logger.Logf(t, "validate")
+	})
+}