status command retrieves release status

state/state.go

@@ -298,6 +298,49 @@ func (state *HelmState) DiffReleases(helm helmexec.Interface, additionalValues [
 	return nil
 }
 
+func (state *HelmState) ReleaseStatuses(helm helmexec.Interface, workerLimit int) []error {
+	var errs []error
+	jobQueue := make(chan ReleaseSpec)
+	doneQueue := make(chan bool)
+	errQueue := make(chan error)
+
+	if workerLimit < 1 {
+		workerLimit = len(state.Releases)
+	}
+	for w := 1; w <= workerLimit; w++ {
+		go func() {
+			for release := range jobQueue {
+				if err := helm.ReleaseStatus(release.Name); err != nil {
+					errQueue <- err
+				}
+				doneQueue <- true
+			}
+		}()
+	}
+
+	go func() {
+		for _, release := range state.Releases {
+			jobQueue <- release
+		}
+		close(jobQueue)
+	}()
+
+	for i := 0; i < len(state.Releases); {
+		select {
+		case err := <-errQueue:
+			errs = append(errs, err)
+		case <-doneQueue:
+			i++
+		}
+	}
+
+	if len(errs) != 0 {
+		return errs
+	}
+
+	return nil
+}
+
 // DeleteReleases wrapper for executing helm delete on the releases
 func (state *HelmState) DeleteReleases(helm helmexec.Interface) []error {
 	var wg sync.WaitGroup

state/state_test.go

@@ -468,8 +468,9 @@ func Test_normalizeChart(t *testing.T) {
 // mocking helmexec.Interface
 
 type mockHelmExec struct {
-	charts []string
-	repo   []string
+	charts   []string
+	repo     []string
+	releases []string
 }
 
 func (helm *mockHelmExec) UpdateDeps(chart string) error {
@@ -496,6 +497,13 @@ func (helm *mockHelmExec) SyncRelease(name, chart string, flags ...string) error
 func (helm *mockHelmExec) DiffRelease(name, chart string, flags ...string) error {
 	return nil
 }
+func (helm *mockHelmExec) ReleaseStatus(release string) error {
+	if strings.Contains(release, "error") {
+		return errors.New("error")
+	}
+	helm.releases = append(helm.releases, release)
+	return nil
+}
 func (helm *mockHelmExec) DeleteRelease(name string) error {
 	return nil
 }
@@ -591,3 +599,50 @@ func TestHelmState_UpdateDeps(t *testing.T) {
 		t.Errorf("HelmState.UpdateDeps() - no errors, but got: %v", len(errs))
 	}
 }
+
+func TestHelmState_ReleaseStatuses(t *testing.T) {
+	tests := []struct {
+		name     string
+		releases []ReleaseSpec
+		helm     *mockHelmExec
+		want     []string
+		wantErr  bool
+	}{
+		{
+			name: "happy path",
+			releases: []ReleaseSpec{
+				{
+					Name: "releaseA",
+				},
+			},
+			helm: &mockHelmExec{},
+			want: []string{"releaseA"},
+		},
+		{
+			name: "happy path",
+			releases: []ReleaseSpec{
+				{
+					Name: "error",
+				},
+			},
+			helm:    &mockHelmExec{},
+			wantErr: true,
+		},
+	}
+	for _, tt := range tests {
+		i := func(t *testing.T) {
+			state := &HelmState{
+				Releases: tt.releases,
+			}
+			errs := state.ReleaseStatuses(tt.helm, 1)
+			if (errs != nil) != tt.wantErr {
+				t.Errorf("ReleaseStatuses() for %s error = %v, wantErr %v", tt.name, errs, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(tt.helm.releases, tt.want) {
+				t.Errorf("HelmState.ReleaseStatuses() for [%s] = %v, want %v", tt.name, tt.helm.releases, tt.want)
+			}
+		}
+		t.Run(tt.name, i)
+	}
+}