feat(msp-withdrawal -toggle): add logic to enable and disable withdrawal toggle

lib/lambda/setupIndex.test.ts

@@ -51,6 +51,9 @@ describe("handler", () => {
         deleted: { type: "boolean" },
         finalDispositionDate: { type: "date" },
         proposedDate: { type: "date" },
+        raiId: { type: "keyword" },
+        raiName: { type: "keyword" },
+        raiWithdrawnToggleDate: { type: "date" },
         spaWaiverId: { type: "keyword" },
         smartRecordType: { type: "keyword" },
         statusDate: { type: "date" },
@@ -67,7 +70,16 @@ describe("handler", () => {
         statusDate: { type: "date" },
       },
     );
-    expect(updateMappingSpy).toHaveBeenCalledTimes(2);
+    expect(updateMappingSpy).toHaveBeenCalledWith(
+      OPENSEARCH_DOMAIN,
+      `${OPENSEARCH_INDEX_NAMESPACE}changelog`,
+      {
+        raiId: { type: "keyword" },
+        raiName: { type: "keyword" },
+        raiWithdrawnToggleDate: { type: "date" },
+      },
+    );
+    expect(updateMappingSpy).toHaveBeenCalledTimes(3);
 
     expect(callback).toHaveBeenCalledWith(null, { statusCode: 200 });
   });

lib/lambda/setupIndex.ts

@@ -19,6 +19,9 @@ export const handler: Handler = async (event, __, callback) => {
         deleted: { type: "boolean" },
         finalDispositionDate: { type: "date" },
         proposedDate: { type: "date" },
+        raiId: { type: "keyword" },
+        raiName: { type: "keyword" },
+        raiWithdrawnToggleDate: { type: "date" },
         spaWaiverId: { type: "keyword" },
         smartRecordType: { type: "keyword" },
         statusDate: { type: "date" },
@@ -38,6 +41,11 @@ export const handler: Handler = async (event, __, callback) => {
     await manageIndexResource({
       osDomain: event.osDomain,
       index: `${event.indexNamespace}changelog`,
+      update: {
+        raiId: { type: "keyword" },
+        raiName: { type: "keyword" },
+        raiWithdrawnToggleDate: { type: "date" },
+      },
     });
     await manageIndexResource({
       osDomain: event.osDomain,

lib/lambda/sinkSmart.test.ts

@@ -595,12 +595,87 @@ describe("SMART operation dispatch", () => {
     );
   });
 
+  it("consumes MSP_RAI_WITHDRAWAL_TOGGLED for an existing package", async () => {
+    const packageId = "AL-26-0001-GATT";
+    const spaWaiverId = "a0ncp000006RG8TAAW";
+    const raiId = "a0qcp000005GbK5AAK";
+    const toggleDate = "2026-08-17T16:11:46.000Z";
+    const raiTogglePayload = {
+      ...smartEvent,
+      spaWaiverId,
+      id: packageId,
+      correlationId: "",
+      authority: "Medicaid SPA",
+      status: "Pending Second Clock",
+      createdAt: toggleDate,
+      operationType: "MSP_RAI_WITHDRAWAL_TOGGLED",
+      raiId,
+      raiName: `${packageId}-RAI`,
+      raiWithdrawnToggle: true,
+      raiWithdrawnToggleDate: toggleDate,
+    };
+    const existingPackage = {
+      id: packageId,
+      origin: "OneMAC",
+      authority: "Medicaid SPA",
+      state: "AL",
+      seatoolStatus: SEATOOL_STATUS.PENDING,
+      cmsStatus: "Pending",
+      stateStatus: "Under Review",
+      raiRequestedDate: "2026-07-01T12:00:00.000Z",
+      raiReceivedDate: "2026-08-01T12:00:00.000Z",
+      raiWithdrawEnabled: false,
+      spaWaiverId,
+      deleted: false,
+    };
+    getItemSpy.mockResolvedValue({
+      found: true,
+      _id: packageId,
+      _source: existingPackage,
+    } as Awaited<ReturnType<typeof os.getItem>>);
+    searchSpy.mockImplementation(async (_domain, index) =>
+      String(index).endsWith("main")
+        ? { hits: { hits: [{ _id: packageId, _source: existingPackage }] } }
+        : { hits: { hits: [] } },
+    );
+
+    await expect(
+      invokeHandler(createSmartEvent(createSmartRecord(raiTogglePayload, packageId))),
+    ).resolves.toBeUndefined();
+
+    expect(updateItemSpy).toHaveBeenCalledWith(
+      expect.anything(),
+      expect.stringMatching(/main$/),
+      packageId,
+      expect.objectContaining({
+        raiWithdrawEnabled: true,
+        raiId,
+        raiWithdrawnToggleDate: toggleDate,
+      }),
+    );
+    expect(bulkUpdateDataSpy).toHaveBeenCalledWith(
+      expect.anything(),
+      expect.stringMatching(/changelog$/),
+      [
+        expect.objectContaining({
+          packageId,
+          event: "toggle-withdraw-rai",
+          timestamp: Date.parse(toggleDate),
+          isAdminChange: true,
+          raiWithdrawEnabled: true,
+        }),
+      ],
+      { throwOnBulkError: true },
+    );
+    expect(createItemSpy).not.toHaveBeenCalled();
+    expect(publishSmartIngestErrorSpy).not.toHaveBeenCalled();
+  });
+
   it.each([
     "MSP_MANUAL_RECORD_CREATED",
     "MSP_STATUS_UPDATED",
     "MSP_ADMINISTRATIVE_FIELD_UPDATED",
     "MSP_ASSIGNMENT_UPDATED",
-    "MSP_RAI_WITHDRAWAL_TOGGLED",
     "NOT_A_REAL_TYPE",
     undefined,
   ])(
@@ -635,7 +710,6 @@ describe("SMART operation dispatch", () => {
     "MSP_STATUS_UPDATED",
     "MSP_ADMINISTRATIVE_FIELD_UPDATED",
     "MSP_ASSIGNMENT_UPDATED",
-    "MSP_RAI_WITHDRAWAL_TOGGLED",
     "NOT_A_REAL_TYPE",
     undefined,
   ])(