emberflow 1.4.3 → 1.4.4

package/lib/tests/index-utils.test.js

@@ -57,6 +57,7 @@ const projectConfig = {
 admin.initializeApp({
     databaseURL: "https://test-project.firebaseio.com",
 });
+const txnGet = jest.fn();
 jest.mock("../utils/paths", () => {
     const originalModule = jest.requireActual("../utils/paths");
     return Object.assign(Object.assign({}, originalModule), { expandAndGroupDocPathsByEntity: jest.fn() });
@@ -415,7 +416,7 @@ describe("distribute", () => {
                     }],
             ]]);
         (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, []);
-        await indexUtils.distribute(userDocsByDstPath);
+        await indexUtils.distributeFnNonTransactional(userDocsByDstPath);
         expect(admin.firestore().doc).toHaveBeenCalledTimes(1);
         expect(admin.firestore().doc).toHaveBeenCalledWith("/users/test-user-id/documents/test-doc-id");
         expect(queueInstructionsSpy).toHaveBeenCalledTimes(1);
@@ -442,7 +443,7 @@ describe("distribute", () => {
                         dstPath: "/users/test-user-id/documents/test-doc-id",
                     }],
             ]]);
-        await indexUtils.distribute(userDocsByDstPath);
+        await indexUtils.distributeFnNonTransactional(userDocsByDstPath);
         expect(admin.firestore().doc).toHaveBeenCalledTimes(1);
         expect(admin.firestore().doc).toHaveBeenCalledWith("/users/test-user-id/documents/test-doc-id");
         expect(dbDoc.get).toHaveBeenCalledTimes(1);
@@ -483,6 +484,9 @@ describe("distributeLater", () => {
     });
 });
 describe("validateForm", () => {
+    beforeEach(() => {
+        (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, []);
+    });
     it("returns an object with empty validationResult when document is valid", async () => {
         const entity = "user";
         const document = {
@@ -593,8 +597,6 @@ describe("runBusinessLogics", () => {
     let logicFn5;
     let logicFn6;
     let dbSpy;
-    let simulateSubmitFormSpy;
-    let createMetricExecutionSpy;
     let actionRef;
     beforeEach(() => {
         distributeFn = jest.fn();
@@ -604,8 +606,6 @@ describe("runBusinessLogics", () => {
         logicFn4 = jest.fn().mockResolvedValue({ status: "finished" });
         logicFn5 = jest.fn().mockResolvedValue({ status: "finished" });
         logicFn6 = jest.fn().mockResolvedValue({ status: "finished" });
-        simulateSubmitFormSpy = jest.spyOn(indexUtils._mockable, "simulateSubmitForm").mockResolvedValue();
-        createMetricExecutionSpy = jest.spyOn(indexUtils._mockable, "createMetricExecution").mockResolvedValue();
         const dbDoc = {
             get: jest.fn().mockResolvedValue({
                 data: jest.fn().mockReturnValue({
@@ -633,8 +633,6 @@ describe("runBusinessLogics", () => {
         logicFn3.mockRestore();
         distributeFn.mockRestore();
         dbSpy.mockRestore();
-        simulateSubmitFormSpy.mockRestore();
-        createMetricExecutionSpy.mockRestore();
     });
     it("should call all matching logics and pass their results to distributeFn", async () => {
         const logics = [
@@ -691,71 +689,25 @@ describe("runBusinessLogics", () => {
             },
         ];
         (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, logics);
-        const runStatus = await indexUtils.runBusinessLogics(actionRef, action, distributeFn);
-        expect(logicFn1).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(logicFn2).toHaveBeenCalledWith(action, new Map(), undefined);
+        const runStatus = await indexUtils.runBusinessLogics(txnGet, action);
+        expect(logicFn1).toHaveBeenCalledWith(txnGet, action, new Map());
+        expect(logicFn2).toHaveBeenCalledWith(txnGet, action, new Map());
         expect(logicFn3).not.toHaveBeenCalled();
         expect(logicFn4).not.toHaveBeenCalled();
         expect(logicFn5).not.toHaveBeenCalled();
         expect(logicFn6).not.toHaveBeenCalled();
-        expect(distributeFn).toHaveBeenCalledTimes(1);
-        expect(distributeFn).toHaveBeenCalledWith(actionRef, [expect.objectContaining({
-                status: "finished",
-                execTime: expect.any(Number),
-                timeFinished: expect.any(Timestamp),
-            }),
-            expect.objectContaining({
-                status: "finished",
-                execTime: expect.any(Number),
-                timeFinished: expect.any(Timestamp),
-            })], 0);
-        expect(runStatus).toEqual("done");
-        expect(createMetricExecutionSpy).toHaveBeenCalledTimes(1);
-        expect(simulateSubmitFormSpy).toHaveBeenCalledTimes(1);
-    });
-    it("should recall logic when it returns \"partial-result\" status", async () => {
-        logicFn2.mockResolvedValueOnce({ status: "partial-result", nextPage: {} });
-        const logics = [
-            {
-                name: "Logic 1",
-                actionTypes: ["create"],
-                modifiedFields: ["field1"],
-                entities: ["user"],
-                logicFn: logicFn1,
-            },
-            {
-                name: "Logic 2",
-                actionTypes: "all",
-                modifiedFields: ["field2"],
-                entities: ["user"],
-                logicFn: logicFn2,
-            },
-        ];
-        (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, logics);
-        const runStatus = await indexUtils.runBusinessLogics(actionRef, action, distributeFn);
-        expect(logicFn1).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(logicFn2).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(distributeFn).toHaveBeenCalledTimes(2);
-        expect(distributeFn.mock.calls[0]).toEqual([actionRef,
-            [expect.objectContaining({
+        expect(runStatus).toEqual({
+            status: "done",
+            logicResults: [{
                     status: "finished",
                     execTime: expect.any(Number),
                     timeFinished: expect.any(Timestamp),
-                }),
-                expect.objectContaining({
-                    status: "partial-result",
-                    nextPage: {},
-                    execTime: expect.any(Number),
-                    timeFinished: expect.any(Timestamp),
-                })], 0]);
-        expect(logicFn2).toHaveBeenCalledWith(action, new Map(), {});
-        expect(distributeFn.mock.calls[1]).toEqual([actionRef,
-            [expect.objectContaining({
+                }, {
                     status: "finished",
                     execTime: expect.any(Number),
                     timeFinished: expect.any(Timestamp),
-                })], 1]);
-        expect(runStatus).toEqual("done");
+                }],
+        });
     });
     it("should not call any logic when no matching logics are found but distributeFn should still be " +
         "called", async () => {
@@ -783,76 +735,14 @@ describe("runBusinessLogics", () => {
             },
         ];
         (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, logics);
-        const runStatus = await indexUtils.runBusinessLogics(actionRef, action, distributeFn);
+        const runStatus = await indexUtils.runBusinessLogics(txnGet, action);
         expect(logicFn1).not.toHaveBeenCalled();
         expect(logicFn2).not.toHaveBeenCalled();
         expect(logicFn3).not.toHaveBeenCalled();
-        expect(distributeFn).toHaveBeenCalledTimes(1);
-        expect(distributeFn).toHaveBeenCalledWith(actionRef, [], 0);
-        expect(runStatus).toEqual("no-matching-logics");
-    });
-    it("should recall logic when it returns \"partial-result\" status indefinitely up to the config maxLogicResultPages", async () => {
-        logicFn2.mockResolvedValue({ status: "partial-result", nextPage: {} });
-        const logics = [
-            {
-                name: "Logic 1",
-                actionTypes: ["create"],
-                modifiedFields: ["field1"],
-                entities: ["user"],
-                logicFn: logicFn1,
-            },
-            {
-                name: "Logic 2",
-                actionTypes: "all",
-                modifiedFields: ["field2"],
-                entities: ["user"],
-                logicFn: logicFn2,
-            },
-        ];
-        (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, logics);
-        const runStatus = await indexUtils.runBusinessLogics(actionRef, action, distributeFn);
-        expect(logicFn1).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(logicFn2).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(logicFn1).toHaveBeenCalledTimes(1);
-        expect(logicFn2).toHaveBeenCalledTimes(10);
-        expect(distributeFn).toHaveBeenCalledTimes(10);
-        expect(runStatus).toEqual("done");
-    });
-    it("should return when a logic returns a \"cancel-then-retry\" status", async () => {
-        logicFn2.mockResolvedValue({ status: "cancel-then-retry" });
-        logicFn3.mockResolvedValue({ status: "finished" });
-        const logics = [
-            {
-                name: "Logic 1",
-                actionTypes: ["create"],
-                modifiedFields: ["field1"],
-                entities: ["user"],
-                logicFn: logicFn1,
-            },
-            {
-                name: "Logic 2",
-                actionTypes: "all",
-                modifiedFields: ["field2"],
-                entities: ["user"],
-                logicFn: logicFn2,
-            },
-            {
-                name: "Logic 3",
-                actionTypes: "all",
-                modifiedFields: ["field1"],
-                entities: ["user"],
-                logicFn: logicFn3,
-            },
-        ];
-        (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, logics);
-        const runStatus = await indexUtils.runBusinessLogics(actionRef, action, distributeFn);
-        expect(logicFn1).toHaveBeenCalledTimes(1);
-        expect(logicFn1).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(logicFn2).toHaveBeenCalledTimes(1);
-        expect(logicFn2).toHaveBeenCalledWith(action, new Map(), undefined);
-        expect(logicFn3).not.toHaveBeenCalled();
-        expect(distributeFn).not.toHaveBeenCalled();
-        expect(runStatus).toEqual("cancel-then-retry");
+        expect(runStatus).toEqual({
+            status: "no-matching-logics",
+            logicResults: [],
+        });
     });
     it("should pass data from previous logic to the next logic via shared map", async () => {
         const expectedSharedMap = new Map();
@@ -864,11 +754,11 @@ describe("runBusinessLogics", () => {
             "@id": "test-document-id",
             "title": "Test Document Title",
         });
-        logicFn2.mockImplementation((action, sharedMap) => {
+        logicFn2.mockImplementation((txnGet, action, sharedMap) => {
             sharedMap.set("another-document-id", expectedSharedMap.get("another-document-id"));
             return { status: "finished" };
         });
-        logicFn3.mockImplementation((action, sharedMap) => {
+        logicFn3.mockImplementation((txnGet, action, sharedMap) => {
             sharedMap.set("test-document-id", expectedSharedMap.get("test-document-id"));
             return { status: "finished" };
         });
@@ -896,361 +786,16 @@ describe("runBusinessLogics", () => {
             },
         ];
         (0, index_1.initializeEmberFlow)(projectConfig, admin, db_structure_1.dbStructure, db_structure_1.Entity, security_1.securityConfig, validators_1.validatorConfig, logics);
-        const runStatus = await indexUtils.runBusinessLogics(actionRef, action, distributeFn);
-        expect(logicFn1).toHaveBeenCalledWith(action, expectedSharedMap, undefined);
-        expect(logicFn2).toHaveBeenCalledWith(action, expectedSharedMap, undefined);
-        expect(logicFn3).toHaveBeenCalledWith(action, expectedSharedMap, undefined);
-        expect(runStatus).toEqual("done");
-    });
-});
-describe("simulateSubmitForm", () => {
-    const eventContext = {
-        id: "test-event-id",
-        uid: "test-user-id",
-        docPath: "servers/test-doc-id",
-        docId: "test-doc-id",
-        formId: "test-form-id",
-        entity: "servers",
-    };
-    const user = {
-        "@id": "test-user-id",
-        "username": "Topic Creator",
-        "avatarUrl": "Avatar URL",
-        "firstName": "Topic",
-        "lastName": "Creator",
-    };
-    const action = {
-        actionType: "create",
-        eventContext: eventContext,
-        user: user,
-        document: {},
-        status: "new",
-        timeCreated: admin.firestore.Timestamp.now(),
-        modifiedFields: {},
-    };
-    const distributeFn = jest.fn();
-    let runBusinessLogicsSpy;
-    let dataMock;
-    let docMock;
-    let now;
-    beforeEach(() => {
-        jest.spyOn(console, "debug").mockImplementation();
-        jest.spyOn(console, "warn").mockImplementation();
-        dataMock = jest.fn().mockReturnValue({});
-        const getMock = {
-            data: dataMock,
-        };
-        docMock = {
-            set: jest.fn(),
-            get: jest.fn().mockResolvedValue(getMock),
-            collection: jest.fn(() => collectionMock),
-        };
-        const collectionMock = {
-            doc: jest.fn(() => docMock),
-        };
-        jest.spyOn(admin.firestore(), "doc").mockReturnValue(docMock);
-        jest.spyOn(admin.firestore(), "collection").mockReturnValue(collectionMock);
-        now = Timestamp.now();
-        jest.spyOn(indexUtils._mockable, "createNowTimestamp").mockReturnValue(now);
-        runBusinessLogicsSpy =
-            jest.spyOn(indexUtils, "runBusinessLogics").mockResolvedValue("done");
-    });
-    afterEach(() => {
-        jest.restoreAllMocks();
-    });
-    it("should skip when there is no logic result doc with 'simulate-submit-form' action", async () => {
-        await indexUtils._mockable.simulateSubmitForm([], action, distributeFn);
-        expect(console.debug).toHaveBeenCalledTimes(1);
-        expect(console.debug).toHaveBeenCalledWith("Simulating submit form: ", 0);
-        expect(runBusinessLogicsSpy).not.toHaveBeenCalled();
-    });
-    it("should skip when entity is undefined", async () => {
-        const logicResults = [];
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "no-entity/sample-doc-id",
-            doc: {
-                "@id": "no-entity/sample-doc-id",
-            },
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(console.warn).toHaveBeenCalledTimes(1);
-        expect(console.warn).toHaveBeenCalledWith("No matching entity found for logic no-entity/sample-doc-id. Skipping");
-        expect(runBusinessLogicsSpy).not.toHaveBeenCalled();
-    });
-    it("should skip when logic result doc is undefined", async () => {
-        const logicResults = [];
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(console.warn).toHaveBeenCalledTimes(1);
-        expect(console.warn).toHaveBeenCalledWith("LogicResultDoc.doc should not be undefined. Skipping");
-        expect(runBusinessLogicsSpy).not.toHaveBeenCalled();
-    });
-    it("should skip when @actionType is undefined", async () => {
-        const logicResults = [];
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-            doc: { "@id": "sample-server-id" },
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(console.warn).toHaveBeenCalledTimes(1);
-        expect(console.warn).toHaveBeenCalledWith("No @actionType found. Skipping");
-        expect(runBusinessLogicsSpy).not.toHaveBeenCalled();
-    });
-    it("should skip when submitFormAs data is undefined", async () => {
-        dataMock.mockReturnValueOnce(undefined);
-        const logicResults = [];
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-            doc: {
-                "@actionType": "create",
-                "@submitFormAs": "test-user-id",
-            },
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(console.warn).toHaveBeenCalledTimes(1);
-        expect(console.warn).toHaveBeenCalledWith("User test-user-id not found. Skipping");
-        expect(runBusinessLogicsSpy).not.toHaveBeenCalled();
-    });
-    it("should simulate submit form correctly when submitFormAs is defined", async () => {
-        dataMock.mockReturnValueOnce({ "@id": "test-user-id" });
-        const logicResults = [];
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-            doc: {
-                "@actionType": "create",
-                "@submitFormAs": "test-user-id",
-                "name": "sample-server-name",
-                "createdAt": now,
-            },
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        const expectedEventContext = {
-            id: action.eventContext.id + "-1",
-            uid: action.eventContext.uid,
-            formId: action.eventContext.formId + "-1",
-            docId: "sample-server-id",
-            docPath: logicResultDoc.dstPath,
-            entity: "server",
-        };
-        const expectedAction = {
-            eventContext: expectedEventContext,
-            actionType: "create",
-            document: {},
-            modifiedFields: {
-                "name": "sample-server-name",
-                "createdAt": now,
-            },
-            user: { "@id": "test-user-id" },
-            status: "new",
-            timeCreated: now,
-        };
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(docMock.set).toHaveBeenCalledTimes(1);
-        expect(docMock.set).toHaveBeenCalledWith(expectedAction);
-        expect(runBusinessLogicsSpy).toHaveBeenCalledTimes(1);
-        expect(runBusinessLogicsSpy).toHaveBeenCalledWith(docMock, expectedAction, distributeFn);
-    });
-    it("should simulate submit form correctly when submitFormAs is undefined", async () => {
-        const logicResults = [];
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-            doc: {
-                "@actionType": "create",
-                "name": "sample-server-name",
-                "createdAt": now,
-            },
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        const expectedEventContext = {
-            id: action.eventContext.id + "-1",
-            uid: action.eventContext.uid,
-            formId: action.eventContext.formId + "-1",
-            docId: "sample-server-id",
-            docPath: logicResultDoc.dstPath,
-            entity: "server",
-        };
-        const expectedAction = {
-            eventContext: expectedEventContext,
-            actionType: "create",
-            document: {},
-            modifiedFields: {
-                "name": "sample-server-name",
-                "createdAt": now,
-            },
-            user: user,
-            status: "new",
-            timeCreated: now,
-        };
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(docMock.set).toHaveBeenCalledTimes(1);
-        expect(docMock.set).toHaveBeenCalledWith(expectedAction);
-        expect(runBusinessLogicsSpy).toHaveBeenCalledTimes(1);
-        expect(runBusinessLogicsSpy).toHaveBeenCalledWith(docMock, expectedAction, distributeFn);
-    });
-    it("should simulate submit form correctly with multiple logic result docs", async () => {
-        const logicResults = [];
-        const logicResultDoc1 = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-            doc: {
-                "@actionType": "create",
-                "name": "sample-server-name",
-                "createdAt": now,
-            },
-        };
-        const logicResultDoc2 = {
-            action: "merge",
-            dstPath: "servers/merge-server-id",
-            doc: {
-                "@actionType": "update",
-                "name": "sample-server-name",
-            },
-        };
-        const logicResultDoc3 = {
-            action: "simulate-submit-form",
-            dstPath: "users/sample-user-id",
-            doc: {
-                "@actionType": "create",
-                "name": "sample-user-name",
-            },
-        };
-        const logicResult1 = {
-            name: "serverLogicResult",
-            status: "finished",
-            documents: [logicResultDoc1, logicResultDoc2],
-        };
-        const logicResult2 = {
-            name: "userLogicResult",
-            status: "finished",
-            documents: [logicResultDoc3],
-        };
-        logicResults.push(logicResult1);
-        logicResults.push(logicResult2);
-        const expectedServerEventContext = {
-            id: action.eventContext.id + "-1",
-            uid: action.eventContext.uid,
-            formId: action.eventContext.formId + "-1",
-            docId: "sample-server-id",
-            docPath: logicResultDoc1.dstPath,
-            entity: "server",
-        };
-        const expectedServerAction = {
-            eventContext: expectedServerEventContext,
-            actionType: "create",
-            document: {},
-            modifiedFields: {
-                "name": "sample-server-name",
-                "createdAt": now,
-            },
-            user: user,
-            status: "new",
-            timeCreated: now,
-        };
-        const expectedUserEventContext = {
-            id: action.eventContext.id + "-2",
-            uid: action.eventContext.uid,
-            formId: action.eventContext.formId + "-2",
-            docId: "sample-user-id",
-            docPath: logicResultDoc3.dstPath,
-            entity: "user",
-        };
-        const expectedUserAction = {
-            eventContext: expectedUserEventContext,
-            actionType: "create",
-            document: {},
-            modifiedFields: {
-                "name": "sample-user-name",
-            },
-            user: user,
-            status: "new",
-            timeCreated: now,
-        };
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(docMock.set).toHaveBeenCalledTimes(2);
-        expect(docMock.set).toHaveBeenNthCalledWith(1, expectedServerAction);
-        expect(docMock.set).toHaveBeenNthCalledWith(2, expectedUserAction);
-        expect(runBusinessLogicsSpy).toHaveBeenCalledTimes(2);
-        expect(runBusinessLogicsSpy).toHaveBeenNthCalledWith(1, docMock, expectedServerAction, distributeFn);
-        expect(runBusinessLogicsSpy).toHaveBeenNthCalledWith(2, docMock, expectedUserAction, distributeFn);
-    });
-    it("should skip when maximum retry count is reached", async () => {
-        runBusinessLogicsSpy.mockResolvedValueOnce("cancel-then-retry")
-            .mockResolvedValueOnce("cancel-then-retry")
-            .mockResolvedValueOnce("cancel-then-retry")
-            .mockResolvedValueOnce("cancel-then-retry")
-            .mockResolvedValueOnce("cancel-then-retry")
-            .mockResolvedValueOnce("cancel-then-retry");
-        const logicResults = [];
-        const doc = {
-            "@actionType": "create",
-        };
-        const logicResultDoc = {
-            action: "simulate-submit-form",
-            dstPath: "servers/sample-server-id",
-            doc: doc,
-        };
-        const logicResult = {
-            name: "sampleLogicResult",
-            status: "finished",
-            documents: [logicResultDoc],
-        };
-        logicResults.push(logicResult);
-        const dateSpy = jest.spyOn(Date, "now");
-        dateSpy.mockReturnValueOnce(now.toMillis())
-            .mockReturnValueOnce(now.toMillis() + 2000)
-            .mockReturnValueOnce(now.toMillis())
-            .mockReturnValueOnce(now.toMillis() + 4000)
-            .mockReturnValueOnce(now.toMillis())
-            .mockReturnValueOnce(now.toMillis() + 8000)
-            .mockReturnValueOnce(now.toMillis())
-            .mockReturnValueOnce(now.toMillis() + 16000)
-            .mockReturnValueOnce(now.toMillis())
-            .mockReturnValueOnce(now.toMillis() + 32000);
-        await indexUtils._mockable.simulateSubmitForm(logicResults, action, distributeFn);
-        expect(console.warn).toHaveBeenCalledTimes(1);
-        expect(console.warn).toHaveBeenCalledWith("Maximum retry count reached for logic servers/sample-server-id");
+        const runStatus = await indexUtils.runBusinessLogics(txnGet, action);
+        expect(logicFn1).toHaveBeenCalledWith(txnGet, action, expectedSharedMap);
+        expect(logicFn2).toHaveBeenCalledWith(txnGet, action, expectedSharedMap);
+        expect(logicFn3).toHaveBeenCalledWith(txnGet, action, expectedSharedMap);
+        expect(runStatus.status).toEqual("done");
+        const logicResults = runStatus.logicResults;
+        expect(logicResults.length).toEqual(3);
+        expect(logicResults[0].status).toEqual("finished");
+        expect(logicResults[1].status).toEqual("finished");
+        expect(logicResults[2].status).toEqual("finished");
     });
 });
 describe("groupDocsByTargetDocPath", () => {