import { Kysely } from 'kysely'; import { DateTime } from 'luxon'; import { DB } from 'src/db'; import { ImmichEnvironment, JobName, JobStatus } from 'src/enum'; import { UserService } from 'src/services/user.service'; import { mediumFactory, newMediumService } from 'test/medium.factory'; import { factory } from 'test/small.factory'; import { getKyselyDB } from 'test/utils'; describe(UserService.name, () => { let defaultDatabase: Kysely; const createSut = (db?: Kysely) => { process.env.IMMICH_ENV = ImmichEnvironment.TESTING; return newMediumService(UserService, { database: db || defaultDatabase, repos: { user: 'real', crypto: 'real', config: 'real', job: 'mock', systemMetadata: 'real', }, }); }; beforeAll(async () => { defaultDatabase = await getKyselyDB(); const { repos } = createSut(); await repos.user.create({ isAdmin: true, email: 'admin@immich.cloud' }); }); describe('create', () => { it('should create a user', async () => { const { sut } = createSut(); const user = mediumFactory.userInsert(); await expect(sut.createUser({ name: user.name, email: user.email })).resolves.toEqual( expect.objectContaining({ name: user.name, email: user.email }), ); }); it('should reject user with duplicate email', async () => { const { sut } = createSut(); const user = mediumFactory.userInsert(); await expect(sut.createUser({ email: user.email })).resolves.toMatchObject({ email: user.email }); await expect(sut.createUser({ email: user.email })).rejects.toThrow('User exists'); }); it('should not return password', async () => { const { sut } = createSut(); const dto = mediumFactory.userInsert({ password: 'password' }); const user = await sut.createUser({ email: dto.email, password: 'password' }); expect((user as any).password).toBeUndefined(); }); }); describe('search', () => { it('should get users', async () => { const { sut, repos } = createSut(); const user1 = mediumFactory.userInsert(); const user2 = mediumFactory.userInsert(); await Promise.all([repos.user.create(user1), repos.user.create(user2)]); const auth = factory.auth({ user: user1 }); await expect(sut.search(auth)).resolves.toEqual( expect.arrayContaining([ expect.objectContaining({ email: user1.email }), expect.objectContaining({ email: user2.email }), ]), ); }); }); describe('get', () => { it('should get a user', async () => { const { sut, repos } = createSut(); const user = mediumFactory.userInsert(); await repos.user.create(user); await expect(sut.get(user.id)).resolves.toEqual( expect.objectContaining({ id: user.id, name: user.name, email: user.email, }), ); }); it('should not return password', async () => { const { sut, repos } = createSut(); const user = mediumFactory.userInsert(); await repos.user.create(user); const result = await sut.get(user.id); expect((result as any).password).toBeUndefined(); }); }); describe('updateMe', () => { it('should update a user', async () => { const { sut, repos: repositories } = createSut(); const before = await repositories.user.create(mediumFactory.userInsert()); const auth = factory.auth({ user: { id: before.id } }); const after = await sut.updateMe(auth, { name: `${before.name} Updated` }); expect(before.updatedAt).toBeDefined(); expect(after.updatedAt).toBeDefined(); expect(before.updatedAt).not.toEqual(after.updatedAt); }); }); describe('setLicense', () => { it('should set a license', async () => { const license = { licenseKey: 'IMCL-FF69-TUK1-RWZU-V9Q8-QGQS-S5GC-X4R2-UFK4', activationKey: 'KuX8KsktrBSiXpQMAH0zLgA5SpijXVr_PDkzLdWUlAogCTMBZ0I3KCHXK0eE9EEd7harxup8_EHMeqAWeHo5VQzol6LGECpFv585U9asXD4Zc-UXt3mhJr2uhazqipBIBwJA2YhmUCDy8hiyiGsukDQNu9Rg9C77UeoKuZBWVjWUBWG0mc1iRqfvF0faVM20w53czAzlhaMxzVGc3Oimbd7xi_CAMSujF_2y8QpA3X2fOVkQkzdcH9lV0COejl7IyH27zQQ9HrlrXv3Lai5Hw67kNkaSjmunVBxC5PS0TpKoc9SfBJMaAGWnaDbjhjYUrm-8nIDQnoeEAidDXVAdPw', }; const { sut, repos } = createSut(); const user = mediumFactory.userInsert(); await repos.user.create(user); const auth = factory.auth({ user: { id: user.id } }); await expect(sut.getLicense(auth)).rejects.toThrowError(); const after = await sut.setLicense(auth, license); expect(after.licenseKey).toEqual(license.licenseKey); expect(after.activationKey).toEqual(license.activationKey); const getResponse = await sut.getLicense(auth); expect(getResponse).toEqual(after); }); }); describe.sequential('handleUserDeleteCheck', () => { beforeEach(async () => { const { sut } = createSut(); // These tests specifically have to be sequential otherwise we hit race conditions with config changes applying in incorrect tests const config = await sut.getConfig({ withCache: false }); config.user.deleteDelay = 7; await sut.updateConfig(config); }); it('should work when there are no deleted users', async () => { const { sut, mocks } = createSut(); mocks.job.queueAll.mockResolvedValue(void 0); await expect(sut.handleUserDeleteCheck()).resolves.toEqual(JobStatus.SUCCESS); expect(mocks.job.queueAll).toHaveBeenCalledExactlyOnceWith([]); }); it('should work when there is a user to delete', async () => { const { sut, repos, mocks } = createSut(await getKyselyDB()); mocks.job.queueAll.mockResolvedValue(void 0); const user = mediumFactory.userInsert({ deletedAt: DateTime.now().minus({ days: 60 }).toJSDate() }); await repos.user.create(user); await expect(sut.handleUserDeleteCheck()).resolves.toEqual(JobStatus.SUCCESS); expect(mocks.job.queueAll).toHaveBeenCalledExactlyOnceWith([ { name: JobName.USER_DELETION, data: { id: user.id } }, ]); }); it('should skip a recently deleted user', async () => { const { sut, repos, mocks } = createSut(await getKyselyDB()); mocks.job.queueAll.mockResolvedValue(void 0); const user = mediumFactory.userInsert({ deletedAt: DateTime.now().minus({ days: 5 }).toJSDate() }); await repos.user.create(user); await expect(sut.handleUserDeleteCheck()).resolves.toEqual(JobStatus.SUCCESS); expect(mocks.job.queueAll).toHaveBeenCalledExactlyOnceWith([]); }); it('should respect a custom user delete delay', async () => { const { sut, repos, mocks } = createSut(await getKyselyDB()); mocks.job.queueAll.mockResolvedValue(void 0); const user = mediumFactory.userInsert({ deletedAt: DateTime.now().minus({ days: 25 }).toJSDate() }); await repos.user.create(user); const config = await sut.getConfig({ withCache: false }); config.user.deleteDelay = 30; await sut.updateConfig(config); await expect(sut.handleUserDeleteCheck()).resolves.toEqual(JobStatus.SUCCESS); expect(mocks.job.queueAll).toHaveBeenCalledExactlyOnceWith([]); }); }); });