Prepare an object to be snapshoted by jest

prepareObjectSnapshot.js

/**
 * Prepare an object to be snapshoted by jest
 * replace objectID
 * replace datetime
 * frozen same specific keys
 * @param obj
 * @returns {{}}
 */
export function prepareObject(obj, frozenKeys = []) {
  const placeholder = {};
  let counter = 0;

  const objectIdToNumber = {};

  Object.keys(obj).map((key) => {
    const value = obj[key];

    if (frozenKeys.indexOf(key) > -1) {
      placeholder[key] = `${key}-FROZED`;
      return;
    }

    // Handle objectID
    if (value != 0 && ObjectId.isValid(""+value)) {
      if (value in objectIdToNumber) {
      } else {
        objectIdToNumber[value] = counter;
        counter++;
      }

      const internalId = objectIdToNumber[value];

      placeholder[key] = `OBJECTID_${internalId}`;
      return;
    }

    // Handle Date
    if (typeof value !== 'number' && isValidDate(value)) {
      placeholder[key] = 'FAKE-DA-TET00:00:00.000Z';
      return;
    }

    // Handle array
    if (Array.isArray(value)) {
      placeholder[key] = value.map((item) => prepareObject(item));
      return;
    }

    if (typeof value === 'object') {
      placeholder[key] = prepareObject(value);
      return;
    }

    // if (value && typeof value !== 'undefined') {
    //   console.log('valid: ', value);
    //
    //   if (value._id) {
    //     placeholder[key] = prepareMongooseObject(value);
    //     return;
    //   }
    // }

    placeholder[key] = value;
  });

  return placeholder;
}

/**
 * Extract object ids from mongoose objects to make it possible to use with snapshot feature
 * @param obj
 * @returns {{}}
 */
export function prepareMongooseObject(obj) {
  return prepareObject(obj.toJSON());
}