medusa-store/integration-tests/plugins/__tests__/customer/admin/update-customer-address.spec.ts

import { ModuleRegistrationName } from "@medusajs/modules-sdk"
import { ICustomerModuleService } from "@medusajs/types"
import path from "path"
import { startBootstrapApp } from "../../../../environment-helpers/bootstrap-app"
import { useApi } from "../../../../environment-helpers/use-api"
import { getContainer } from "../../../../environment-helpers/use-container"
import { initDb, useDb } from "../../../../environment-helpers/use-db"
import adminSeeder from "../../../../helpers/admin-seeder"

jest.setTimeout(50000)

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

describe("POST /admin/customers/:id/addresses/:address_id", () => {
  let dbConnection
  let appContainer
  let shutdownServer
  let customerModuleService: ICustomerModuleService

  beforeAll(async () => {
    const cwd = path.resolve(path.join(__dirname, "..", "..", ".."))
    dbConnection = await initDb({ cwd, env } as any)
    shutdownServer = await startBootstrapApp({ cwd, env })
    appContainer = getContainer()
    customerModuleService = appContainer.resolve(
      ModuleRegistrationName.CUSTOMER
    )
  })

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

  beforeEach(async () => {
    await adminSeeder(dbConnection)
  })

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

  it("should update a customer address", async () => {
    const customer = await customerModuleService.create({
      first_name: "John",
      last_name: "Doe",
    })

    const address = await customerModuleService.addAddresses({
      customer_id: customer.id,
      first_name: "John",
      last_name: "Doe",
      address_1: "Test street 1",
    })

    const api = useApi() as any
    const response = await api.post(
      `/admin/customers/${customer.id}/addresses/${address.id}`,
      {
        first_name: "Jane",
      },
      adminHeaders
    )

    expect(response.status).toEqual(200)
    expect(response.data.address).toEqual(
      expect.objectContaining({
        id: expect.any(String),
        first_name: "Jane",
        last_name: "Doe",
      })
    )
  })

  it("updates a new address to be default and unsets old one", async () => {
    const customer = await customerModuleService.create({
      first_name: "John",
      last_name: "Doe",
    })
    const [, address] = await customerModuleService.addAddresses([
      {
        customer_id: customer.id,
        first_name: "John",
        last_name: "Doe",
        address_1: "Test street 1",
        is_default_shipping: true,
      },
      {
        customer_id: customer.id,
        first_name: "John",
        last_name: "Doe",
        address_1: "Test street 2",
      },
    ])

    const api = useApi() as any
    const response = await api.post(
      `/admin/customers/${customer.id}/addresses/${address.id}`,
      {
        first_name: "jane",
        is_default_shipping: true,
      },
      adminHeaders
    )

    expect(response.status).toEqual(200)

    const [defaultAddress] = await customerModuleService.listAddresses({
      customer_id: customer.id,
      is_default_shipping: true,
    })

    expect(defaultAddress.first_name).toEqual("jane")
    expect(defaultAddress.address_1).toEqual("Test street 2")
  })

  // do the same as above but for billing address
  it("updates a new address to be default and unsets old one", async () => {
    const customer = await customerModuleService.create({
      first_name: "John",
      last_name: "Doe",
    })
    const [, address] = await customerModuleService.addAddresses([
      {
        customer_id: customer.id,
        first_name: "John",
        last_name: "Doe",
        address_1: "Test street 1",
        is_default_billing: true,
      },
      {
        customer_id: customer.id,
        first_name: "John",
        last_name: "Doe",
        address_1: "Test street 2",
      },
    ])

    const api = useApi() as any
    const response = await api.post(
      `/admin/customers/${customer.id}/addresses/${address.id}`,
      {
        first_name: "jane",
        is_default_billing: true,
      },
      adminHeaders
    )

    expect(response.status).toEqual(200)

    const [defaultAddress] = await customerModuleService.listAddresses({
      customer_id: customer.id,
      is_default_billing: true,
    })

    expect(defaultAddress.first_name).toEqual("jane")
    expect(defaultAddress.address_1).toEqual("Test street 2")
  })
})