Function TestSessionWaitStreamsUntilStopped

internal/cli/session_test.go:522–574

(t *testing.T)

}

func TestSessionWaitStreamsUntilStopped(t *testing.T) {
	t.Parallel()

	getCalls := 0
	deps := newTestDeps(t, &stubClient{
		getSessionFn: func(context.Context, string) (SessionRecord, error) {
			getCalls++
			state := session.StateActive
			if getCalls > 1 {
				state = session.StateStopped
			}
			return SessionRecord{
				ID:            "sess-1",
				AgentName:     "coder",
				WorkspaceID:   "ws-1",
				WorkspacePath: "/workspace/project",
				State:         state,
				CreatedAt:     fixedTestNow,
				UpdatedAt:     fixedTestNow,
			}, nil
		},
		streamSessionFn: func(_ context.Context, id string, _ SessionEventQuery, _ string, handler SSEHandler) error {
			return handler(SSEEvent{
				ID:    "2",
				Event: session.EventTypeSessionStopped,
				Data: mustJSON(t, SessionEventRecord{
					ID:        "evt-2",
					SessionID: id,
					Sequence:  2,
					Type:      session.EventTypeSessionStopped,
					AgentName: "coder",
					Timestamp: fixedTestNow,
				}),
			})
		},
	})

	stdout, _, err := executeRootCommand(t, deps, "session", "wait", "sess-1", "-o", "json")
	if err != nil {
		t.Fatalf("executeRootCommand() error = %v", err)
	}
	if getCalls != 2 {
		t.Fatalf("GetSession() calls = %d, want 2", getCalls)
	}

	var decoded SessionRecord
	if err := json.Unmarshal([]byte(stdout), &decoded); err != nil {
		t.Fatalf("json.Unmarshal() error = %v", err)
	}
	if decoded.State != session.StateStopped {
		t.Fatalf("decoded.State = %q, want %q", decoded.State, session.StateStopped)
	}
}

func TestSessionStopFetchesUpdatedSession(t *testing.T) {
	t.Parallel()