From 15e1fa510dc1acd468af91c38750458d5e9ea136 Mon Sep 17 00:00:00 2001
From: David Gardiner <dmgardiner25@github.com>
Date: Tue, 30 May 2023 14:19:15 -0700
Subject: [PATCH] Add test

---
 pkg/cmd/codespace/view_test.go | 128 +++++++++++++++++++++++++++++++++
 1 file changed, 128 insertions(+)
 create mode 100644 pkg/cmd/codespace/view_test.go

diff --git a/pkg/cmd/codespace/view_test.go b/pkg/cmd/codespace/view_test.go
new file mode 100644
index 000000000..549a53fc6
--- /dev/null
+++ b/pkg/cmd/codespace/view_test.go
@@ -0,0 +1,128 @@
+package codespace
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/cli/cli/v2/internal/codespaces/api"
+	"github.com/cli/cli/v2/pkg/iostreams"
+)
+
+func Test_NewCmdView(t *testing.T) {
+	type fields struct {
+		apiClient apiClient
+	}
+	tests := []struct {
+		tName         string
+		codespaceName string
+		opts          *viewOptions
+		cliArgs       []string
+		wantErr       bool
+		wantStdout    string
+		errMsg        string
+	}{
+		{
+			tName:   "selector throws because no terminal found",
+			opts:    &viewOptions{},
+			wantErr: true,
+			errMsg:  "choosing codespace: error getting answers: no terminal",
+		},
+		{
+			tName:         "command fails because provided codespace doesn't exist",
+			codespaceName: "i-dont-exist",
+			opts:          &viewOptions{},
+			wantErr:       true,
+			errMsg:        "getting full codespace details: codespace not found",
+		},
+		{
+			tName:         "command succeeds because codespace exists (no details)",
+			codespaceName: "monalisa-cli-cli-abcdef",
+			opts:          &viewOptions{},
+			wantErr:       false,
+			wantStdout:    "Name\tmonalisa-cli-cli-abcdef\nGit Status\t - 0↓ 0↑\nIdle Timeout\t0 minutes\n",
+		},
+		{
+			tName:         "command succeeds because codespace exists (with details)",
+			codespaceName: "monalisa-cli-cli-hijklm",
+			opts:          &viewOptions{},
+			wantErr:       false,
+			wantStdout:    "Name\tmonalisa-cli-cli-hijklm\nGit Status\tmain* - 2↓ 1↑\nIdle Timeout\t30 minutes\nRetention Period\t1 day\n",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.tName, func(t *testing.T) {
+			ios, _, stdout, _ := iostreams.Test()
+			a := &App{
+				apiClient: testViewApiMock(),
+				io:        ios,
+			}
+			selector := &CodespaceSelector{api: a.apiClient, codespaceName: tt.codespaceName}
+			tt.opts.selector = selector
+
+			var err error
+			if tt.cliArgs == nil {
+				if tt.opts.selector == nil {
+					t.Fatalf("selector must be set in opts if cliArgs are not provided")
+				}
+
+				err = a.ViewCodespace(context.Background(), tt.opts)
+			} else {
+				cmd := newViewCmd(a)
+				cmd.SilenceUsage = true
+				cmd.SilenceErrors = true
+				cmd.SetOut(ios.ErrOut)
+				cmd.SetErr(ios.ErrOut)
+				cmd.SetArgs(tt.cliArgs)
+				_, err = cmd.ExecuteC()
+			}
+
+			if tt.wantErr {
+				if err == nil {
+					t.Error("Edit() expected error, got nil")
+				} else if err.Error() != tt.errMsg {
+					t.Errorf("Edit() error = %q, want %q", err, tt.errMsg)
+				}
+			} else if err != nil {
+				t.Errorf("Edit() expected no error, got %v", err)
+			}
+
+			if out := stdout.String(); out != tt.wantStdout {
+				t.Errorf("stdout = %q, want %q", out, tt.wantStdout)
+			}
+		})
+	}
+}
+
+func testViewApiMock() *apiClientMock {
+	codespaceWithNoDetails := &api.Codespace{
+		Name: "monalisa-cli-cli-abcdef",
+	}
+	codespaceWithDetails := &api.Codespace{
+		Name: "monalisa-cli-cli-hijklm",
+		GitStatus: api.CodespaceGitStatus{
+			Ahead:                 1,
+			Behind:                2,
+			Ref:                   "main",
+			HasUnpushedChanges:    true,
+			HasUncommittedChanges: true,
+		},
+		IdleTimeoutMinutes:     30,
+		RetentionPeriodMinutes: 1440,
+	}
+	return &apiClientMock{
+		GetCodespaceFunc: func(_ context.Context, name string, _ bool) (*api.Codespace, error) {
+			if name == codespaceWithDetails.Name {
+				return codespaceWithDetails, nil
+			} else if name == codespaceWithNoDetails.Name {
+				return codespaceWithNoDetails, nil
+			}
+
+			return nil, fmt.Errorf("codespace not found")
+		},
+		ListCodespacesFunc: func(ctx context.Context, opts api.ListCodespacesOptions) ([]*api.Codespace, error) {
+			return []*api.Codespace{codespaceWithNoDetails, codespaceWithDetails}, nil
+		},
+	}
+}