add unit tests

AbhineetJain · AbhineetJain · commit ccddf238b60a · 2022-06-17T19:08:35.000Z
diff --git a/provisioner/terraform/serve_internal_test.go b/provisioner/terraform/serve_internal_test.go
@@ -0,0 +1,95 @@
+package terraform
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func Test_getAbsoluteBinaryPath(t *testing.T) {
+	t.Parallel()
+	type args struct {
+		ctx context.Context
+	}
+	tests := []struct {
+		name                   string
+		args                   args
+		terraformVersion       string
+		expectedAbsoluteBinary string
+		expectedOk             bool
+	}{
+		{
+			name:                   "TestCorrectVersion",
+			args:                   args{ctx: context.Background()},
+			terraformVersion:       "1.1.9",
+			expectedAbsoluteBinary: "",
+			expectedOk:             true,
+		},
+		{
+			name:                   "TestOldVersion",
+			args:                   args{ctx: context.Background()},
+			terraformVersion:       "1.0.9",
+			expectedAbsoluteBinary: "",
+			expectedOk:             false,
+		},
+		{
+			name:                   "TestNewVersion",
+			args:                   args{ctx: context.Background()},
+			terraformVersion:       "1.2.9",
+			expectedAbsoluteBinary: "",
+			expectedOk:             false,
+		},
+		{
+			name:                   "TestMalformedVersion",
+			args:                   args{ctx: context.Background()},
+			terraformVersion:       "version",
+			expectedAbsoluteBinary: "",
+			expectedOk:             false,
+		},
+	}
+	// nolint:paralleltest
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// Create a temp dir with the binary
+			tempDir := t.TempDir()
+			terraformBinaryOutput := fmt.Sprintf(`#!/bin/sh
+			cat <<-EOF
+			{
+				"terraform_version": "%s",
+				"platform": "linux_amd64",
+				"provider_selections": {},
+				"terraform_outdated": false
+			}
+			EOF`, tt.terraformVersion)
+
+			// #nosec
+			err := os.WriteFile(
+				filepath.Join(tempDir, "terraform"),
+				[]byte(terraformBinaryOutput),
+				0770,
+			)
+			require.NoError(t, err)
+
+			// Add the binary to PATH
+			pathVariable := os.Getenv("PATH")
+			t.Setenv("PATH", strings.Join([]string{tempDir, pathVariable}, ":"))
+
+			if tt.expectedOk {
+				tt.expectedAbsoluteBinary = filepath.Join(tempDir, "terraform")
+			}
+
+			actualAbsoluteBinary, actualOk := getAbsoluteBinaryPath(tt.args.ctx)
+			if actualAbsoluteBinary != tt.expectedAbsoluteBinary {
+				t.Errorf("getAbsoluteBinaryPath() absoluteBinaryPath, actual = %v, expected %v", actualAbsoluteBinary, tt.expectedAbsoluteBinary)
+			}
+			if actualOk != tt.expectedOk {
+				t.Errorf("getAbsoluteBinaryPath() ok, actual = %v, expected %v", actualOk, tt.expectedOk)
+			}
+		})
+	}
+}
