From 59efc5fea7982c615d0fa1d16f9e9e277dc07d1f Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Tue, 28 Jun 2022 10:14:15 +0800
Subject: [PATCH] Add test

---
 packages/api/src/elastic/types.ts           |  2 ++
 packages/api/test/resolvers/article.test.ts | 12 ++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/packages/api/src/elastic/types.ts b/packages/api/src/elastic/types.ts
index 3a89c8da5..99dc06890 100644
--- a/packages/api/src/elastic/types.ts
+++ b/packages/api/src/elastic/types.ts
@@ -232,6 +232,8 @@ export interface SearchItem {
   readAt?: Date
   savedAt: Date
   updatedAt?: Date
+  labels?: Label[]
+  highlights?: Highlight[]
 }
 
 const keys = ['_id', 'url', 'slug', 'userId', 'uploadFileId', 'state'] as const
diff --git a/packages/api/test/resolvers/article.test.ts b/packages/api/test/resolvers/article.test.ts
index ef82378f3..8353b68ca 100644
--- a/packages/api/test/resolvers/article.test.ts
+++ b/packages/api/test/resolvers/article.test.ts
@@ -180,6 +180,9 @@ const searchQuery = (keyword = '') => {
             url
             createdAt
             updatedAt
+            highlights {
+              id
+            }
           }
         }
         pageInfo {
@@ -1021,6 +1024,15 @@ describe('Article API', () => {
         expect(res.body.data.search.edges[3].node.id).to.eq(pages[1].id)
         expect(res.body.data.search.edges[4].node.id).to.eq(pages[0].id)
       })
+
+      it('should return highlights in pages', async () => {
+        const res = await graphqlRequest(query, authToken).expect(200)
+
+        expect(res.body.data.search.edges[0].node.highlights.length).to.eql(1)
+        expect(res.body.data.search.edges[0].node.highlights[0].id).to.eq(
+          highlights[4].id
+        )
+      })
     })
 
     context('when type:highlights is in the query', () => {