@getjones/test-infra 3.0.3 → 3.1.3

package/src/factory/records/records.js

@@ -1,169 +1,225 @@
-const path = require('path');
-
-const { getEnv } = require('../../env');
-const { credentials } = getEnv();
-
-const auth = require('../../saas/internal-api/auth');
-const coi = require('../../saas/internal-api/coi');
-const compliance = require('../../saas/internal-api/compliance');
-const contractors = require('../../saas/internal-api/contractors');
-const requests = require('../../saas/internal-api/requests');
-
-const { complianceStatuses, getAuditConfigByStatus } = require('../../common/utils/complianceUtil');
-const httpResponseStatusCodes = require('../../common/httpResponseStatusCodes.json');
-const activationStatuses = require('../../common/consts/activationStatuses');
-const timeUtils = require('../../common/utils/timeUtils');
-const recordFlows = require('../../common/commonFlows/records/records');
-
-const filePath = path.join(__dirname, '../../resources/coi/coi.pdf');
-
-const prepareRecord = async (recordCompany, recordEmail, status, tabId, requirementId, context, { shouldSendNotificationToPM = false, recordConfig } = {}) => {
-	const requestData = await requests.services.getOrCreateRequest(
-		context,
-		tabId,
-		requirementId,
-		{ // record
-			inviterEmail: credentials.superUser,
-			email: recordEmail,
-			companyName: recordCompany
-		}
-	);
-
-	if (!(requestData && requestData.id)) {
-		throw Error(`Failed to create record ${recordCompany}\n${requestData}`);
-	}
-
-	const accountId = requestData.record.userId;
-	const projectId = requestData.projectInfo.projectId;
-	const recordId = requestData.record.contractorId;
-
-	recordConfig && await recordFlows.editRequestAndRecordDetails(requestData, recordConfig, context);
-
-	const coiUpload = await coi.utils.uploadCoi(
-		recordId,
-		filePath,
-		uploaderType = coi.consts.UPLOADER_TYPES.ICS,
-		uploaderEmail = credentials.superUser
-	);
-
-	if (!(coiUpload && coiUpload.name)) {
-		throw Error(`Failed to upload COI for record ${recordCompany}\n${coiUpload}`);
-	}
-
-	if (complianceStatuses.underReview === status) {
-		return recordId;
-	}
-
-	const auditorContext = await auth.services.getAuditorCookie();
-
-	const auditConfig = getAuditConfigByStatus(status);
-
-	let auditResponse, retries = 5;
-	do {
-		await timeUtils.waitFor(1).seconds;
-		auditResponse = await compliance.services.audit.auditRecord(
-			auditorContext,
-			accountId,
-			projectId,
-			requirementId,
-			recordId,
-			auditConfig,
-			{ shouldSendNotificationToPM }
-		);
-	} while (auditResponse.status !== httpResponseStatusCodes.NO_CONTENT && retries--);
-
-	if (auditResponse.status !== httpResponseStatusCodes.NO_CONTENT) {
-		throw Error(`Failed to audit record ${recordCompany}\n${auditResponse}`);
-	}
-
-	return recordId;
-};
-
-const prepareCompliantRecord = (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
-	return prepareRecord(recordCompany, recordEmail, complianceStatuses.complaint, tabId, requirementId, context, { recordConfig });
-};
-
-const prepareNotCompliantRecord = (recordCompany, recordEmail, tabId, requirementId, context, { shouldSendNotificationToPM = false, recordConfig } = {}) => {
-	return prepareRecord(recordCompany, recordEmail, complianceStatuses.notCompliant, tabId, requirementId, context, {	shouldSendNotificationToPM, recordConfig });
-};
-
-const prepareExpiredRecord = (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
-	return prepareRecord(recordCompany, recordEmail, complianceStatuses.expired, tabId, requirementId, context, { recordConfig });
-};
-
-const prepareAboutToExpiredRecord = (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
-	return prepareRecord(recordCompany, recordEmail, complianceStatuses.aboutToExpired, tabId, requirementId, context, { recordConfig });
-};
-
-const prepareWaivedRecord = async (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
-	const recordId = await prepareRecord(recordCompany, recordEmail, complianceStatuses.notCompliant, tabId, requirementId, context, { recordConfig });
-
-	const { data: record } = await contractors.services.getContractorById(recordId, context);
-	await compliance.services.overrideGap.waiveAll(record, context);
-
-	return recordId;
-};
-
-const prepareAwaitingNewCoiRecord = async (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
-	const recordId = await prepareRecord(recordCompany, recordEmail, complianceStatuses.awaitingNewCoi, tabId, requirementId, context, { recordConfig });
-
-	const { id: requestId } = await requests.services.getRequestFromTabByRecordId(recordId, tabId, context);
-	const recordVerificationModel = await contractors.services.getRecordVerificationModel(recordId, context);
-	const sendGapsResponse = await contractors.services.sendGaps(requestId, recordVerificationModel, context);
-
-	if (sendGapsResponse.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to send gaps to ${recordCompany}\n${sendGapsResponse}`);
-	}
-
-	return recordId;
-};
-const prepareUnderReviewRecord = (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
-	return prepareRecord(recordCompany, recordEmail, complianceStatuses.underReview, tabId, requirementId, context, { recordConfig });
-};
-const prepareDraftRecord = async (recordCompany, tabId, context, { recordEmail, requirementId, uploadCoi, recordConfig } = {}) => {
-	const requestData = await requests.services.getOrCreateRequest(
-		context,
-		tabId,
-		requirementId,
-		{
-			inviterEmail: credentials.superUser,
-			email: recordEmail,
-			companyName: recordCompany
-		},
-		activationStatuses.DRAFT
-	);
-
-	if (!(requestData && requestData.id)) {
-		throw Error(`Failed to create record ${recordCompany}\n${requestData}`);
-	}
-
-	recordConfig && await recordFlows.editRequestAndRecordDetails(requestData, recordConfig, context);
-
-	if (uploadCoi) {
-		const recordId = requestData.record.contractorId;
-
-		const coiUpload = await coi.utils.uploadCoi(
-			recordId,
-			filePath,
-			uploaderType = coi.consts.UPLOADER_TYPES.ICS,
-			uploaderEmail = credentials.superUser
-		);
-
-		if (!(coiUpload && coiUpload.name)) {
-			throw Error(`Failed to upload COI for record ${recordCompany}\n${coiUpload}`);
-		}
-	}
-
-	return requestData;
-};
-
-module.exports = {
-	prepareCompliantRecord,
-	prepareNotCompliantRecord,
-	prepareExpiredRecord,
-	prepareWaivedRecord,
-	prepareAboutToExpiredRecord,
-	prepareAwaitingNewCoiRecord,
-	prepareUnderReviewRecord,
-	prepareDraftRecord
-};
+const path = require('path');
+
+const { getEnv } = require('../../env');
+const { credentials } = getEnv();
+
+const auth = require('../../saas/internal-api/auth');
+const coi = require('../../saas/internal-api/coi');
+const compliance = require('../../saas/internal-api/compliance');
+const contractors = require('../../saas/internal-api/contractors');
+const requests = require('../../saas/internal-api/requests');
+
+const { complianceStatuses, getAuditConfigByStatus, expiryStatuses } = require('../../common/utils/complianceUtil');
+const httpResponseStatusCodes = require('../../common/httpResponseStatusCodes.json');
+const activationStatuses = require('../../common/consts/activationStatuses');
+const timeUtils = require('../../common/utils/timeUtils');
+const recordFlows = require('../../common/commonFlows/records/records');
+const recordTypes = require('../../common/consts/recordTypes');
+
+const filePath = path.join(__dirname, '../../resources/coi/coi.pdf');
+
+const prepareRecord = async (recordCompany, recordEmail, status, tabId, requirementId, context, { expiryStatus, shouldSendNotificationToPM = false, recordConfig } = {}) => {
+	validateInsuranceAndExpiryStatus(status, expiryStatus);
+
+	const requestData = await requests.services.getOrCreateRequest(
+		context,
+		tabId,
+		requirementId,
+		{ // record
+			inviterEmail: credentials.superUser,
+			email: recordEmail,
+			companyName: recordCompany
+		}
+	);
+
+	if (!(requestData && requestData.id)) {
+		throw Error(`Failed to create record ${recordCompany}\n${requestData}`);
+	}
+
+	const accountId = requestData.record.userId;
+	const projectId = requestData.projectInfo.projectId;
+	const recordId = requestData.record.contractorId;
+
+	recordConfig && await recordFlows.editRequestAndRecordDetails(requestData, recordConfig, context);
+
+	if (status === complianceStatuses.awaitingCoi) {
+		return recordId;
+	}
+
+	const coiUpload = await coi.utils.uploadCoi(
+		recordId,
+		filePath,
+		uploaderType = coi.consts.UPLOADER_TYPES.ICS,
+		uploaderEmail = credentials.superUser
+	);
+
+	if (!(coiUpload && coiUpload.name)) {
+		throw Error(`Failed to upload COI for record ${recordCompany}\n${coiUpload}`);
+	}
+
+	if (complianceStatuses.underReview === status) {
+		return recordId;
+	}
+
+	const auditorContext = await auth.services.getAuditorCookie();
+
+	const auditConfig = getAuditConfigByStatus(status, { expiryStatus });
+
+	let auditResponse, retries = 5;
+	do {
+		await timeUtils.waitFor(1).seconds;
+		auditResponse = await compliance.services.audit.auditRecord(
+			auditorContext,
+			accountId,
+			projectId,
+			requirementId,
+			recordId,
+			auditConfig,
+			{ shouldSendNotificationToPM }
+		);
+	} while (auditResponse.status !== httpResponseStatusCodes.NO_CONTENT && retries--);
+
+	if (auditResponse.status !== httpResponseStatusCodes.NO_CONTENT) {
+		throw Error(`Failed to audit record ${recordCompany}\n${auditResponse}`);
+	}
+
+	if (status === complianceStatuses.waived) {
+		await waiveAllGaps(recordId, context);
+	} else if (status === complianceStatuses.awaitingNewCoi) {
+		await sendGaps(recordId, tabId, context);
+	}
+
+	return recordId;
+};
+
+const prepareCompliantRecord = (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig, expiryStatus } = {}) => {
+	return prepareRecord(recordCompany, recordEmail, complianceStatuses.complaint, tabId, requirementId, context, { recordConfig, expiryStatus });
+};
+
+const prepareNotCompliantRecord = (recordCompany, recordEmail, tabId, requirementId, context, { shouldSendNotificationToPM = false, recordConfig, expiryStatus } = {}) => {
+	return prepareRecord(recordCompany, recordEmail, complianceStatuses.notCompliant, tabId, requirementId, context, {	shouldSendNotificationToPM, recordConfig, expiryStatus });
+};
+
+const prepareWaivedRecord = async (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig, expiryStatus } = {}) => {
+	const recordId = await prepareRecord(recordCompany, recordEmail, complianceStatuses.notCompliant, tabId, requirementId, context, { recordConfig, expiryStatus });
+	await waiveAllGaps(recordId, context);
+	return recordId;
+};
+
+const prepareAwaitingNewCoiRecord = async (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig, expiryStatus, isNotResponsive = false } = {}) => {
+	const recordId = await prepareRecord(recordCompany, recordEmail, complianceStatuses.awaitingNewCoi, tabId, requirementId, context, { recordConfig, expiryStatus });
+
+	if (isNotResponsive) {
+		await makeRecordNotResponsive(recordId, context);
+	}
+
+	return recordId;
+};
+
+const prepareAwaitingCoiRecord = async (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig, isNotResponsive = false } = {}) => {
+	const recordId = await prepareRecord(recordCompany, recordEmail, complianceStatuses.awaitingCoi, tabId, requirementId, context, { recordConfig });
+
+	if (isNotResponsive) {
+		await makeRecordNotResponsive(recordId, context);
+	}
+
+	return recordId;
+};
+
+const prepareUnderReviewRecord = (recordCompany, recordEmail, tabId, requirementId, context, { recordConfig } = {}) => {
+	return prepareRecord(recordCompany, recordEmail, complianceStatuses.underReview, tabId, requirementId, context, { recordConfig });
+};
+
+const makeRecordNotResponsive = async (recordId, context) => {
+	const { data: record } = await contractors.services.getContractorById(recordId, context);
+	const isTenant = record.recordType === recordTypes.TENANTS;
+	await recordFlows.makeRecordNotResponsive(recordId, context, { isTenant });
+};
+
+const sendGaps = async (recordId, tabId, context) => {
+	const { id: requestId } = await requests.services.getRequestFromTabByRecordId(recordId, tabId, context);
+	const recordVerificationModel = await contractors.services.getRecordVerificationModel(recordId, context);
+	const sendGapsResponse = await contractors.services.sendGaps(requestId, recordVerificationModel, context);
+
+	if (sendGapsResponse.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to send gaps to ${recordCompany}\n${sendGapsResponse}`);
+	}
+};
+
+const waiveAllGaps = async (recordId, context) => {
+	const { data: record } = await contractors.services.getContractorById(recordId, context);
+	await compliance.services.overrideGap.waiveAll(record, context);
+};
+
+const validateInsuranceAndExpiryStatus = async (insuranceStatus, expiryStatus) => {
+	let isValid = false;
+	const isWithoutExpiryStatus = Boolean(expiryStatus);
+
+	switch (insuranceStatus) {
+		case complianceStatuses.complaint:
+		case complianceStatuses.waived:
+			isValid = isWithoutExpiryStatus || expiryStatus === expiryStatuses.aboutToExpired ;
+			break;
+		case complianceStatuses.awaitingNewCoi:
+		case complianceStatuses.notCompliant:
+			isValid = isWithoutExpiryStatus || Object.values(expiryStatuses).includes(expiryStatus) ;
+			break;
+		case complianceStatuses.awaitingCoi:
+		case complianceStatuses.underReview:
+			isValid = true;
+			break;
+		default:
+			throw new Error(`Invalid insurance status: ${insuranceStatus}`);
+	}
+
+	if (!isValid) {
+		throw new Error(`Invalid insurance and expiry status combination: insuranceStatus=${insuranceStatus}, expiryStatus=${expiryStatus}`);
+	}
+};
+
+const prepareDraftRecord = async (recordCompany, tabId, context, { recordEmail, requirementId, uploadCoi, recordConfig } = {}) => {
+	const requestData = await requests.services.getOrCreateRequest(
+		context,
+		tabId,
+		requirementId,
+		{
+			inviterEmail: credentials.superUser,
+			email: recordEmail,
+			companyName: recordCompany
+		},
+		activationStatuses.DRAFT
+	);
+
+	if (!(requestData && requestData.id)) {
+		throw Error(`Failed to create record ${recordCompany}\n${requestData}`);
+	}
+
+	recordConfig && await recordFlows.editRequestAndRecordDetails(requestData, recordConfig, context);
+
+	if (uploadCoi) {
+		const recordId = requestData.record.contractorId;
+
+		const coiUpload = await coi.utils.uploadCoi(
+			recordId,
+			filePath,
+			uploaderType = coi.consts.UPLOADER_TYPES.ICS,
+			uploaderEmail = credentials.superUser
+		);
+
+		if (!(coiUpload && coiUpload.name)) {
+			throw Error(`Failed to upload COI for record ${recordCompany}\n${coiUpload}`);
+		}
+	}
+
+	return requestData;
+};
+
+module.exports = {
+	prepareCompliantRecord,
+	prepareNotCompliantRecord,
+	prepareWaivedRecord,
+	prepareAwaitingNewCoiRecord,
+	prepareUnderReviewRecord,
+	prepareAwaitingCoiRecord,
+	prepareDraftRecord
+};

package/src/factory/users/users.js

@@ -1,77 +1,77 @@
-const { getEnv } = require('../../env');
-const httpResponseStatusCodes = require('../../common/httpResponseStatusCodes.json');
-
-const { credentials } = getEnv();
-
-const users = require('../../saas/internal-api/users');
-const { InviteUserModel } = users.models;
-
-const defaultsPassword = credentials.strongPassword;
-const roles = {
-	readOnly: 'Read-only User',
-	inviter: 'Inviter',
-	cm: 'Compliance Manager'
-};
-
-const inviteSaasUser = async (username, role, projectIds, context) => {
-	const userResponse = await users.services.getUserByEmail(username, context);
-
-	if (userResponse.data && userResponse.data.subuserId) {
-		return;
-	}
-
-	const inviteUserModel = new InviteUserModel(username, role, projectIds, context);
-	const inviteResponse = await users.services.inviteUser([inviteUserModel], context);
-
-	if (inviteResponse.status !== httpResponseStatusCodes.NO_CONTENT) {
-		throw new Error(`Failed to invite user with email: ${username}\n${inviteResponse}`);
-	}
-};
-
-const signupSaasUser = async (username, role, context) => {
-	const userResponse = await users.services.getUserByEmail(username, context);
-
-	//Already signup user
-	if (userResponse.data.isVerified) {
-		return;
-	}
-
-	const signupUserModel = await users.util.getSignupUserModel(username, role, role, defaultsPassword, context);
-	const signupResponse = await users.services.signupUser(signupUserModel, context);
-
-	if (signupResponse.status !== httpResponseStatusCodes.OK) {
-		throw new Error(`Failed to invite ${role} user with email: ${username}\n${signupResponse}`);
-	}
-};
-
-const prepareUser = async (username, role, projectIds, context, { signup = true } = {}) => {
-	await inviteSaasUser(username, role, projectIds, context);
-
-	signup && await signupSaasUser(username, role, context);
-
-	const userResponse = await users.services.getUserByEmail(username, context);
-
-	if (userResponse.status !== httpResponseStatusCodes.OK) {
-		throw new Error(`Failed to prepare ${role} user with email: ${username}\n${userResponse}`);
-	}
-
-	return userResponse.data;
-};
-
-const prepareComplianceManagerUser = (username, projectIds, context, { signup } = {}) => {
-	return prepareUser(username, roles.cm, projectIds, context, { signup });
-};
-
-const prepareInviterUser = (username, projectIds, context, { signup } = {}) => {
-	return prepareUser(username, roles.inviter, projectIds, context, { signup });
-};
-
-const prepareReadOnlyUser = (username, projectIds, context, { signup } = {}) => {
-	return prepareUser(username, roles.readOnly, projectIds, context, { signup });
-};
-
-module.exports = {
-	prepareComplianceManagerUser,
-	prepareInviterUser,
-	prepareReadOnlyUser
-};
+const { getEnv } = require('../../env');
+const httpResponseStatusCodes = require('../../common/httpResponseStatusCodes.json');
+
+const { credentials } = getEnv();
+
+const users = require('../../saas/internal-api/users');
+const { InviteUserModel } = users.models;
+
+const defaultsPassword = credentials.strongPassword;
+const roles = {
+	readOnly: 'Read-only User',
+	inviter: 'Inviter',
+	cm: 'Compliance Manager'
+};
+
+const inviteSaasUser = async (username, role, projectIds, context) => {
+	const userResponse = await users.services.getUserByEmail(username, context);
+
+	if (userResponse.data && userResponse.data.subuserId) {
+		return;
+	}
+
+	const inviteUserModel = new InviteUserModel(username, role, projectIds, context);
+	const inviteResponse = await users.services.inviteUser([inviteUserModel], context);
+
+	if (inviteResponse.status !== httpResponseStatusCodes.NO_CONTENT) {
+		throw new Error(`Failed to invite user with email: ${username}\n${inviteResponse}`);
+	}
+};
+
+const signupSaasUser = async (username, role, context) => {
+	const userResponse = await users.services.getUserByEmail(username, context);
+
+	//Already signup user
+	if (userResponse.data.isVerified) {
+		return;
+	}
+
+	const signupUserModel = await users.util.getSignupUserModel(username, role, role, defaultsPassword, context);
+	const signupResponse = await users.services.signupUser(signupUserModel, context);
+
+	if (signupResponse.status !== httpResponseStatusCodes.OK) {
+		throw new Error(`Failed to invite ${role} user with email: ${username}\n${signupResponse}`);
+	}
+};
+
+const prepareUser = async (username, role, projectIds, context, { signup = true } = {}) => {
+	await inviteSaasUser(username, role, projectIds, context);
+
+	signup && await signupSaasUser(username, role, context);
+
+	const userResponse = await users.services.getUserByEmail(username, context);
+
+	if (userResponse.status !== httpResponseStatusCodes.OK) {
+		throw new Error(`Failed to prepare ${role} user with email: ${username}\n${userResponse}`);
+	}
+
+	return userResponse.data;
+};
+
+const prepareComplianceManagerUser = (username, projectIds, context, { signup } = {}) => {
+	return prepareUser(username, roles.cm, projectIds, context, { signup });
+};
+
+const prepareInviterUser = (username, projectIds, context, { signup } = {}) => {
+	return prepareUser(username, roles.inviter, projectIds, context, { signup });
+};
+
+const prepareReadOnlyUser = (username, projectIds, context, { signup } = {}) => {
+	return prepareUser(username, roles.readOnly, projectIds, context, { signup });
+};
+
+module.exports = {
+	prepareComplianceManagerUser,
+	prepareInviterUser,
+	prepareReadOnlyUser
+};