fix: hydrate poll state with thread parent messages when querying thread (#1696)

Author: khushal87

src/client.ts

@@ -3358,6 +3358,10 @@ export class StreamChat {
       requestBody,
     );
 
+    // Hydrate the polls for the parent messages of the threads
+    const parentMessages = response.threads.map((thread) => thread.parent_message);
+    this.polls.hydratePollCache(parentMessages);
+
     return {
       threads: response.threads.map(
         (thread) => new Thread({ client: this, threadData: thread }),

test/unit/client.test.js

@@ -7,6 +7,7 @@ import { StreamChat } from '../../src/client';
 import { ConnectionState } from '../../src/connection_fallback';
 import { StableWSConnection } from '../../src/connection';
 import { mockChannelQueryResponse } from './test-utils/mockChannelQueryResponse';
+import { generateThreadResponse } from './test-utils/generateThreadResponse';
 import { DEFAULT_QUERY_CHANNEL_MESSAGE_LIST_PAGE_SIZE } from '../../src/constants';
 
 import {
@@ -787,6 +788,32 @@ describe('StreamChat.queryChannels', async () => {
 	});
 });
 
+describe('StreamChat.queryThreads', () => {
+	it('returns threads and next, and hydrates poll cache with parent messages', async () => {
+		const client = await getClientWithUser();
+		const parentMessage = generateMsg();
+		const rawThread = generateThreadResponse(
+			mockChannelQueryResponse.channel,
+			parentMessage,
+		);
+		const apiResponse = { threads: [rawThread], next: undefined };
+
+		const postStub = sinon.stub(client, 'post');
+		postStub.onFirstCall().resolves(apiResponse);
+		const hydratePollCacheSpy = sinon.spy(client.polls, 'hydratePollCache');
+
+		const result = await client.queryThreads();
+
+		expect(result.threads).to.have.lengthOf(1);
+		expect(result.threads[0].id).to.equal(parentMessage.id);
+		expect(result.next).to.be.undefined;
+		expect(hydratePollCacheSpy.calledOnce).to.be.true;
+		expect(hydratePollCacheSpy.calledWith([parentMessage])).to.be.true;
+
+		postStub.restore();
+	});
+});
+
 describe('StreamChat.queryReactions', () => {
 	let client;
 	let dispatchSpy;