Function TestListProcesses

agent/agentproc/api_test.go:476–713

(t *testing.T)

}

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

	t.Run("NoProcesses", func(t *testing.T) {
		t.Parallel()

		handler := newTestAPI(t)
		w := getList(t, handler)
		require.Equal(t, http.StatusOK, w.Code)

		var resp workspacesdk.ListProcessesResponse
		err := json.NewDecoder(w.Body).Decode(&resp)
		require.NoError(t, err)
		require.NotNil(t, resp.Processes)
		require.Empty(t, resp.Processes)
	})

	t.Run("FilterByChatID", func(t *testing.T) {
		t.Parallel()

		handler := newTestAPI(t)

		chatA := uuid.New().String()
		chatB := uuid.New().String()
		headersA := http.Header{workspacesdk.CoderChatIDHeader: {chatA}}
		headersB := http.Header{workspacesdk.CoderChatIDHeader: {chatB}}

		// Start processes with different chat IDs.
		id1 := startAndGetID(t, handler, workspacesdk.StartProcessRequest{
			Command: "echo chat-a",
		}, headersA)
		waitForExit(t, handler, id1)

		id2 := startAndGetID(t, handler, workspacesdk.StartProcessRequest{
			Command: "echo chat-b",
		}, headersB)
		waitForExit(t, handler, id2)

		id3 := startAndGetID(t, handler, workspacesdk.StartProcessRequest{
			Command: "echo chat-a-2",
		}, headersA)
		waitForExit(t, handler, id3)

		// List with chat A header should return 2 processes.
		w := getListWithChatHeader(t, handler, chatA)
		require.Equal(t, http.StatusOK, w.Code)

		var resp workspacesdk.ListProcessesResponse
		err := json.NewDecoder(w.Body).Decode(&resp)
		require.NoError(t, err)
		require.Len(t, resp.Processes, 2)

		ids := make(map[string]bool)
		for _, p := range resp.Processes {
			ids[p.ID] = true
		}
		require.True(t, ids[id1])
		require.True(t, ids[id3])