medusa-store/integration-tests/api/__tests__/admin/variant.js

const path = require("path")

const setupServer = require("../../../environment-helpers/setup-server")
const { useApi } = require("../../../environment-helpers/use-api")
const { initDb, useDb } = require("../../../environment-helpers/use-db")
const { simpleProductFactory } = require("../../../factories")

const adminSeeder = require("../../../helpers/admin-seeder")
const adminVariantsSeeder = require("../../../helpers/admin-variants-seeder")
const productSeeder = require("../../../helpers/product-seeder")

const adminHeaders = {
  headers: {
    "x-medusa-access-token": "test_token",
  },
}

jest.setTimeout(30000)

describe("/admin/products", () => {
  let medusaProcess
  let dbConnection

  beforeAll(async () => {
    const cwd = path.resolve(path.join(__dirname, "..", ".."))
    dbConnection = await initDb({ cwd })
    medusaProcess = await setupServer({ cwd })
  })

  afterAll(async () => {
    const db = useDb()
    await db.shutdown()

    medusaProcess.kill()
  })

  describe("GET /admin/product-variants", () => {
    beforeEach(async () => {
      await productSeeder(dbConnection)
      await adminSeeder(dbConnection)
    })

    afterEach(async () => {
      const db = useDb()
      await db.teardown()
    })

    it("lists all product variants", async () => {
      const api = useApi()

      const response = await api
        .get("/admin/variants/", adminHeaders)
        .catch((err) => {
          console.log(err)
        })

      expect(response.status).toEqual(200)
      expect(response.data.variants).toEqual(
        expect.arrayContaining([
          expect.objectContaining(
            {
              id: "test-variant",
            },
            {
              id: "test-variant_2",
            },
            {
              id: "test-variant_1",
            }
          ),
        ])
      )
    })

    it("lists all product variants matching a specific sku", async () => {
      const api = useApi()
      const response = await api
        .get("/admin/variants?q=sku2", adminHeaders)
        .catch((err) => {
          console.log(err)
        })

      expect(response.status).toEqual(200)
      expect(response.data.variants.length).toEqual(1)
      expect(response.data.variants).toEqual(
        expect.arrayContaining([
          expect.objectContaining({
            sku: "test-sku2",
          }),
        ])
      )
    })

    it("lists all product variants matching a specific variant title", async () => {
      const api = useApi()
      const response = await api
        .get("/admin/variants?q=rank (1)", adminHeaders)
        .catch((err) => {
          console.log(err)
        })

      expect(response.status).toEqual(200)
      expect(response.data.variants.length).toEqual(1)
      expect(response.data.variants).toEqual(
        expect.arrayContaining([
          expect.objectContaining({
            id: "test-variant_1",
            sku: "test-sku1",
          }),
        ])
      )
    })

    it("lists all product variants matching a specific product title", async () => {
      const api = useApi()

      const response = await api
        .get("/admin/variants?q=Test product1", adminHeaders)
        .catch((err) => {
          console.log(err)
        })

      expect(response.status).toEqual(200)
      expect(response.data.variants.length).toEqual(2)
      expect(response.data.variants).toEqual(
        expect.arrayContaining([
          expect.objectContaining({
            product_id: "test-product1",
            id: "test-variant_3",
            sku: "test-sku3",
          }),
          expect.objectContaining({
            product_id: "test-product1",
            id: "test-variant_4",
            sku: "test-sku4",
          }),
        ])
      )
    })
  })

  describe("GET /admin/variants price selection strategy", () => {
    beforeEach(async () => {
      try {
        await adminVariantsSeeder(dbConnection)
      } catch (err) {
        console.log(err)
      }
      await adminSeeder(dbConnection)
    })

    afterEach(async () => {
      const db = useDb()
      await db.teardown()
    })

    it("selects prices based on the passed currency code", async () => {
      const api = useApi()

      const response = await api.get(
        "/admin/variants?id=test-variant&currency_code=usd",
        adminHeaders
      )

      expect(response.data).toMatchSnapshot({
        variants: [
          {
            id: "test-variant",
            original_price: 100,
            calculated_price: 80,
            calculated_price_type: "sale",
            original_price_incl_tax: null,
            calculated_price_incl_tax: null,
            original_tax: null,
            calculated_tax: null,
            options: expect.any(Array),
            prices: expect.any(Array),
            product: expect.any(Object),
            created_at: expect.any(String),
            updated_at: expect.any(String),
          },
        ],
      })
    })

    it("selects prices based on the passed region id", async () => {
      const api = useApi()

      const response = await api.get(
        "/admin/variants?id=test-variant&region_id=reg-europe",
        adminHeaders
      )

      expect(response.data).toMatchSnapshot({
        variants: [
          {
            id: "test-variant",
            original_price: 100,
            calculated_price: 80,
            calculated_price_type: "sale",
            original_price_incl_tax: 100,
            calculated_price_incl_tax: 80,
            original_tax: 0,
            calculated_tax: 0,
            options: expect.any(Array),
            prices: expect.any(Array),
            product: expect.any(Object),
            created_at: expect.any(String),
            updated_at: expect.any(String),
          },
        ],
      })
    })

    it("selects prices based on the passed region id and customer id", async () => {
      const api = useApi()

      const response = await api.get(
        "/admin/variants?id=test-variant&region_id=reg-europe&customer_id=test-customer",
        adminHeaders
      )

      expect(response.data).toMatchSnapshot({
        variants: [
          {
            id: "test-variant",
            original_price: 100,
            calculated_price: 40,
            calculated_price_type: "sale",
            original_price_incl_tax: 100,
            calculated_price_incl_tax: 40,
            original_tax: 0,
            calculated_tax: 0,
            prices: expect.any(Array),
            options: expect.any(Array),
            product: expect.any(Object),
            created_at: expect.any(String),
            updated_at: expect.any(String),
          },
        ],
      })
    })

    it("returns a list of variants matching the given ids", async () => {
      const api = useApi()

      const productData = {
        id: "test-product_filtering_by_variant_id",
        title: "Test product filtering by variant id",
        handle: "test-product_filtering_by_variant_id",
        options: [
          {
            id: "test-product_filtering_by_variant_id-option",
            title: "Size",
          },
        ],
        variants: [],
      }

      for (let i = 0; i < 25; i++) {
        productData.variants.push({
          product_id: productData.id,
          sku: `test-product_filtering_by_variant_id-${i}`,
          title: `test-product_filtering_by_variant_id-${i}`,
          options: [
            {
              option_id: "test-product_filtering_by_variant_id-option",
              value: `Large-${i}`,
            },
          ],
        })
      }

      const product = await simpleProductFactory(dbConnection, productData)

      const variantIds = product.variants.map((v) => v.id)
      const qs = "id[]=" + variantIds.join("&id[]=")

      const response = await api
        .get("/admin/variants?limit=30&" + qs, adminHeaders)
        .catch((err) => {
          console.log(err)
        })

      expect(response.status).toEqual(200)
      expect(response.data.variants.length).toEqual(variantIds.length)

      for (const variant of response.data.variants) {
        expect(variantIds).toContain(variant.id)
      }
    })
  })
})