@lobehub/chat 0.162.24 → 0.163.0

package/src/database/server/models/__tests__/plugin.test.ts

@@ -0,0 +1,172 @@
+// @vitest-environment node
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { getTestDBInstance } from '@/database/server/core/dbForTest';
+
+import { NewInstalledPlugin, installedPlugins, users } from '../../schemas/lobechat';
+import { PluginModel } from '../plugin';
+
+let serverDB = await getTestDBInstance();
+
+vi.mock('@/database/server/core/db', async () => ({
+  get serverDB() {
+    return serverDB;
+  },
+}));
+
+const userId = 'plugin-db';
+const pluginModel = new PluginModel(userId);
+
+beforeEach(async () => {
+  await serverDB.transaction(async (trx) => {
+    await trx.delete(users);
+    await trx.insert(users).values([{ id: userId }, { id: '456' }]);
+  });
+});
+
+afterEach(async () => {
+  await serverDB.delete(users);
+});
+
+describe('PluginModel', () => {
+  describe('create', () => {
+    it('should create a new installed plugin', async () => {
+      const params = {
+        type: 'plugin',
+        identifier: 'test-plugin',
+        manifest: { identifier: 'Test Plugin' },
+        customParams: { manifestUrl: 'abc123' },
+      } as NewInstalledPlugin;
+
+      const result = await pluginModel.create(params);
+
+      expect(result.userId).toBe(userId);
+      expect(result.type).toBe(params.type);
+      expect(result.identifier).toBe(params.identifier);
+      expect(result.manifest).toEqual(params.manifest);
+      expect(result.customParams).toEqual(params.customParams);
+    });
+  });
+
+  describe('delete', () => {
+    it('should delete an installed plugin by identifier', async () => {
+      await serverDB.insert(installedPlugins).values({
+        userId,
+        type: 'plugin',
+        identifier: 'test-plugin',
+        manifest: { name: 'Test Plugin' },
+      } as unknown as NewInstalledPlugin);
+
+      await pluginModel.delete('test-plugin');
+
+      const result = await serverDB.select().from(installedPlugins);
+      expect(result).toHaveLength(0);
+    });
+  });
+
+  describe('deleteAll', () => {
+    it('should delete all installed plugins for the user', async () => {
+      await serverDB.insert(installedPlugins).values([
+        {
+          userId,
+          type: 'plugin',
+          identifier: 'test-plugin-1',
+          manifest: { name: 'Test Plugin 1' },
+        },
+        {
+          userId,
+          type: 'plugin',
+          identifier: 'test-plugin-2',
+          manifest: { name: 'Test Plugin 2' },
+        },
+        {
+          userId: '456',
+          type: 'plugin',
+          identifier: 'test-plugin-3',
+          manifest: { name: 'Test Plugin 3' },
+        },
+      ] as unknown as NewInstalledPlugin[]);
+
+      await pluginModel.deleteAll();
+
+      const result = await serverDB.select().from(installedPlugins);
+      expect(result).toHaveLength(1);
+      expect(result[0].userId).toBe('456');
+    });
+  });
+
+  describe('query', () => {
+    it('should query installed plugins for the user', async () => {
+      await serverDB.insert(installedPlugins).values([
+        {
+          userId,
+          type: 'plugin',
+          identifier: 'test-plugin-1',
+          manifest: { name: 'Test Plugin 1' },
+          createdAt: new Date('2023-01-01'),
+        },
+        {
+          userId,
+          type: 'plugin',
+          identifier: 'test-plugin-2',
+          manifest: { name: 'Test Plugin 2' },
+          createdAt: new Date('2023-02-01'),
+        },
+        {
+          userId: '456',
+          type: 'plugin',
+          identifier: 'test-plugin-3',
+          manifest: { name: 'Test Plugin 3' },
+          createdAt: new Date('2023-03-01'),
+        },
+      ] as unknown as NewInstalledPlugin[]);
+
+      const result = await pluginModel.query();
+
+      expect(result).toHaveLength(2);
+      expect(result[0].identifier).toBe('test-plugin-2');
+      expect(result[1].identifier).toBe('test-plugin-1');
+    });
+  });
+
+  describe('findById', () => {
+    it('should find an installed plugin by identifier', async () => {
+      await serverDB.insert(installedPlugins).values([
+        {
+          userId,
+          type: 'plugin',
+          identifier: 'test-plugin-1',
+          manifest: { name: 'Test Plugin 1' },
+        },
+        {
+          userId: '456',
+          type: 'plugin',
+          identifier: 'test-plugin-2',
+          manifest: { name: 'Test Plugin 2' },
+        },
+      ] as unknown as NewInstalledPlugin[]);
+
+      const result = await pluginModel.findById('test-plugin-1');
+
+      expect(result?.userId).toBe(userId);
+      expect(result?.identifier).toBe('test-plugin-1');
+    });
+  });
+
+  describe('update', () => {
+    it('should update an installed plugin', async () => {
+      await serverDB.insert(installedPlugins).values({
+        userId,
+        type: 'plugin',
+        identifier: 'test-plugin',
+        manifest: {},
+        settings: { enabled: true },
+      } as unknown as NewInstalledPlugin);
+
+      await pluginModel.update('test-plugin', { settings: { enabled: false } });
+
+      const result = await pluginModel.findById('test-plugin');
+      expect(result?.settings).toEqual({ enabled: false });
+    });
+  });
+});