All files / server/src/features/assignment/run-assignment runAssignmentTestUtils.ts

77.96% Statements 46/59
66.66% Branches 16/24
100% Functions 8/8
77.96% Lines 46/59

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162                                        4x   4x         327x 223x       104x 3702x     104x 104x               4x     8x   8x 8x     8x   8x 8x     8x   8x 8x     8x   8x         8x   8x             1120x 327x 7688x     327x           327x 327x         327x   327x   572x           327x                     8x       8x     4x     8x   8x 281x 297x   281x     8x 8x     4x             8x             8x   8x    
import { expect } from "vitest";
import dayjs from "dayjs";
import { unsafelyUnwrap } from "server/test/utils/unsafelyUnwrapResult";
import { EventLogAction } from "shared/types/models/eventLog";
import { logger } from "server/utils/logger";
import { User } from "shared/types/models/user";
import { findUsers } from "server/features/user/userRepository";
import { findDirectSignups } from "server/features/direct-signup/directSignupRepository";
import {
  Result,
  makeErrorResult,
  makeSuccessResult,
} from "shared/utils/result";
import { MongoDbError } from "shared/types/api/errors";
import { findProgramItems } from "server/features/program-item/programItemRepository";
import { getLotteryParticipantDirectSignups } from "server/features/assignment/utils/prepareAssignmentParams";
import { generateTestUsers } from "server/test/test-data-generation/generators/generateTestData";
import { createProgramItems } from "server/test/test-data-generation/generators/createProgramItems";
import { createLotterySignups } from "server/test/test-data-generation/generators/createLotterySignups";
 
export const firstLotterySignupSlot = 3;
 
const getGroupCreator = (
  users: User[],
  user: User,
): Result<User, MongoDbError> => {
  // Group creator (or not in a group) -> user is their own creator
  if (user.isGroupCreator || user.groupCode === "0") {
    return makeSuccessResult(user);
  }
 
  // Group member -> find the group's creator
  const groupCreator = users.find(
    (creator) => creator.isGroupCreator && creator.groupCode === user.groupCode,
  );
 
  Eif (groupCreator) {
    return makeSuccessResult(groupCreator);
  }
 
  logger.error(new Error(`Group creator not found for user ${user.username}`));
 
  return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
};
 
export const verifyUserSignups = async (): Promise<
  Result<void, MongoDbError>
> => {
  logger.info("Verify lottery signups and signups match for users");
 
  const usersResult = await findUsers();
  Iif (!usersResult.ok) {
    return usersResult;
  }
  const users = usersResult.value;
 
  const signupsResult = await findDirectSignups();
  Iif (!signupsResult.ok) {
    return signupsResult;
  }
  const signups = signupsResult.value;
 
  const programItemsResult = await findProgramItems();
  Iif (!programItemsResult.ok) {
    return programItemsResult;
  }
  const programItems = programItemsResult.value;
 
  const lotteryParticipantDirectSignups = getLotteryParticipantDirectSignups(
    signups,
    programItems,
  );
 
  let mismatchFound = false;
 
  for (const {
    programItemId,
    userSignups,
  } of lotteryParticipantDirectSignups) {
    // Verify group member signups match with group creators lotterySignups
    // If not in group -> user is group creator
 
    for (const userSignup of userSignups) {
      const matchingUser = users.find(
        (user) => user.username === userSignup.username,
      );
 
      Iif (!matchingUser) {
        logger.error(new Error(`No matching user: ${userSignup.username}`));
        mismatchFound = true;
        continue;
      }
 
      const groupCreatorResult = getGroupCreator(users, matchingUser);
      Iif (!groupCreatorResult.ok) {
        mismatchFound = true;
        continue;
      }
 
      const groupCreator = groupCreatorResult.value;
 
      const matchingCreatorLotterySignup = groupCreator.lotterySignups.some(
        (creatorLotterySignup) =>
          creatorLotterySignup.programItemId === programItemId &&
          dayjs(creatorLotterySignup.signedToStartTime).isSame(
            userSignup.signedToStartTime,
          ),
      );
 
      Iif (!matchingCreatorLotterySignup) {
        logger.error(
          new Error(
            `No matching signed program item found from group creator: ${userSignup.username} - ${programItemId}`,
          ),
        );
        mismatchFound = true;
      }
    }
  }
 
  Iif (mismatchFound) {
    return makeErrorResult(MongoDbError.UNKNOWN_ERROR);
  }
 
  return makeSuccessResult();
};
 
export const assertUserUpdatedCorrectly = async (
  usernames: string[],
): Promise<void> => {
  const users = unsafelyUnwrap(await findUsers(usernames));
 
  for (const user of users) {
    const newAssignmentEventLogItems = user.eventLogItems.filter(
      (eventLogItem) => eventLogItem.action === EventLogAction.NEW_ASSIGNMENT,
    );
    expect(newAssignmentEventLogItems).toHaveLength(1);
  }
 
  const verifyResult = await verifyUserSignups();
  expect(verifyResult.ok).toBe(true);
};
 
export const generateTestData = async (
  newUsersCount: number,
  newProgramItemsCount: number,
  groupSize: number,
  numberOfGroups: number,
  testUsersCount: number,
): Promise<void> => {
  await generateTestUsers(
    newUsersCount,
    groupSize,
    numberOfGroups,
    testUsersCount,
  );
 
  await createProgramItems(newProgramItemsCount);
 
  await createLotterySignups();
};