immich/server/test/medium/specs/sync/sync-partner-asset-exif.spec.ts

import { Kysely } from 'kysely';
import { DB } from 'src/db';
import { SyncEntityType, SyncRequestType } from 'src/enum';
import { mediumFactory, newSyncAuthUser, newSyncTest } from 'test/medium.factory';
import { factory } from 'test/small.factory';
import { getKyselyDB, wait } from 'test/utils';

let defaultDatabase: Kysely<DB>;

const setup = async (db?: Kysely<DB>) => {
  const database = db || defaultDatabase;
  const result = newSyncTest({ db: database });
  const { auth, create } = newSyncAuthUser();
  await create(database);
  return { ...result, auth };
};

beforeAll(async () => {
  defaultDatabase = await getKyselyDB();
});

describe.concurrent(SyncRequestType.PartnerAssetExifsV1, () => {
  it('should detect and sync the first partner asset exif', async () => {
    const { auth, sut, getRepository, testSync } = await setup();

    const userRepo = getRepository('user');
    const user2 = mediumFactory.userInsert();
    await userRepo.create(user2);

    const partnerRepo = getRepository('partner');
    await partnerRepo.create({ sharedById: user2.id, sharedWithId: auth.user.id });

    const assetRepo = getRepository('asset');
    const asset = mediumFactory.assetInsert({ ownerId: user2.id });
    await assetRepo.create(asset);
    await assetRepo.upsertExif({ assetId: asset.id, make: 'Canon' });

    const initialSyncResponse = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    expect(initialSyncResponse).toHaveLength(1);
    expect(initialSyncResponse).toEqual(
      expect.arrayContaining([
        {
          ack: expect.any(String),
          data: {
            assetId: asset.id,
            city: null,
            country: null,
            dateTimeOriginal: null,
            description: '',
            exifImageHeight: null,
            exifImageWidth: null,
            exposureTime: null,
            fNumber: null,
            fileSizeInByte: null,
            focalLength: null,
            fps: null,
            iso: null,
            latitude: null,
            lensModel: null,
            longitude: null,
            make: 'Canon',
            model: null,
            modifyDate: null,
            orientation: null,
            profileDescription: null,
            projectionType: null,
            rating: null,
            state: null,
            timeZone: null,
          },
          type: SyncEntityType.PartnerAssetExifV1,
        },
      ]),
    );

    const acks = [initialSyncResponse[0].ack];
    await sut.setAcks(auth, { acks });

    const ackSyncResponse = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    expect(ackSyncResponse).toHaveLength(0);
  });

  it('should not sync partner asset exif for own user', async () => {
    const { auth, getRepository, testSync } = await setup();

    const userRepo = getRepository('user');
    const user2 = mediumFactory.userInsert();
    await userRepo.create(user2);

    const partnerRepo = getRepository('partner');
    await partnerRepo.create({ sharedById: user2.id, sharedWithId: auth.user.id });

    const assetRepo = getRepository('asset');
    const asset = mediumFactory.assetInsert({ ownerId: auth.user.id });
    await assetRepo.create(asset);
    await assetRepo.upsertExif({ assetId: asset.id, make: 'Canon' });

    await expect(testSync(auth, [SyncRequestType.AssetExifsV1])).resolves.toHaveLength(1);
    await expect(testSync(auth, [SyncRequestType.PartnerAssetExifsV1])).resolves.toHaveLength(0);
  });

  it('should not sync partner asset exif for unrelated user', async () => {
    const { auth, getRepository, testSync } = await setup();

    const userRepo = getRepository('user');

    const user2 = mediumFactory.userInsert();
    const user3 = mediumFactory.userInsert();
    await Promise.all([userRepo.create(user2), userRepo.create(user3)]);

    const partnerRepo = getRepository('partner');
    await partnerRepo.create({ sharedById: user2.id, sharedWithId: auth.user.id });

    const assetRepo = getRepository('asset');
    const asset = mediumFactory.assetInsert({ ownerId: user3.id });
    await assetRepo.create(asset);
    await assetRepo.upsertExif({ assetId: asset.id, make: 'Canon' });

    const sessionRepo = getRepository('session');
    const session = mediumFactory.sessionInsert({ userId: user3.id });
    await sessionRepo.create(session);

    const authUser3 = factory.auth({ session, user: user3 });
    await expect(testSync(authUser3, [SyncRequestType.AssetExifsV1])).resolves.toHaveLength(1);
    await expect(testSync(auth, [SyncRequestType.PartnerAssetExifsV1])).resolves.toHaveLength(0);
  });

  it('should backfill partner asset exif when a partner shared their library with you', async () => {
    const { auth, sut, getRepository, testSync } = await setup();

    const userRepo = getRepository('user');
    const user2 = mediumFactory.userInsert();
    const user3 = mediumFactory.userInsert();
    await userRepo.create(user2);
    await userRepo.create(user3);

    const assetRepo = getRepository('asset');
    const assetUser3 = mediumFactory.assetInsert({ ownerId: user3.id });
    const assetUser2 = mediumFactory.assetInsert({ ownerId: user2.id });
    await assetRepo.create(assetUser3);
    await assetRepo.upsertExif({ assetId: assetUser3.id, make: 'Canon' });
    await wait(2);
    await assetRepo.create(assetUser2);
    await assetRepo.upsertExif({ assetId: assetUser2.id, make: 'Canon' });

    const partnerRepo = getRepository('partner');
    await partnerRepo.create({ sharedById: user2.id, sharedWithId: auth.user.id });

    const response = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    expect(response).toHaveLength(1);
    expect(response).toEqual(
      expect.arrayContaining([
        {
          ack: expect.any(String),
          data: expect.objectContaining({
            assetId: assetUser2.id,
          }),
          type: SyncEntityType.PartnerAssetExifV1,
        },
      ]),
    );

    const acks = response.map(({ ack }) => ack);
    await sut.setAcks(auth, { acks });

    await partnerRepo.create({ sharedById: user3.id, sharedWithId: auth.user.id });
    const backfillResponse = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    expect(backfillResponse).toHaveLength(2);
    expect(backfillResponse).toEqual(
      expect.arrayContaining([
        {
          ack: expect.any(String),
          data: expect.objectContaining({
            assetId: assetUser3.id,
          }),
          type: SyncEntityType.PartnerAssetExifBackfillV1,
        },
        {
          ack: expect.any(String),
          data: {},
          type: SyncEntityType.SyncAckV1,
        },
      ]),
    );

    const backfillAck = backfillResponse[1].ack;
    await sut.setAcks(auth, { acks: [backfillAck] });

    const finalResponse = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    const finalAcks = finalResponse.map(({ ack }) => ack);
    expect(finalAcks).toEqual([]);
  });

  it('should only backfill partner assets created prior to the current partner asset checkpoint', async () => {
    const { auth, sut, getRepository, testSync } = await setup();

    const userRepo = getRepository('user');
    const user2 = mediumFactory.userInsert();
    const user3 = mediumFactory.userInsert();
    await userRepo.create(user2);
    await userRepo.create(user3);

    const assetRepo = getRepository('asset');
    const assetUser3 = mediumFactory.assetInsert({ ownerId: user3.id });
    const assetUser2 = mediumFactory.assetInsert({ ownerId: user2.id });
    const asset2User3 = mediumFactory.assetInsert({ ownerId: user3.id });
    await assetRepo.create(assetUser3);
    await assetRepo.upsertExif({ assetId: assetUser3.id, make: 'Canon' });
    await wait(2);
    await assetRepo.create(assetUser2);
    await assetRepo.upsertExif({ assetId: assetUser2.id, make: 'Canon' });
    await wait(2);
    await assetRepo.create(asset2User3);
    await assetRepo.upsertExif({ assetId: asset2User3.id, make: 'Canon' });

    const partnerRepo = getRepository('partner');
    await partnerRepo.create({ sharedById: user2.id, sharedWithId: auth.user.id });

    const response = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    expect(response).toHaveLength(1);
    expect(response).toEqual(
      expect.arrayContaining([
        {
          ack: expect.any(String),
          data: expect.objectContaining({
            assetId: assetUser2.id,
          }),
          type: SyncEntityType.PartnerAssetExifV1,
        },
      ]),
    );

    const acks = response.map(({ ack }) => ack);
    await sut.setAcks(auth, { acks });

    await partnerRepo.create({ sharedById: user3.id, sharedWithId: auth.user.id });
    const backfillResponse = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    expect(backfillResponse).toHaveLength(3);
    expect(backfillResponse).toEqual(
      expect.arrayContaining([
        {
          ack: expect.any(String),
          data: expect.objectContaining({
            assetId: assetUser3.id,
          }),
          type: SyncEntityType.PartnerAssetExifBackfillV1,
        },
        {
          ack: expect.stringContaining(SyncEntityType.PartnerAssetExifBackfillV1),
          data: {},
          type: SyncEntityType.SyncAckV1,
        },
        {
          ack: expect.any(String),
          data: expect.objectContaining({
            assetId: asset2User3.id,
          }),
          type: SyncEntityType.PartnerAssetExifV1,
        },
      ]),
    );

    const backfillAck = backfillResponse[1].ack;
    const partnerAssetAck = backfillResponse[2].ack;
    await sut.setAcks(auth, { acks: [backfillAck, partnerAssetAck] });

    const finalResponse = await testSync(auth, [SyncRequestType.PartnerAssetExifsV1]);

    const finalAcks = finalResponse.map(({ ack }) => ack);
    expect(finalAcks).toEqual([]);
  });
});