feat: Update service zone

integration-tests/api/__tests__/admin/fulfillment-sets.spec.ts

@@ -26,7 +26,7 @@ medusaIntegrationTestRunner({
     })
 
     describe("POST /admin/fulfillment-sets/:id/service-zones", () => {
-      it("should create a service zone for a fulfillment set", async () => {
+      it("should create, update, and delete a service zone for a fulfillment set", async () => {
         const stockLocationResponse = await api.post(
           `/admin/stock-locations`,
           {
@@ -120,6 +120,77 @@ medusaIntegrationTestRunner({
             ]),
           })
         )
+
+        const serviceZoneId = fset.service_zones[0].id
+
+        const countryGeoZone = fset.service_zones[0].geo_zones.find(
+          (z) => z.type === "country"
+        )
+
+        // Updates an existing and creates a new one
+        const updateResponse = await api.post(
+          `/admin/fulfillment-sets/${fulfillmentSetId}/service-zones/${serviceZoneId}`,
+          {
+            name: "Test Zone Updated",
+            geo_zones: [
+              {
+                id: countryGeoZone.id,
+                country_code: "us",
+                type: "country",
+              },
+              {
+                country_code: "ca",
+                type: "country",
+              },
+            ],
+          },
+          adminHeaders
+        )
+
+        const updatedFset = updateResponse.data.fulfillment_set
+
+        expect(updateResponse.status).toEqual(200)
+        expect(updatedFset).toEqual(
+          expect.objectContaining({
+            name: "Fulfillment Set",
+            type: "shipping",
+            service_zones: expect.arrayContaining([
+              expect.objectContaining({
+                id: serviceZoneId,
+                name: "Test Zone Updated",
+                fulfillment_set_id: updatedFset.id,
+                geo_zones: expect.arrayContaining([
+                  expect.objectContaining({
+                    id: countryGeoZone.id,
+                    country_code: "us",
+                    type: "country",
+                  }),
+                  expect.objectContaining({
+                    country_code: "ca",
+                    type: "country",
+                  }),
+                ]),
+              }),
+            ]),
+          })
+        )
+
+        const deleteResponse = await api.delete(
+          `/admin/fulfillment-sets/${fulfillmentSetId}/service-zones/${serviceZoneId}`,
+          adminHeaders
+        )
+
+        expect(deleteResponse.status).toEqual(200)
+        expect(deleteResponse.data).toEqual(
+          expect.objectContaining({
+            id: serviceZoneId,
+            object: "service-zone",
+            deleted: true,
+            parent: expect.objectContaining({
+              id: fulfillmentSetId,
+            }),
+          })
+        )
       })
 
       it("should throw if invalid type is passed", async () => {
@@ -251,60 +322,6 @@ medusaIntegrationTestRunner({
       })
 
       describe("POST /admin/fulfillment-sets/:id/service-zones", () => {
-        it("should delete a service zone for a fulfillment set", async () => {
-          const stockLocationResponse = await api.post(
-            `/admin/stock-locations`,
-            {
-              name: "test location",
-            },
-            adminHeaders
-          )
-
-          const stockLocationId = stockLocationResponse.data.stock_location.id
-
-          const locationWithFSetResponse = await api.post(
-            `/admin/stock-locations/${stockLocationId}/fulfillment-sets?fields=id,*fulfillment_sets`,
-            {
-              name: "Fulfillment Set",
-              type: "shipping",
-            },
-            adminHeaders
-          )
-
-          const fulfillmentSetId =
-            locationWithFSetResponse.data.stock_location.fulfillment_sets[0].id
-
-          const response = await api.post(
-            `/admin/fulfillment-sets/${fulfillmentSetId}/service-zones`,
-            {
-              name: "Test Zone",
-              geo_zones: [],
-            },
-            adminHeaders
-          )
-
-          const fset = response.data.fulfillment_set
-
-          const serviceZoneId = fset.service_zones[0].id
-
-          const deleteResponse = await api.delete(
-            `/admin/fulfillment-sets/${fulfillmentSetId}/service-zones/${serviceZoneId}`,
-            adminHeaders
-          )
-
-          expect(deleteResponse.status).toEqual(200)
-          expect(deleteResponse.data).toEqual(
-            expect.objectContaining({
-              id: serviceZoneId,
-              object: "service-zone",
-              deleted: true,
-              parent: expect.objectContaining({
-                id: fulfillmentSetId,
-              }),
-            })
-          )
-        })
-
         it("should throw when fulfillment set doesn't exist", async () => {
           const deleteResponse = await api
             .delete(
@@ -314,7 +331,9 @@ medusaIntegrationTestRunner({
             .catch((e) => e.response)
 
           expect(deleteResponse.status).toEqual(404)
-          expect(deleteResponse.data.message).toEqual("FulfillmentSet with id: foo was not found")
+          expect(deleteResponse.data.message).toEqual(
+            "FulfillmentSet with id: foo was not found"
+          )
         })
       })
     })

packages/core-flows/src/fulfillment/steps/update-service-zones.ts

@@ -0,0 +1,43 @@
+import { ModuleRegistrationName } from "@medusajs/modules-sdk"
+import { FulfillmentWorkflow, IFulfillmentModuleService } from "@medusajs/types"
+import { getSelectsAndRelationsFromObjectArray } from "@medusajs/utils"
+import { StepResponse, createStep } from "@medusajs/workflows-sdk"
+
+type StepInput = FulfillmentWorkflow.UpdateServiceZonesWorkflowInput
+
+export const updateServiceZonesStepId = "update-service-zones"
+export const updateServiceZonesStep = createStep(
+  updateServiceZonesStepId,
+  async (input: StepInput, { container }) => {
+    const service = container.resolve<IFulfillmentModuleService>(
+      ModuleRegistrationName.FULFILLMENT
+    )
+
+    const { selects, relations } = getSelectsAndRelationsFromObjectArray([
+      input.update,
+    ])
+
+    const prevData = await service.listServiceZones(input.selector, {
+      select: selects,
+      relations,
+    })
+
+    const updatedServiceZones = await service.updateServiceZones(
+      input.selector,
+      input.update
+    )
+
+    return new StepResponse(updatedServiceZones, prevData)
+  },
+  async (prevData, { container }) => {
+    if (!prevData?.length) {
+      return
+    }
+
+    const service = container.resolve<IFulfillmentModuleService>(
+      ModuleRegistrationName.FULFILLMENT
+    )
+
+    await service.upsertServiceZones(prevData)
+  }
+)

packages/core-flows/src/fulfillment/workflows/index.ts

@@ -1,5 +1,6 @@
 export * from "./add-rules-to-fulfillment-shipping-option"
 export * from "./create-service-zones"
-export * from "./delete-service-zones"
 export * from "./create-shipping-options"
+export * from "./delete-service-zones"
 export * from "./remove-rules-from-fulfillment-shipping-option"
+export * from "./update-service-zones"

packages/core-flows/src/fulfillment/workflows/update-service-zones.ts

@@ -0,0 +1,15 @@
+import { FulfillmentWorkflow } from "@medusajs/types"
+import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import { updateServiceZonesStep } from "../steps/update-service-zones"
+
+export const updateServiceZonesWorkflowId = "update-service-zones-workflow"
+export const updateServiceZonesWorkflow = createWorkflow(
+  updateServiceZonesWorkflowId,
+  (
+    input: WorkflowData<FulfillmentWorkflow.UpdateServiceZonesWorkflowInput>
+  ): WorkflowData => {
+    const serviceZones = updateServiceZonesStep(input)
+
+    return serviceZones
+  }
+)