fix: add unit test

Author: hakhandelwal11

pkg/cmd/hello/hello_test.go

@@ -1 +1,39 @@
 package hello
+
+import "testing"
+
+type shellOnboardingPollDoneCase struct {
+	name string
+	res  *OnboardingObject
+	want bool
+}
+
+var shellOnboardingPollDoneCases = []shellOnboardingPollDoneCase{
+	{
+		name: "hasRunBrevShell",
+		res:  &OnboardingObject{HasRunBrevShell: true},
+		want: true,
+	},
+	{
+		name: "brevOpenOnly",
+		res:  &OnboardingObject{HasRunBrevOpen: true, HasRunBrevShell: false},
+		want: false,
+	},
+	{
+		name: "nil",
+		res:  nil,
+		want: false,
+	},
+}
+
+func TestShellOnboardingPollDone(t *testing.T) {
+	t.Parallel()
+	for _, c := range shellOnboardingPollDoneCases {
+		t.Run(c.name, func(t *testing.T) {
+			t.Parallel()
+			if got := shellOnboardingPollDone(c.res); got != c.want {
+				t.Fatalf("got %v, want %v", got, c.want)
+			}
+		})
+	}
+}

pkg/cmd/hello/onboarding_utils.go

@@ -123,6 +123,10 @@ type OnboardingObject struct {
 	HasRunBrevOpen  bool `json:"hasRunBrevOpen"`
 }
 
+func shellOnboardingPollDone(res *OnboardingObject) bool {
+	return res != nil && res.HasRunBrevShell
+}
+
 func SetupDefaultOnboardingFile() error {
 	// get path
 	path, err := GetOnboardingFilePath()

pkg/cmd/hello/steps.go

@@ -298,7 +298,7 @@ func doVsCodeOnboarding(
 		if err1 != nil {
 			return breverrors.WrapAndTrace(err1)
 		}
-		if res.HasRunBrevOpen {
+		if shellOnboardingPollDone(res) {
 			spinner.Suffix = spinnerSuffix
 			time.Sleep(250 * time.Millisecond)
 			spinner.Stop()