Function Test_viewRun

pkg/cmd/agent-task/view/view_test.go:160–1324

(t *testing.T)

}

func Test_viewRun(t *testing.T) {
	sampleDate := time.Now().Add(-6 * time.Hour) // 6h ago
	sampleCompletedAt := sampleDate.Add(5 * time.Minute)

	tests := []struct {
		name             string
		tty              bool
		opts             ViewOptions
		promptStubs      func(*testing.T, *prompter.MockPrompter)
		capiStubs        func(*testing.T, *capi.CapiClientMock)
		logRendererStubs func(*testing.T, *shared.LogRendererMock)
		jsonFields       []string
		wantOut          string
		wantErr          error
		wantStderr       string
		wantBrowserURL   string
	}{
		{
			name: "with session id, not found (tty)",
			tty:  true,
			opts: ViewOptions{
				SelectorArg: "some-session-id",
				SessionID:   "some-session-id",
			},
			capiStubs: func(t *testing.T, m *capi.CapiClientMock) {
				m.GetSessionFunc = func(_ context.Context, _ string) (*capi.Session, error) {
					return nil, capi.ErrSessionNotFound
				}
			},
			wantStderr: "session not found\n",
			wantErr:    cmdutil.SilentError,
		},
		{
			name: "with session id, api error (tty)",
			tty:  true,
			opts: ViewOptions{
				SelectorArg: "some-session-id",
				SessionID:   "some-session-id",
			},
			capiStubs: func(t *testing.T, m *capi.CapiClientMock) {
				m.GetSessionFunc = func(_ context.Context, _ string) (*capi.Session, error) {
					return nil, errors.New("some error")
				}
			},
			wantErr: errors.New("some error"),
		},
		{
			name: "with session id, success, with pr and user data (tty)",
			tty:  true,
			opts: ViewOptions{
				SelectorArg: "some-session-id",
				SessionID:   "some-session-id",
			},
			capiStubs: func(t *testing.T, m *capi.CapiClientMock) {
				m.GetSessionFunc = func(_ context.Context, id string) (*capi.Session, error) {
					assert.Equal(t, "some-session-id", id)
					return &capi.Session{
						ID:              "some-session-id",