Function waitForChatProcessed

coderd/x/chatd/chatd_test.go:5766–5787

waitForChatProcessed waits for a wake-triggered processOnce to fully complete for the given chat. It polls until the chat leaves both pending and running states (meaning processChat has finished its cleanup and updated the DB), then calls WaitUntilIdleForTest. Waiting for a terminal state (not just

(
	ctx context.Context,
	t *testing.T,
	db database.Store,
	chatID uuid.UUID,
	server *chatd.Server,
)

// waited for status != pending, we could call Wait() while Add(1)
// hasn't happened yet.
func waitForChatProcessed(
	ctx context.Context,
	t *testing.T,
	db database.Store,
	chatID uuid.UUID,
	server *chatd.Server,
) {
	t.Helper()
	require.Eventually(t, func() bool {
		c, err := db.GetChatByID(ctx, chatID)
		if err != nil {
			return false
		}
		// Wait until the chat reaches a terminal state. Neither
		// pending (waiting to be acquired) nor running (being
		// processed). This guarantees that inflight.Add(1) has
		// already been called by processOnce.
		return c.Status != database.ChatStatusPending &&
			c.Status != database.ChatStatusRunning
	}, testutil.WaitShort, testutil.IntervalFast)
	chatd.WaitUntilIdleForTest(server)
}

// newTestServer creates a passive server that never calls
// processOnce on its own.