9970: Update Entity name for Trial Session so it can be validated for…

… migration.
flexion · Jul 7, 2023 · a9d8bfc · a9d8bfc
1 parent 843cc94
commit a9d8bfc
Show file tree

Hide file tree

Showing 7 changed files with 16 additions and 10 deletions.
diff --git a/shared/src/business/entities/trialSessions/ClosedTrialSession.ts b/shared/src/business/entities/trialSessions/ClosedTrialSession.ts
@@ -3,7 +3,7 @@ import { TrialSession } from './TrialSession';
 
 export class ClosedTrialSession extends TrialSession {
   constructor(rawSession: RawClosedTrialSession) {
-    super(rawSession, 'ClosedTrialSession');
+    super(rawSession, { entityName: 'ClosedTrialSession' });
 
     this.sessionStatus = SESSION_STATUS_TYPES.closed;
   }

diff --git a/shared/src/business/entities/trialSessions/NewTrialSession.ts b/shared/src/business/entities/trialSessions/NewTrialSession.ts
@@ -8,7 +8,7 @@ export class NewTrialSession extends TrialSession {
   public trialClerkId?: string;
 
   constructor(rawSession: RawNewTrialSession, { applicationContext }) {
-    super(rawSession, 'NewTrialSession');
+    super(rawSession, { entityName: 'NewTrialSession' });
 
     if (!applicationContext) {
       throw new TypeError('applicationContext must be defined');

diff --git a/shared/src/business/entities/trialSessions/OpenTrialSession.ts b/shared/src/business/entities/trialSessions/OpenTrialSession.ts
@@ -17,7 +17,7 @@ export class OpenTrialSession extends TrialSession {
   public thirtyDaysBeforeTrialFormatted: string; // TODO: should be a computed property
 
   constructor(rawProps) {
-    super(rawProps, 'OpenTrialSession');
+    super(rawProps, { entityName: 'OpenTrialSession' });
 
     const formattedStartDate = formatDateString(this.startDate, FORMATS.MMDDYY);
     const trialStartDateString = prepareDateFromString(

diff --git a/shared/src/business/entities/trialSessions/TrialSession.ts b/shared/src/business/entities/trialSessions/TrialSession.ts
@@ -230,7 +230,10 @@ export class TrialSession extends JoiValidationEntity {
     trialSessionId: JoiValidationConstants.UUID.required(),
   };
 
-  constructor(rawSession, entityName: string) {
+  constructor(
+    rawSession,
+    { entityName = 'TrialSession' }: { entityName: string },
+  ) {
     super(entityName);
 
     this.address1 = rawSession.address1;

diff --git a/web-api/src/applicationContext.ts b/web-api/src/applicationContext.ts
@@ -25,10 +25,13 @@ import { AwsSigv4Signer } from '@opensearch-project/opensearch/aws';
 import { Case } from '../../shared/src/business/entities/cases/Case';
 import { CaseDeadline } from '../../shared/src/business/entities/CaseDeadline';
 import { Client } from '@opensearch-project/opensearch';
+import { ClosedTrialSession } from '../../shared/src/business/entities/trialSessions/ClosedTrialSession';
 import { Correspondence } from '../../shared/src/business/entities/Correspondence';
 import { DocketEntry } from '../../shared/src/business/entities/DocketEntry';
 import { IrsPractitioner } from '../../shared/src/business/entities/IrsPractitioner';
 import { Message } from '../../shared/src/business/entities/Message';
+import { NewTrialSession } from '../../shared/src/business/entities/trialSessions/NewTrialSession';
+import { OpenTrialSession } from '../../shared/src/business/entities/trialSessions/OpenTrialSession';
 import { Practitioner } from '../../shared/src/business/entities/Practitioner';
 import { PrivatePractitioner } from '../../shared/src/business/entities/PrivatePractitioner';
 import { TrialSession } from '../../shared/src/business/entities/trialSessions/TrialSession';
@@ -167,10 +170,13 @@ let notificationServiceCache;
 const entitiesByName = {
   Case,
   CaseDeadline,
+  ClosedTrialSession,
   Correspondence,
   DocketEntry,
   IrsPractitioner,
   Message,
+  NewTrialSession,
+  OpenTrialSession,
   Practitioner,
   PrivatePractitioner,
   TrialSession,

diff --git a/...igrations/0000-validate-all-items.test.js → ...igrations/0000-validate-all-items.test.ts b/...igrations/0000-validate-all-items.test.js → ...igrations/0000-validate-all-items.test.ts
diff --git a/...das/migrations/0000-validate-all-items.js → ...das/migrations/0000-validate-all-items.ts b/...das/migrations/0000-validate-all-items.js → ...das/migrations/0000-validate-all-items.ts
@@ -1,9 +1,8 @@
-const {
-  createApplicationContext,
-} = require('../../../../../src/applicationContext');
+import { createApplicationContext } from '../../../../../src/applicationContext';
+
 const applicationContext = createApplicationContext({});
 
-const migrateItems = items => {
+export const migrateItems = items => {
   for (const item of items) {
     const entityConstructor = applicationContext.getEntityByName(
       item.entityName,
@@ -17,5 +16,3 @@ const migrateItems = items => {
   }
   return items;
 };
-
-exports.migrateItems = migrateItems;