From d2c16624c3e7935ea1d06209d751111fc2974453 Mon Sep 17 00:00:00 2001
From: Jackson Harper <jacksonh@gmail.com>
Date: Tue, 20 Aug 2024 17:05:07 +0800
Subject: [PATCH] Add tests for shortcuts service

---
 packages/api/test/routers/shortcuts.test.ts | 78 +++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 packages/api/test/routers/shortcuts.test.ts

diff --git a/packages/api/test/routers/shortcuts.test.ts b/packages/api/test/routers/shortcuts.test.ts
new file mode 100644
index 000000000..50fb7b2cb
--- /dev/null
+++ b/packages/api/test/routers/shortcuts.test.ts
@@ -0,0 +1,78 @@
+import { expect } from 'chai'
+import 'mocha'
+import { In } from 'typeorm'
+import { StatusType, User } from '../../src/entity/user'
+import {
+  createUsers,
+  deleteUser,
+  deleteUsers,
+  findUsersByIds,
+} from '../../src/services/user'
+import { request } from '../util'
+import { createTestUser } from '../db'
+import { v4 as uuid } from 'uuid'
+
+import {
+  getShortcuts,
+  setShortcuts,
+} from '../../src/services/user_personalization'
+
+describe('Shortcuts Router', () => {
+  const token = process.env.PUBSUB_VERIFICATION_TOKEN || ''
+  let user: User
+
+  describe('default', () => {
+    before(async () => {
+      user = await createTestUser('fakeUser')
+    })
+
+    after(async () => {
+      await deleteUser(user.id)
+    })
+
+    it('gets the default shortcuts', async () => {
+      const shortcuts = await getShortcuts(user.id)
+      expect(shortcuts.length).to.eq(3) // labels, subscriptions, searches
+    })
+
+    it('can set the shortcuts', async () => {
+      const shortcuts = await setShortcuts(user.id, [
+        {
+          id: uuid(),
+          type: 'folder',
+          section: 'library',
+          name: 'test folder',
+        },
+      ])
+      const result = await getShortcuts(user.id)
+      expect(result.length).to.eq(1)
+    })
+
+    it('can modify shortcuts', async () => {
+      await setShortcuts(user.id, [
+        {
+          id: uuid(),
+          type: 'folder',
+          section: 'library',
+          name: 'test folder',
+        },
+      ])
+      await setShortcuts(user.id, [
+        {
+          id: uuid(),
+          type: 'folder',
+          section: 'library',
+          name: 'test folder',
+        },
+        {
+          id: uuid(),
+          type: 'folder',
+          section: 'library',
+          name: 'test folder 2',
+        },
+      ])
+      const result = await getShortcuts(user.id)
+      expect(result.length).to.eq(2)
+    })
+  })
+})