weseek__growi/apps/app/src/features/external-user-group/server/models/external-user-group-relation.integ.ts

import mongoose from 'mongoose';

import ExternalUserGroup from './external-user-group';
import ExternalUserGroupRelation from './external-user-group-relation';

// TODO: use actual user model after ~/server/models/user.js becomes importable in vitest
// ref: https://github.com/vitest-dev/vitest/issues/846
const userSchema = new mongoose.Schema({
  name: { type: String },
  username: { type: String, required: true, unique: true },
  email: { type: String, unique: true, sparse: true },
}, {
  timestamps: true,
});
const User = mongoose.model('User', userSchema);

describe('ExternalUserGroupRelation model', () => {
  let user1;
  const userId1 = new mongoose.Types.ObjectId();

  let user2;
  const userId2 = new mongoose.Types.ObjectId();

  const groupId1 = new mongoose.Types.ObjectId();
  const groupId2 = new mongoose.Types.ObjectId();
  const groupId3 = new mongoose.Types.ObjectId();

  beforeAll(async() => {
    user1 = await User.create({
      _id: userId1, name: 'user1', username: 'user1', email: 'user1@example.com',
    });

    user2 = await User.create({
      _id: userId2, name: 'user2', username: 'user2', email: 'user2@example.com',
    });

    await ExternalUserGroup.insertMany([
      {
        _id: groupId1, name: 'test group 1', externalId: 'testExternalId', provider: 'testProvider',
      },
      {
        _id: groupId2, name: 'test group 2', externalId: 'testExternalId2', provider: 'testProvider',
      },
      {
        _id: groupId3, name: 'test group 3', externalId: 'testExternalId3', provider: 'testProvider',
      },
    ]);
  });

  afterEach(async() => {
    await ExternalUserGroupRelation.deleteMany();
  });

  describe('createRelations', () => {
    it('creates relation for user', async() => {
      await ExternalUserGroupRelation.createRelations([groupId1, groupId2], user1);
      const relations = await ExternalUserGroupRelation.find();
      const idCombinations = relations.map((relation) => {
        return [relation.relatedGroup, relation.relatedUser];
      });
      expect(idCombinations).toStrictEqual([[groupId1, userId1], [groupId2, userId1]]);
    });
  });

  describe('removeAllInvalidRelations', () => {
    beforeAll(async() => {
      const nonExistentGroupId1 = new mongoose.Types.ObjectId();
      const nonExistentGroupId2 = new mongoose.Types.ObjectId();
      await ExternalUserGroupRelation.createRelations([nonExistentGroupId1, nonExistentGroupId2], user1);
    });

    it('removes invalid relations', async() => {
      const relationsBeforeRemoval = await ExternalUserGroupRelation.find();
      expect(relationsBeforeRemoval.length).not.toBe(0);

      await ExternalUserGroupRelation.removeAllInvalidRelations();

      const relationsAfterRemoval = await ExternalUserGroupRelation.find();
      expect(relationsAfterRemoval.length).toBe(0);
    });
  });

  describe('findAllUserIdsForUserGroups', () => {
    beforeAll(async() => {
      await ExternalUserGroupRelation.createRelations([groupId1, groupId2], user1);
      await ExternalUserGroupRelation.create({ relatedGroup: groupId3, relatedUser: user2._id });
    });

    it('finds all unique user ids for specified user groups', async() => {
      const userIds = await ExternalUserGroupRelation.findAllUserIdsForUserGroups([groupId1, groupId2, groupId3]);
      expect(userIds).toStrictEqual([userId1.toString(), user2._id.toString()]);
    });
  });

  describe('findAllUserGroupIdsRelatedToUser', () => {
    beforeAll(async() => {
      await ExternalUserGroupRelation.createRelations([groupId1, groupId2], user1);
      await ExternalUserGroupRelation.create({ relatedGroup: groupId3, relatedUser: user2._id });
    });

    it('finds all group ids related to user', async() => {
      const groupIds = await ExternalUserGroupRelation.findAllUserGroupIdsRelatedToUser(user1);
      expect(groupIds).toStrictEqual([groupId1, groupId2]);

      const groupIds2 = await ExternalUserGroupRelation.findAllUserGroupIdsRelatedToUser(user2);
      expect(groupIds2).toStrictEqual([groupId3]);
    });
  });

  describe('findAllGroupsForUser', () => {
    beforeAll(async() => {
      await ExternalUserGroupRelation.createRelations([groupId1, groupId2], user1);
      await ExternalUserGroupRelation.create({ relatedGroup: groupId3, relatedUser: user2._id });
    });

    it('finds all groups related to user', async() => {
      const groups = await ExternalUserGroupRelation.findAllGroupsForUser(user1);
      const groupIds = groups.map(group => group._id);
      expect(groupIds).toStrictEqual([groupId1, groupId2]);

      const groups2 = await ExternalUserGroupRelation.findAllGroupsForUser(user2);
      const groupIds2 = groups2.map(group => group._id);
      expect(groupIds2).toStrictEqual([groupId3]);
    });
  });
});