Add unit tests for new python hook

Author: yorinasub17

.circleci/config.yml

@@ -0,0 +1,31 @@
+version: 2.1
+
+jobs:
+  unit_test_py3:
+    docker:
+      - image: python:3
+    steps:
+      - checkout
+      - run:
+          name: run tests
+          command: |
+            python test/check-skip-env/check_skip_env_test.py
+  unit_test_py2:
+    docker:
+      - image: python:2
+    steps:
+      - checkout
+      - run:
+          name: run tests
+          command: |
+            python test/check-skip-env/check_skip_env_test.py
+
+
+workflows:
+  version: 2
+  test:
+    jobs:
+      - unit_test:
+          filters:
+            tags:
+              only: /^v.*/

test/check-skip-env/check_skip_env_test.py

@@ -0,0 +1,60 @@
+import os
+import glob
+import unittest
+import subprocess
+
+
+class TestCheckSkipEnv(unittest.TestCase):
+    def setUp(self):
+        self.root = get_git_root()
+        self.hook_script = os.path.join(self.root, 'hooks', 'check_skip_env.py')
+        self.fixture_dir = os.path.join(self.root, 'test', 'check-skip-env', 'fixtures')
+
+    def _check_success(self, files):
+        subprocess.run([self.hook_script] + files, check=True)
+
+    def _check_failure(self, files, failed_files):
+        result = subprocess.run([self.hook_script] + files, stderr=subprocess.PIPE)
+        self.assertEqual(result.returncode, 1)
+        for f in failed_files:
+            self.assertIn(f, result.stderr.decode('utf-8'))
+
+    def test_everything_commented(self):
+        self._check_success([os.path.join(self.fixture_dir, 'everything_commented_test.go')])
+
+    def test_non_skip_uncommented(self):
+        self._check_success([os.path.join(self.fixture_dir, 'non_skip_uncommented_test.go')])
+
+    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_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])
+
+    def test_nested_uncommented(self):
+        test_file_path = os.path.join(self.fixture_dir, 'nested_uncommented_test.go')
+        self._check_failure([test_file_path], [test_file_path])
+
+    def test_everything(self):
+        all_test_files = glob.glob(os.path.join(self.fixture_dir, '*.go'))
+        failed_files = [
+            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'),
+        ]
+        self._check_failure(all_test_files, failed_files)
+
+
+def get_git_root():
+    """ Returns the root directory of the git repository, assuming this script is run from within the repository. """
+    result = subprocess.run(['git', 'rev-parse', '--show-toplevel'], stdout=subprocess.PIPE, check=True)
+    if result.stdout is None:
+        # TODO: concrete exception
+        raise Exception('Did not get any output from git: stderr is "{}"'.format(result.stderr))
+    return result.stdout.decode('utf-8').rstrip('\n')
+
+
+if __name__ == '__main__':
+    unittest.main()

test/check-skip-env/fixtures/everything_commented_test.go

@@ -0,0 +1,35 @@
+package basecase
+
+import (
+	"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")
+	})
+}

test/check-skip-env/fixtures/multiple_skip_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")
+	})
+}

test/check-skip-env/fixtures/nested_uncommented_test.go

@@ -0,0 +1,40 @@
+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_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")
+	})
+
+	for _, data := range testCases {
+		t.Run(data.name, func(t *testing.T) {
+			os.Setenv("SKIP_validate")
+			test_structure.RunTestStage(t, "validate", func() {
+				logger.Logf(t, "validate")
+			})
+		})
+	}
+}

test/check-skip-env/fixtures/non_skip_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")
+	})
+}

test/check-skip-env/fixtures/skip_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")
+	})
+}