@getjones/test-infra 3.0.1 → 3.0.2

package/src/factory/records/records.js

@@ -1,183 +1,169 @@
-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 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;
-
-	if (recordConfig) {
-		const tabModel = new requests.models.TabModel(requestData.tab.type);
-		const recordDetailsModel = contractors.services.getContractorDetailsModel(recordConfig);
-		const record = new contractors.models.edit.ContractorModel(
-			recordId,
-			requestData.record.recordType,
-			recordConfig.company,
-			recordConfig.emailAddress,
-			recordDetailsModel
-		);
-		const requestEditModel = new requests.models.RequestModel(requestData.id, record, tabModel, {});
-
-		const editRequestResponse = await requests.services.editRequestAndRecordDetails(requestEditModel, context);
-
-		if (editRequestResponse.status !== httpResponseStatusCodes.OK) {
-			throw Error(`Failed to edit record ${recordCompany}\n${editRequestResponse.data.message}`);
-		}
-	}
-
-	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 } = {}) => {
-	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}`);
-	}
-
-	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 } = 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
+};

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
+};