medusa-store/integration-tests/modules/__tests__/link-modules/shipping-option-price-set.spec.ts

import {
  IFulfillmentModuleService,
  IPricingModuleService,
} from "@medusajs/types"
import { ContainerRegistrationKeys, Modules } from "@medusajs/utils"
import { medusaIntegrationTestRunner } from "@medusajs/test-utils"
import {
  adminHeaders,
  createAdminUser,
} from "../../../helpers/create-admin-user"

jest.setTimeout(50000)

const env = { MEDUSA_FF_MEDUSA_V2: true }

medusaIntegrationTestRunner({
  env,
  testSuite: ({ dbConnection, getContainer, api }) => {
    describe("Region and Payment Providers", () => {
      let appContainer
      let fulfillmentModule: IFulfillmentModuleService
      let pricingModule: IPricingModuleService
      let remoteQuery

      beforeAll(async () => {
        appContainer = getContainer()
        fulfillmentModule = appContainer.resolve(Modules.FULFILLMENT)
        pricingModule = appContainer.resolve(Modules.PRICING)
        remoteQuery = appContainer.resolve(
          ContainerRegistrationKeys.REMOTE_QUERY
        )
        await createAdminUser(dbConnection, adminHeaders, getContainer())
      })

      it("should query shipping option and price set link with remote query", async () => {
        const shippingProfile = await fulfillmentModule.createShippingProfiles({
          name: "Test",
          type: "default",
        })
        const fulfillmentSet = await fulfillmentModule.createFulfillmentSets({
          name: "Test",
          type: "test-type",
        })
        const remoteLink = appContainer.resolve(
          ContainerRegistrationKeys.REMOTE_LINK
        )
        const location = (
          await api.post(
            `/admin/stock-locations`,
            { name: "Test location" },
            adminHeaders
          )
        ).data.stock_location

        await remoteLink.create([
          {
            [Modules.STOCK_LOCATION]: {
              stock_location_id: location.id,
            },
            [Modules.FULFILLMENT]: {
              fulfillment_set_id: fulfillmentSet.id,
            },
          },
        ])

        await api.post(
          `/admin/stock-locations/${location.id}/fulfillment-providers`,
          { add: ["manual_test-provider"] },
          adminHeaders
        )

        const serviceZone = await fulfillmentModule.createServiceZones({
          name: "Test",
          fulfillment_set_id: fulfillmentSet.id,
          geo_zones: [
            {
              type: "country",
              country_code: "us",
            },
          ],
        })

        const shippingOption = await fulfillmentModule.createShippingOptions({
          name: "Test shipping option",
          service_zone_id: serviceZone.id,
          shipping_profile_id: shippingProfile.id,
          provider_id: "manual_test-provider",
          price_type: "flat",
          type: {
            label: "Test type",
            description: "Test description",
            code: "test-code",
          },
        })

        const priceSet = await pricingModule.createPriceSets({
          prices: [
            {
              amount: 3000,
              currency_code: "usd",
            },
            {
              amount: 5000,
              currency_code: "eur",
            },
          ],
        })

        await remoteLink.create([
          {
            [Modules.FULFILLMENT]: {
              shipping_option_id: shippingOption.id,
            },
            [Modules.PRICING]: {
              price_set_id: priceSet.id,
            },
          },
        ])

        const link = await remoteQuery({
          shipping_option: {
            fields: ["id"],
            price_set_link: {
              fields: ["id", "price_set_id", "shipping_option_id"],
            },
            prices: {
              fields: ["amount", "currency_code"],
            },
            calculated_price: {
              fields: ["calculated_amount", "currency_code"],
              __args: {
                context: {
                  currency_code: "eur",
                },
              },
            },
          },
        })

        expect(link).toHaveLength(1)
        expect(link).toEqual(
          expect.arrayContaining([
            expect.objectContaining({
              id: shippingOption.id,
              price_set_link: expect.objectContaining({
                price_set_id: priceSet.id,
                shipping_option_id: shippingOption.id,
              }),
              prices: expect.arrayContaining([
                expect.objectContaining({
                  amount: 5000,
                  currency_code: "eur",
                }),
                expect.objectContaining({
                  amount: 3000,
                  currency_code: "usd",
                }),
              ]),
              calculated_price: expect.objectContaining({
                calculated_amount: 5000,
                currency_code: "eur",
              }),
            }),
          ])
        )
      })
    })
  },
})