add test

Author: samtholiya

internal/exec/version_test.go

@@ -8,6 +8,7 @@ import (
 	"github.com/cloudposse/atmos/pkg/version"
 	"github.com/golang/mock/gomock"
 	"github.com/pkg/errors"
+	"github.com/stretchr/testify/assert"
 )
 
 // Mock interfaces generated by mockgen
@@ -194,3 +195,61 @@ func TestIsCheckVersionEnabled(t *testing.T) {
 		})
 	}
 }
+
+func TestGetLatestVersion(t *testing.T) {
+	tests := []struct {
+		name                string
+		version             string
+		latestRelease       string
+		getLatestReleaseErr error
+		expectedVersion     string
+		expectedOk          bool
+	}{
+		{
+			name:            "Same version",
+			version:         "v1.0.0",
+			latestRelease:   "v1.0.0",
+			expectedVersion: "",
+			expectedOk:      false,
+		},
+		{
+			name:            "Newer version available",
+			version:         "v1.0.0",
+			latestRelease:   "v1.1.0",
+			expectedVersion: "1.1.0",
+			expectedOk:      true,
+		},
+		{
+			name:                "Error fetching latest release",
+			version:             "v1.0.0",
+			getLatestReleaseErr: errors.New("error"),
+			expectedVersion:     "",
+			expectedOk:          false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ctrl := gomock.NewController(t)
+			oldVersion := version.Version
+			version.Version = tt.version
+			t.Cleanup(func() {
+				version.Version = oldVersion // Reset to a known state
+			})
+			defer ctrl.Finish()
+
+			mockExec := NewMockVersionExecutor(ctrl)
+			mockExec.EXPECT().GetLatestGitHubRepoRelease().
+				Return(tt.latestRelease, tt.getLatestReleaseErr).AnyTimes()
+
+			v := versionExec{
+				atmosConfig:                &schema.AtmosConfiguration{},
+				getLatestGitHubRepoRelease: mockExec.GetLatestGitHubRepoRelease,
+			}
+
+			version, ok := v.GetLatestVersion(true)
+			assert.Equal(t, tt.expectedVersion, version, "Expected version mismatch")
+			assert.Equal(t, tt.expectedOk, ok, "Expected ok mismatch")
+		})
+	}
+}