element-plus/packages/components/calendar/__tests__/calendar.spec.ts

import { nextTick } from 'vue'
import { mount } from '@vue/test-utils'
import Calendar from '../src/calendar.vue'

const _mount = (template: string, data?, otherObj?) =>
  mount({
    components: {
      'el-calendar': Calendar,
    },
    template,
    data,
    ...otherObj,
  })

describe('Calendar.vue', () => {
  it('create', async () => {
    const wrapper = _mount(
      `
    <el-calendar v-model="value"></el-calendar>
    `,
      () => ({ value: new Date('2019-04-01') })
    )
    const titleEl = wrapper.find('.el-calendar__title')
    expect(
      /2019.*April/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    expect(wrapper.element.querySelectorAll('thead th').length).toBe(7)
    const rows = wrapper.element.querySelectorAll('.el-calendar-table__row')
    expect(rows.length).toBe(6)
    ;(rows[5].firstElementChild as HTMLElement).click()

    await nextTick()
    expect(
      /2019.*May/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    const vm = wrapper.vm as any
    const date = vm.value
    expect(date.getFullYear()).toBe(2019)
    expect(date.getMonth()).toBe(4)
    expect(
      (wrapper.find('.is-selected span').element as HTMLElement).innerHTML
    ).toBe('5')
  })

  it('range', () => {
    const wrapper = _mount(`
    <el-calendar :range="[new Date(2019, 2, 4), new Date(2019, 2, 24)]"></el-calendar>
    `)
    const titleEl = wrapper.find('.el-calendar__title')
    expect(
      /2019.*March/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    const rows = wrapper.element.querySelectorAll('.el-calendar-table__row')
    expect(rows.length).toBe(4)
    expect(
      wrapper.element.querySelector('.el-calendar__button-group')
    ).toBeNull()
  })

  // https://github.com/element-plus/element-plus/issues/3155
  it('range when the start date will be calculated to last month', () => {
    const wrapper = _mount(`
    <el-calendar :range="[new Date(2021, 1, 2), new Date(2021, 1, 28)]"></el-calendar>
    `)
    const titleEl = wrapper.find('.el-calendar__title')
    expect(
      /2021.*January/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    const rows = wrapper.element.querySelectorAll('.el-calendar-table__row')
    expect(rows.length).toBe(5)
    expect(
      wrapper.element.querySelector('.el-calendar__button-group')
    ).toBeNull()
  })

  it('range tow monthes', async () => {
    const wrapper = _mount(`
    <el-calendar :range="[new Date(2019, 3, 14), new Date(2019, 4, 18)]"></el-calendar>
    `)
    const titleEl = wrapper.find('.el-calendar__title')
    expect(
      /2019.*April/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    const dateTables = wrapper.element.querySelectorAll(
      '.el-calendar-table.is-range'
    )
    expect(dateTables.length).toBe(2)
    const rows = wrapper.element.querySelectorAll('.el-calendar-table__row')
    expect(rows.length).toBe(5)
    const cell = rows[rows.length - 1].firstElementChild
    ;(cell as HTMLElement).click()

    await nextTick()

    expect(
      /2019.*May/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    expect(cell.classList.contains('is-selected')).toBeTruthy()
  })

  // https://github.com/element-plus/element-plus/issues/3155
  it('range tow monthes when the start date will be calculated to last month', async () => {
    const wrapper = _mount(`
    <el-calendar :range="[new Date(2021, 1, 2), new Date(2021, 2, 21)]"></el-calendar>
    `)
    const titleEl = wrapper.find('.el-calendar__title')
    expect(
      /2021.*January/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    const dateTables = wrapper.element.querySelectorAll(
      '.el-calendar-table.is-range'
    )
    expect(dateTables.length).toBe(3)
    const rows = wrapper.element.querySelectorAll('.el-calendar-table__row')
    expect(rows.length).toBe(8)
    const cell = rows[rows.length - 1].firstElementChild
    ;(cell as HTMLElement).click()

    await nextTick()

    expect(
      /2021.*March/.test((titleEl.element as HTMLElement).innerHTML)
    ).toBeTruthy()
    expect(cell.classList.contains('is-selected')).toBeTruthy()
  })

  it('firstDayOfWeek', async () => {
    // default en locale, weekStart 0 Sunday
    const wrapper = _mount(
      `
    <el-calendar v-model="value"></el-calendar>
    `,
      () => ({ value: new Date('2019-04-01') })
    )
    const head = wrapper.element.querySelector('.el-calendar-table thead')
    expect((head.firstElementChild as HTMLElement).innerHTML).toBe('Sun')
    expect((head.lastElementChild as HTMLElement).innerHTML).toBe('Sat')
    const firstRow = wrapper.element.querySelector('.el-calendar-table__row')
    expect((firstRow.firstElementChild as HTMLElement).innerHTML).toContain(
      '31'
    )
    expect((firstRow.lastElementChild as HTMLElement).innerHTML).toContain('6')
  })

  it('firstDayOfWeek in range mode', async () => {
    const wrapper = _mount(
      `
    <el-calendar v-model="value" :first-day-of-week="7" :range="[new Date(2019, 1, 3), new Date(2019, 2, 23)]"></el-calendar>
    `,
      () => ({ value: new Date('2019-03-04') })
    )
    const head = wrapper.element.querySelector('.el-calendar-table thead')
    expect((head.firstElementChild as HTMLElement).innerHTML).toBe('Sun')
    expect((head.lastElementChild as HTMLElement).innerHTML).toBe('Sat')
    const firstRow = wrapper.element.querySelector('.el-calendar-table__row')
    expect((firstRow.firstElementChild as HTMLElement).innerHTML).toContain('3')
    expect((firstRow.lastElementChild as HTMLElement).innerHTML).toContain('9')
  })

  it('click previous month or next month', async () => {
    const wrapper = _mount(
      `
    <el-calendar v-model="value"></el-calendar>
    `,
      () => ({ value: new Date('2019-04-01') })
    )
    await nextTick()
    const btns = wrapper.findAll('.el-button')
    const prevBtn = btns.at(0)
    const nextBtn = btns.at(2)
    await prevBtn.trigger('click')
    expect(wrapper.find('.is-selected').text()).toBe('1')
    await nextBtn.trigger('click')
    expect(wrapper.find('.is-selected').text()).toBe('1')
  })
})