feat(core-flows,types,promotion): register promotion campaign usage upon cart completion (#8970)

* feat(core-flows,types,promotion): register promotion campaign usage upon cart completion * Apply suggestions from code review Co-authored-by: Oli Juhl <59018053+olivermrbl@users.noreply.github.com> --------- Co-authored-by: Oli Juhl <59018053+olivermrbl@users.noreply.github.com>
2024-09-05 10:43:29 +02:00
parent 2a055b71ef
commit 5d7179b7d0
8 changed files with 421 additions and 56 deletions
@@ -0,0 +1,129 @@
+import { IPromotionModuleService } from "@medusajs/types"
+import { Modules } from "@medusajs/utils"
+import { moduleIntegrationTestRunner, SuiteOptions } from "medusa-test-utils"
+import { createCampaigns } from "../../../__fixtures__/campaigns"
+import { createDefaultPromotion } from "../../../__fixtures__/promotion"
+
+jest.setTimeout(30000)
+
+moduleIntegrationTestRunner({
+  moduleName: Modules.PROMOTION,
+  testSuite: ({
+    MikroOrmWrapper,
+    service,
+  }: SuiteOptions<IPromotionModuleService>) => {
+    describe("Promotion Service: campaign usage", () => {
+      beforeEach(async () => {
+        await createCampaigns(MikroOrmWrapper.forkManager())
+      })
+
+      describe("revertUsage", () => {
+        it("should revert usage for type spend", async () => {
+          const createdPromotion = await createDefaultPromotion(service, {})
+          const createdPromotion2 = await createDefaultPromotion(service, {
+            code: "PROMO_2",
+            campaign_id: createdPromotion.campaign?.id,
+          })
+          const createdPromotion3 = await createDefaultPromotion(service, {
+            code: "PROMO_3",
+            campaign_id: createdPromotion.campaign?.id,
+          })
+
+          await service.registerUsage([
+            { amount: 200, code: createdPromotion.code! },
+            { amount: 100, code: createdPromotion.code! },
+          ])
+
+          await service.registerUsage([
+            { amount: 100, code: createdPromotion2.code! },
+            { amount: 200, code: createdPromotion2.code! },
+          ])
+
+          await service.registerUsage([
+            { amount: 50, code: createdPromotion3.code! },
+            { amount: 250, code: createdPromotion3.code! },
+          ])
+
+          await service.revertUsage([
+            { amount: 200, code: createdPromotion.code! },
+            { amount: 100, code: createdPromotion.code! },
+          ])
+
+          await service.revertUsage([
+            { amount: 50, code: createdPromotion3.code! },
+            { amount: 250, code: createdPromotion3.code! },
+          ])
+
+          const campaign = await service.retrieveCampaign(
+            createdPromotion.campaign?.id!,
+            { relations: ["budget"] }
+          )
+
+          expect(campaign.budget).toEqual(
+            expect.objectContaining({
+              type: "spend",
+              limit: 1000,
+              used: 300,
+            })
+          )
+        })
+
+        it("should revert usage for type usage", async () => {
+          const campaignId = "campaign-id-2"
+          const createdPromotion = await createDefaultPromotion(service, {
+            code: "PROMO_1",
+            campaign_id: campaignId,
+          })
+          const createdPromotion2 = await createDefaultPromotion(service, {
+            code: "PROMO_2",
+            campaign_id: campaignId,
+          })
+          const createdPromotion3 = await createDefaultPromotion(service, {
+            code: "PROMO_3",
+            campaign_id: campaignId,
+          })
+
+          await service.registerUsage([
+            { amount: 200, code: createdPromotion.code! },
+            { amount: 500, code: createdPromotion.code! },
+          ])
+
+          await service.registerUsage([
+            { amount: 200, code: createdPromotion2.code! },
+            { amount: 500, code: createdPromotion3.code! },
+          ])
+
+          await service.revertUsage([
+            { amount: 200, code: createdPromotion.code! },
+            { amount: 500, code: createdPromotion.code! },
+          ])
+
+          await service.revertUsage([
+            { amount: 200, code: createdPromotion2.code! },
+            { amount: 500, code: createdPromotion3.code! },
+          ])
+
+          const campaign = await service.retrieveCampaign(campaignId, {
+            relations: ["budget"],
+          })
+
+          expect(campaign.budget).toEqual(
+            expect.objectContaining({
+              type: "usage",
+              limit: 1000,
+              used: 0,
+            })
+          )
+        })
+
+        it("should not throw an error when compute action with code does not exist", async () => {
+          const response = await service
+            .revertUsage([{ amount: 200, code: "DOESNOTEXIST" }])
+            .catch((e) => e)
+
+          expect(response).toEqual(undefined)
+        })
+      })
+    })
+  },
+})