From 2460d85460b1cbc2ba7f2f38aeeda08b9aa40b1f Mon Sep 17 00:00:00 2001
From: Chewbacca <dice.warrant-02@icloud.com>
Date: Wed, 9 Nov 2022 10:28:15 -0500
Subject: [PATCH] Add more tests to chats query

---
 app/soapbox/queries/__tests__/chats.test.ts | 63 ++++++++++++++++++++-
 1 file changed, 62 insertions(+), 1 deletion(-)

diff --git a/app/soapbox/queries/__tests__/chats.test.ts b/app/soapbox/queries/__tests__/chats.test.ts
index 6680bbf13..61ad4f99d 100644
--- a/app/soapbox/queries/__tests__/chats.test.ts
+++ b/app/soapbox/queries/__tests__/chats.test.ts
@@ -8,7 +8,7 @@ import { normalizeRelationship } from 'soapbox/normalizers';
 import { flattenPages } from 'soapbox/utils/queries';
 
 import { IAccount } from '../accounts';
-import { ChatKeys, IChat, IChatMessage, useChat, useChatActions, useChatMessages, useChats } from '../chats';
+import { ChatKeys, IChat, IChatMessage, isLastMessage, useChat, useChatActions, useChatMessages, useChats } from '../chats';
 
 jest.mock('soapbox/utils/queries');
 
@@ -43,6 +43,67 @@ const buildChatMessage = (id: string): IChatMessage => ({
   unread: true,
 });
 
+describe('ChatKeys', () => {
+  it('has a "chat" key', () => {
+    const id = '1';
+
+    expect(ChatKeys.chat(id)).toEqual(['chats', 'chat', id]);
+  });
+
+  it('has a "chatMessages" key', () => {
+    const id = '1';
+
+    expect(ChatKeys.chatMessages(id)).toEqual(['chats', 'messages', id]);
+  });
+
+  it('has a "chatSearch" key', () => {
+    const searchQuery = 'che';
+
+    expect(ChatKeys.chatSearch()).toEqual(['chats', 'search']);
+    expect(ChatKeys.chatSearch(searchQuery)).toEqual(['chats', 'search', searchQuery]);
+  });
+});
+
+describe('isLastMessage', () => {
+  describe('when its the last message', () => {
+    it('is truthy', () => {
+      const id = '5';
+      const newChat = { ...chat, last_message: { id } } as any;
+      const initialQueryData = {
+        pages: [
+          { result: [newChat], hasMore: false, link: undefined },
+        ],
+        pageParams: [undefined],
+      };
+      const initialFlattenedData = flattenPages(initialQueryData);
+      expect(sumBy(initialFlattenedData, (chat: IChat) => chat.unread)).toBe(0);
+
+      queryClient.setQueryData(ChatKeys.chatSearch(), initialQueryData);
+
+      expect(isLastMessage(id)).toBeTruthy();
+    });
+  });
+
+  describe('when its not the last message', () => {
+    it('is not truthy', () => {
+      const id = '5';
+      const newChat = { ...chat, last_message: { id } } as any;
+      const initialQueryData = {
+        pages: [
+          { result: [newChat], hasMore: false, link: undefined },
+        ],
+        pageParams: [undefined],
+      };
+      const initialFlattenedData = flattenPages(initialQueryData);
+      expect(sumBy(initialFlattenedData, (chat: IChat) => chat.unread)).toBe(0);
+
+      queryClient.setQueryData(ChatKeys.chatSearch(), initialQueryData);
+
+      expect(isLastMessage('10')).not.toBeTruthy();
+    });
+  });
+});
+
 describe('useChatMessages', () => {
   let store: ReturnType<typeof mockStore>;