@getjones/test-infra 5.2.10 → 5.2.12

package/src/saas/internal-api/requirements/api/lease/lease.service.js

@@ -1,123 +1,167 @@
-const leaseApi = require('./lease.api');
-const httpResponseStatusCodes = require('../../../../../common/httpResponseStatusCodes.json');
-const timeUtils = require('../../../../../common/utils/timeUtils');
-const { leaseExtractionStatus } = require('../../../../../saas/internal-api/requirements/consts/consts');
-
-const getLeaseByRecordId = async (recordId, context) => {
-	const response = await leaseApi.getLeaseByRecordId(recordId, context);
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to get lease for record: ${recordId}`);
-	}
-
-	return response.data;
-};
-
-const getLeaseQueueData = async (leaseId, context) => {
-	const response = await leaseApi.getLeaseQueueData(leaseId, context);
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to get lease queue status for lease: ${leaseId}`);
-	}
-
-	return response.data;
-};
-
-const getLeaseExtraction = async (leaseId, context) => {
-	const response = await leaseApi.getLeaseExtraction(leaseId, context);
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to get lease extraction data for lease: ${leaseId}`);
-	}
-
-	return response.data;
-};
-
-const getLeaseExtractionAnalysisStatus = async (leaseId, context, { isExpectedSuccessStatus, retries = 27 } = {}) => {
-	let statusResponse;
-
-	while (retries > 0) {
-		await timeUtils.waitFor(5).seconds;
-		statusResponse = await leaseApi.getLeaseExtractionAnalysisStatus(leaseId, context);
-
-		if (statusResponse.status !== httpResponseStatusCodes.OK) {
-			throw Error(`Failed to get lease AI success status for lease: ${leaseId}. Status code: ${statusResponse.status}, message: ${statusResponse.data.message}`);
-		}
-
-		if (!isExpectedSuccessStatus) {
-			return statusResponse.data;
-		}
-
-		if (statusResponse.data === leaseExtractionStatus.success) {
-			return statusResponse.data;
-		}
-
-		// If expecting success but didn't get it yet - retry
-		retries--;
-		console.log(`Lease AI extraction status did not reach SUCCESS for lease: ${leaseId} (retries left: ${retries})`);
-	}
-	return statusResponse.data;
-};
-
-const requestLeaseExtraction = async (recordId, extractionRequestModel, context, { expectError } = {}) => {
-	const response = await leaseApi.requestLeaseExtraction(recordId, extractionRequestModel, context);
-
-	if (expectError) {
-		return response;
-	}
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to request requirements extraction for record: ${recordId}.`);
-	}
-
-	return response.data;
-};
-
-const getRequestCancellationData = async (leaseId, context, { expectError } = {}) => {
-	const response = await leaseApi.getRequestCancellationData(leaseId, context);
-
-	if (expectError) {
-		return response;
-	}
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to get cancellation data for lease: ${leaseId}.`);
-	}
-
-	return response.data;
-};
-
-const cancelExtractionRequest = async (leaseId, context, { expectError } = {}) => {
-	const response = await leaseApi.cancelExtractionRequest(leaseId, context);
-
-	if (expectError) {
-		return response;
-	}
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to cancel extraction request for lease: ${leaseId}.`);
-	}
-
-	return response.data;
-};
-
-const createRequirementsByLease = async (leaseId, requirementId, context) => {
-	const response = await leaseApi.createRequirementsByLease(leaseId, requirementId, context);
-
-	if (response.status !== httpResponseStatusCodes.OK) {
-		throw Error(`Failed to create requirements for lease: ${leaseId}`);
-	}
-
-	return response.data;
-};
-
-module.exports = {
-	getLeaseByRecordId,
-	getLeaseQueueData,
-	getLeaseExtraction,
-	getLeaseExtractionAnalysisStatus,
-	createRequirementsByLease,
-	requestLeaseExtraction,
-	getRequestCancellationData,
-	cancelExtractionRequest
-};
+const leaseApi = require('./lease.api');
+const httpResponseStatusCodes = require('../../../../../common/httpResponseStatusCodes.json');
+const timeUtils = require('../../../../../common/utils/timeUtils');
+const { leaseExtractionStatus } = require('../../../../../saas/internal-api/requirements/consts/consts');
+const OPEN_QUEUE_STATUSES = [leaseExtractionStatus.pending, leaseExtractionStatus.inProgress, leaseExtractionStatus.completed];
+
+const getLeaseByRecordId = async (recordId, context) => {
+	const response = await leaseApi.getLeaseByRecordId(recordId, context);
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to get lease for record: ${recordId}`);
+	}
+
+	return response.data;
+};
+
+const getLeaseQueueData = async (leaseId, context) => {
+	const response = await leaseApi.getLeaseQueueData(leaseId, context);
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to get lease queue status for lease: ${leaseId}`);
+	}
+
+	return response.data;
+};
+
+const getLeaseExtraction = async (leaseId, context) => {
+	const response = await leaseApi.getLeaseExtraction(leaseId, context);
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to get lease extraction data for lease: ${leaseId}`);
+	}
+
+	return response.data;
+};
+
+const getLeaseExtractionAnalysisStatus = async (leaseId, context, { isExpectedSuccessStatus, retries = 27 } = {}) => {
+	let statusResponse;
+
+	while (retries > 0) {
+		await timeUtils.waitFor(5).seconds;
+		statusResponse = await leaseApi.getLeaseExtractionAnalysisStatus(leaseId, context);
+
+		if (statusResponse.status !== httpResponseStatusCodes.OK) {
+			throw Error(`Failed to get lease AI success status for lease: ${leaseId}. Status code: ${statusResponse.status}, message: ${statusResponse.data.message}`);
+		}
+
+		if (!isExpectedSuccessStatus) {
+			return statusResponse.data;
+		}
+
+		if (statusResponse.data === leaseExtractionStatus.success) {
+			return statusResponse.data;
+		}
+
+		// If expecting success but didn't get it yet - retry
+		retries--;
+		console.log(`Lease AI extraction status did not reach SUCCESS for lease: ${leaseId} (retries left: ${retries})`);
+	}
+	return statusResponse.data;
+};
+
+const requestLeaseExtraction = async (recordId, extractionRequestModel, context, { expectError } = {}) => {
+	const response = await leaseApi.requestLeaseExtraction(recordId, extractionRequestModel, context);
+
+	if (expectError) {
+		return response;
+	}
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to request requirements extraction for record: ${recordId}.`);
+	}
+
+	return response.data;
+};
+
+const getRequestCancellationData = async (leaseId, context, { expectError } = {}) => {
+	const response = await leaseApi.getRequestCancellationData(leaseId, context);
+
+	if (expectError) {
+		return response;
+	}
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to get cancellation data for lease: ${leaseId}.`);
+	}
+
+	return response.data;
+};
+
+const cancelExtractionRequest = async (leaseId, context, { expectError } = {}) => {
+	const response = await leaseApi.cancelExtractionRequest(leaseId, context);
+
+	if (expectError) {
+		return response;
+	}
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to cancel extraction request for lease: ${leaseId}.`);
+	}
+
+	return response.data;
+};
+
+const requestTabLeaseExtraction = async (tabId, tabExtractionRequestModel, context, { expectError } = {}) => {
+	const response = await leaseApi.requestTabLeaseExtraction(tabId, tabExtractionRequestModel, context);
+
+	if (expectError) {
+		return response;
+	}
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to request tab requirements extraction for tab: ${tabId}. Status code: ${response.status}`);
+	}
+
+	return response.data;
+};
+
+const getLeasesByTabId = async (tabId, context, { openOnly = true, expectError } = {}) => {
+	const response = await leaseApi.getLeasesByTabId(tabId, context);
+
+	if (expectError) {
+		return response;
+	}
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to get leases for tab: ${tabId}. Status code: ${response.status}`);
+	}
+
+	return openOnly ?
+		response.data.filter(({ queueStatus }) => OPEN_QUEUE_STATUSES.includes(queueStatus)) :
+		response.data;
+};
+
+const getTabDocumentView = async (documentId, context, { expectError } = {}) => {
+	const response = await leaseApi.getTabDocumentView(documentId, context);
+
+	if (!expectError && response.status !== httpResponseStatusCodes.FOUND) {
+		throw Error(`Failed to get view url for tab document: ${documentId}. Status code: ${response.status}`);
+	}
+
+	return response;
+};
+
+const createRequirementsByLease = async (leaseId, requirementId, context) => {
+	const response = await leaseApi.createRequirementsByLease(leaseId, requirementId, context);
+
+	if (response.status !== httpResponseStatusCodes.OK) {
+		throw Error(`Failed to create requirements for lease: ${leaseId}`);
+	}
+
+	return response.data;
+};
+
+module.exports = {
+	getLeaseByRecordId,
+	getLeaseQueueData,
+	getLeaseExtraction,
+	getLeaseExtractionAnalysisStatus,
+	createRequirementsByLease,
+	requestLeaseExtraction,
+	getRequestCancellationData,
+	cancelExtractionRequest,
+	requestTabLeaseExtraction,
+	getLeasesByTabId,
+	getTabDocumentView
+};

package/src/saas/internal-api/requirements/api/requirements.api.js

@@ -1,65 +1,65 @@
-const requests = require('../../../../common/requests/cookieAuth/requests');
-const { backendUrl } = require('../../../../env').getEnv();
-
-const Endpoints = {
-	create: `${backendUrl}/saas/requirement`,
-	getRequirementsByTabId: `${backendUrl}/saas/requirements`,
-	getRequirementById: requirementId => `${backendUrl}/saas/requirement/${requirementId}`,
-	update: `${backendUrl}/saas/requirement/linked/edit`,
-	updateWithNotification: `${backendUrl}/saas/requirement/linked/edit/notify`,
-	delete: requirementId => `${backendUrl}/saas/requirement/${requirementId}`,
-	getLinkedRecordCount: requirementId => `${backendUrl}/saas/requirement/${requirementId}/contractor-count`,
-	updateRequirementsVisibility: `${backendUrl}/saas/requirements/visibility`,
-	getViewOnlyRequirement: `${backendUrl}/_saas/requirement/viewonly`,
-	getPermanentViewOnlyRequirement: `${backendUrl}/_saas/requirement/viewonly/permanent`
-};
-
-const createRequirement = (createRequirementModel, tabId, context) => {
-	const params = { tabId };
-	return requests.postRequestHandler(Endpoints.create, createRequirementModel, context, {}, params);
-};
-
-const getRequirementsByTabId = (tabId, context) => {
-	return requests.getRequestHandler(Endpoints.getRequirementsByTabId, { tabId }, context);
-};
-
-const getRequirementById = (requirementId, context, { isPermanentLink } = {}) => {
-	return requests.getRequestHandler(Endpoints.getRequirementById(requirementId), { isPermanentLink }, context);
-};
-
-const getRequirementLinkedRecordCount = async (requirementId, context, { shouldCountDraft = true } = {}) => {
-	return requests.getRequestHandler(Endpoints.getLinkedRecordCount(requirementId), { shouldCountDraft }, context);
-};
-
-const updateRequirement = (updateRequirementModel, tabId, context) => {
-	return requests.postRequestHandler(Endpoints.update, updateRequirementModel, context, {}, { tabId });
-};
-
-const updateRequirementWithNotification = (updateRequirementModel, tabId, context) => {
-	return requests.postRequestHandler(Endpoints.updateWithNotification, updateRequirementModel, context, {}, { tabId });
-};
-
-const deleteByRequirementId = (requirementId, context) => {
-	return requests.deleteRequestHandler(Endpoints.delete(requirementId), context);
-};
-
-const updateRequirementsVisibility = (data, context) => {
-	return requests.postRequestHandler(Endpoints.updateRequirementsVisibility, data, context);
-};
-
-const getViewOnlyRequirement = (encryptedUrl, context = null, { isPermanent = false } = {}) => {
-	const endpoint = isPermanent ? Endpoints.getPermanentViewOnlyRequirement : Endpoints.getViewOnlyRequirement;
-	return requests.getRequestHandler(endpoint, { encryptedUrl }, context);
-};
-
-module.exports = {
-	createRequirement,
-	getRequirementsByTabId,
-	getRequirementById,
-	getRequirementLinkedRecordCount,
-	updateRequirement,
-	updateRequirementWithNotification,
-	updateRequirementsVisibility,
-	getViewOnlyRequirement,
-	deleteByRequirementId
-};
+const requests = require('../../../../common/requests/cookieAuth/requests');
+const { backendUrl } = require('../../../../env').getEnv();
+
+const Endpoints = {
+	create: `${backendUrl}/saas/requirement`,
+	getRequirementsByTabId: `${backendUrl}/saas/requirements`,
+	getRequirementById: requirementId => `${backendUrl}/saas/requirement/${requirementId}`,
+	update: `${backendUrl}/saas/requirement/linked/edit`,
+	updateWithNotification: `${backendUrl}/saas/requirement/linked/edit/notify`,
+	delete: requirementId => `${backendUrl}/saas/requirement/${requirementId}`,
+	getLinkedRecordCount: requirementId => `${backendUrl}/saas/requirement/${requirementId}/contractor-count`,
+	updateRequirementsVisibility: `${backendUrl}/saas/requirements/visibility`,
+	getViewOnlyRequirement: `${backendUrl}/_saas/requirement/viewonly`,
+	getPermanentViewOnlyRequirement: `${backendUrl}/_saas/requirement/viewonly/permanent`
+};
+
+const createRequirement = (createRequirementModel, tabId, context) => {
+	const params = { tabId };
+	return requests.postRequestHandler(Endpoints.create, createRequirementModel, context, {}, params);
+};
+
+const getRequirementsByTabId = (tabId, context) => {
+	return requests.getRequestHandler(Endpoints.getRequirementsByTabId, { tabId }, context);
+};
+
+const getRequirementById = (requirementId, context, { isPermanentLink } = {}) => {
+	return requests.getRequestHandler(Endpoints.getRequirementById(requirementId), { isPermanentLink }, context);
+};
+
+const getRequirementLinkedRecordCount = async (requirementId, context, { shouldCountDraft = true } = {}) => {
+	return requests.getRequestHandler(Endpoints.getLinkedRecordCount(requirementId), { shouldCountDraft }, context);
+};
+
+const updateRequirement = (updateRequirementModel, tabId, context) => {
+	return requests.postRequestHandler(Endpoints.update, updateRequirementModel, context, {}, { tabId });
+};
+
+const updateRequirementWithNotification = (updateRequirementModel, tabId, context) => {
+	return requests.postRequestHandler(Endpoints.updateWithNotification, updateRequirementModel, context, {}, { tabId });
+};
+
+const deleteByRequirementId = (requirementId, context) => {
+	return requests.deleteRequestHandler(Endpoints.delete(requirementId), context);
+};
+
+const updateRequirementsVisibility = (data, context) => {
+	return requests.postRequestHandler(Endpoints.updateRequirementsVisibility, data, context);
+};
+
+const getViewOnlyRequirement = (encryptedUrl, context = null, { isPermanent = false } = {}) => {
+	const endpoint = isPermanent ? Endpoints.getPermanentViewOnlyRequirement : Endpoints.getViewOnlyRequirement;
+	return requests.getRequestHandler(endpoint, { encryptedUrl }, context);
+};
+
+module.exports = {
+	createRequirement,
+	getRequirementsByTabId,
+	getRequirementById,
+	getRequirementLinkedRecordCount,
+	updateRequirement,
+	updateRequirementWithNotification,
+	updateRequirementsVisibility,
+	getViewOnlyRequirement,
+	deleteByRequirementId
+};