core/packages/runtime-core/__tests__/apiSetupHelpers.spec.ts

import {
  type ComponentInternalInstance,
  type ComputedRef,
  Fragment,
  type Ref,
  type SetupContext,
  Suspense,
  computed,
  createApp,
  createBlock,
  createElementBlock,
  createElementVNode,
  createVNode,
  defineComponent,
  getCurrentInstance,
  h,
  nextTick,
  nodeOps,
  onMounted,
  openBlock,
  ref,
  render,
  serializeInner,
  shallowReactive,
  watch,
} from '@vue/runtime-test'
import {
  createPropsRestProxy,
  defineEmits,
  defineExpose,
  defineProps,
  mergeDefaults,
  mergeModels,
  useAttrs,
  useModel,
  useSlots,
  withAsyncContext,
  withDefaults,
} from '../src/apiSetupHelpers'

describe('SFC <script setup> helpers', () => {
  test('should warn runtime usage', () => {
    defineProps()
    expect(`defineProps() is a compiler-hint`).toHaveBeenWarned()

    defineEmits()
    expect(`defineEmits() is a compiler-hint`).toHaveBeenWarned()

    defineExpose()
    expect(`defineExpose() is a compiler-hint`).toHaveBeenWarned()

    withDefaults({}, {})
    expect(`withDefaults() is a compiler-hint`).toHaveBeenWarned()
  })

  test('useSlots / useAttrs (no args)', () => {
    let slots: SetupContext['slots'] | undefined
    let attrs: SetupContext['attrs'] | undefined
    const Comp = {
      setup() {
        slots = useSlots()
        attrs = useAttrs()
        return () => {}
      },
    }
    const passedAttrs = { id: 'foo' }
    const passedSlots = {
      default: () => {},
      x: () => {},
    }
    render(h(Comp, passedAttrs, passedSlots), nodeOps.createElement('div'))
    expect(typeof slots!.default).toBe('function')
    expect(typeof slots!.x).toBe('function')
    expect(attrs).toMatchObject(passedAttrs)
  })

  test('useSlots / useAttrs (with args)', () => {
    let slots: SetupContext['slots'] | undefined
    let attrs: SetupContext['attrs'] | undefined
    let ctx: SetupContext | undefined
    const Comp = defineComponent({
      setup(_, _ctx) {
        slots = useSlots()
        attrs = useAttrs()
        ctx = _ctx
        return () => {}
      },
    })
    render(h(Comp), nodeOps.createElement('div'))
    expect(slots).toBe(ctx!.slots)
    expect(attrs).toBe(ctx!.attrs)
  })

  describe('mergeDefaults', () => {
    test('object syntax', () => {
      const merged = mergeDefaults(
        {
          foo: null,
          bar: { type: String, required: false },
          baz: String,
        },
        {
          foo: 1,
          bar: 'baz',
          baz: 'qux',
        },
      )
      expect(merged).toMatchObject({
        foo: { default: 1 },
        bar: { type: String, required: false, default: 'baz' },
        baz: { type: String, default: 'qux' },
      })
    })

    test('array syntax', () => {
      const merged = mergeDefaults(['foo', 'bar', 'baz'], {
        foo: 1,
        bar: 'baz',
        baz: 'qux',
      })
      expect(merged).toMatchObject({
        foo: { default: 1 },
        bar: { default: 'baz' },
        baz: { default: 'qux' },
      })
    })

    test('merging with skipFactory', () => {
      const fn = () => {}
      const merged = mergeDefaults(['foo', 'bar', 'baz'], {
        foo: fn,
        __skip_foo: true,
      })
      expect(merged).toMatchObject({
        foo: { default: fn, skipFactory: true },
      })
    })

    test('should warn missing', () => {
      mergeDefaults({}, { foo: 1 })
      expect(
        `props default key "foo" has no corresponding declaration`,
      ).toHaveBeenWarned()
    })
  })

  describe('mergeModels', () => {
    test('array syntax', () => {
      expect(mergeModels(['foo', 'bar'], ['baz'])).toMatchObject([
        'foo',
        'bar',
        'baz',
      ])
    })

    test('object syntax', () => {
      expect(
        mergeModels({ foo: null, bar: { required: true } }, ['baz']),
      ).toMatchObject({
        foo: null,
        bar: { required: true },
        baz: {},
      })

      expect(
        mergeModels(['baz'], { foo: null, bar: { required: true } }),
      ).toMatchObject({
        foo: null,
        bar: { required: true },
        baz: {},
      })
    })

    test('overwrite', () => {
      expect(
        mergeModels(
          { foo: null, bar: { required: true } },
          { bar: {}, baz: {} },
        ),
      ).toMatchObject({
        foo: null,
        bar: {},
        baz: {},
      })
    })
  })

  describe('useModel', () => {
    test('basic', async () => {
      let foo: any
      const update = () => {
        foo.value = 'bar'
      }

      const compRender = vi.fn()
      const Comp = defineComponent({
        props: ['modelValue'],
        emits: ['update:modelValue'],
        setup(props) {
          foo = useModel(props, 'modelValue')
          return () => {
            compRender()
            return foo.value
          }
        },
      })

      const msg = ref('')
      const setValue = vi.fn(v => (msg.value = v))
      const root = nodeOps.createElement('div')
      createApp(() =>
        h(Comp, {
          modelValue: msg.value,
          'onUpdate:modelValue': setValue,
        }),
      ).mount(root)

      expect(foo.value).toBe('')
      expect(msg.value).toBe('')
      expect(setValue).not.toBeCalled()
      expect(compRender).toBeCalledTimes(1)
      expect(serializeInner(root)).toBe('')

      // update from child
      update()

      await nextTick()
      expect(msg.value).toBe('bar')
      expect(foo.value).toBe('bar')
      expect(setValue).toBeCalledTimes(1)
      expect(compRender).toBeCalledTimes(2)
      expect(serializeInner(root)).toBe('bar')

      // update from parent
      msg.value = 'qux'
      expect(msg.value).toBe('qux')

      await nextTick()
      expect(msg.value).toBe('qux')
      expect(foo.value).toBe('qux')
      expect(setValue).toBeCalledTimes(1)
      expect(compRender).toBeCalledTimes(3)
      expect(serializeInner(root)).toBe('qux')
    })

    test('without parent value (local mutation)', async () => {
      let foo: any
      const update = () => {
        foo.value = 'bar'
      }

      const compRender = vi.fn()
      const Comp = defineComponent({
        props: ['foo'],
        emits: ['update:foo'],
        setup(props) {
          foo = useModel(props, 'foo')
          return () => {
            compRender()
            return foo.value
          }
        },
      })

      const root = nodeOps.createElement('div')
      const updateFoo = vi.fn()
      render(h(Comp, { 'onUpdate:foo': updateFoo }), root)
      expect(compRender).toBeCalledTimes(1)
      expect(serializeInner(root)).toBe('<!---->')

      expect(foo.value).toBeUndefined()
      update()
      // when parent didn't provide value, local mutation is enabled
      expect(foo.value).toBe('bar')

      await nextTick()
      expect(updateFoo).toBeCalledTimes(1)
      expect(compRender).toBeCalledTimes(2)
      expect(serializeInner(root)).toBe('bar')
    })

    test('default value', async () => {
      let count: any
      const inc = () => {
        count.value++
      }

      const compRender = vi.fn()
      const Comp = defineComponent({
        props: { count: { default: 0 } },
        emits: ['update:count'],
        setup(props) {
          count = useModel(props, 'count')
          return () => {
            compRender()
            return count.value
          }
        },
      })

      const root = nodeOps.createElement('div')
      const updateCount = vi.fn()
      render(h(Comp, { 'onUpdate:count': updateCount }), root)
      expect(compRender).toBeCalledTimes(1)
      expect(serializeInner(root)).toBe('0')

      expect(count.value).toBe(0)

      inc()
      // when parent didn't provide value, local mutation is enabled
      expect(count.value).toBe(1)

      await nextTick()

      expect(updateCount).toBeCalledTimes(1)
      expect(compRender).toBeCalledTimes(2)
      expect(serializeInner(root)).toBe('1')
    })

    test('parent limiting child value', async () => {
      let childCount: Ref<number>

      const compRender = vi.fn()
      const Comp = defineComponent({
        props: ['count'],
        emits: ['update:count'],
        setup(props) {
          childCount = useModel(props, 'count')
          return () => {
            compRender()
            return childCount.value
          }
        },
      })

      const Parent = defineComponent({
        setup() {
          const count = ref(0)
          watch(count, () => {
            if (count.value < 0) {
              count.value = 0
            }
          })
          return () =>
            h(Comp, {
              count: count.value,
              'onUpdate:count': val => {
                count.value = val
              },
            })
        },
      })

      const root = nodeOps.createElement('div')
      render(h(Parent), root)
      expect(serializeInner(root)).toBe('0')

      // child update
      childCount!.value = 1
      // not yet updated
      expect(childCount!.value).toBe(0)

      await nextTick()
      expect(childCount!.value).toBe(1)
      expect(serializeInner(root)).toBe('1')

      // child update to invalid value
      childCount!.value = -1
      // not yet updated
      expect(childCount!.value).toBe(1)

      await nextTick()
      // limited to 0 by parent
      expect(childCount!.value).toBe(0)
      expect(serializeInner(root)).toBe('0')
    })

    test('has parent value -> no parent value', async () => {
      let childCount: Ref<number>

      const compRender = vi.fn()
      const Comp = defineComponent({
        props: ['count'],
        emits: ['update:count'],
        setup(props) {
          childCount = useModel(props, 'count')
          return () => {
            compRender()
            return childCount.value
          }
        },
      })

      const toggle = ref(true)
      const Parent = defineComponent({
        setup() {
          const count = ref(0)
          return () =>
            toggle.value
              ? h(Comp, {
                  count: count.value,
                  'onUpdate:count': val => {
                    count.value = val
                  },
                })
              : h(Comp)
        },
      })

      const root = nodeOps.createElement('div')
      render(h(Parent), root)
      expect(serializeInner(root)).toBe('0')

      // child update
      childCount!.value = 1
      // not yet updated
      expect(childCount!.value).toBe(0)

      await nextTick()
      expect(childCount!.value).toBe(1)
      expect(serializeInner(root)).toBe('1')

      // parent change
      toggle.value = false

      await nextTick()
      // localValue should be reset
      expect(childCount!.value).toBeUndefined()
      expect(serializeInner(root)).toBe('<!---->')

      // child local mutation should continue to work
      childCount!.value = 2
      expect(childCount!.value).toBe(2)

      await nextTick()
      expect(serializeInner(root)).toBe('2')
    })

    // #9838
    test('pass modelValue to slot (optimized mode) ', async () => {
      let foo: any
      const update = () => {
        foo.value = 'bar'
      }

      const Comp = {
        render(this: any) {
          return this.$slots.default()
        },
      }

      const childRender = vi.fn()
      const slotRender = vi.fn()
      const Child = defineComponent({
        props: ['modelValue'],
        emits: ['update:modelValue'],
        setup(props) {
          foo = useModel(props, 'modelValue')
          return () => {
            childRender()
            return (
              openBlock(),
              createElementBlock(Fragment, null, [
                createVNode(Comp, null, {
                  default: () => {
                    slotRender()
                    return createElementVNode('div', null, foo.value)
                  },
                  _: 1 /* STABLE */,
                }),
              ])
            )
          }
        },
      })

      const msg = ref('')
      const setValue = vi.fn(v => (msg.value = v))
      const root = nodeOps.createElement('div')
      createApp({
        render() {
          return (
            openBlock(),
            createBlock(
              Child,
              {
                modelValue: msg.value,
                'onUpdate:modelValue': setValue,
              },
              null,
              8 /* PROPS */,
              ['modelValue'],
            )
          )
        },
      }).mount(root)

      expect(foo.value).toBe('')
      expect(msg.value).toBe('')
      expect(setValue).not.toBeCalled()
      expect(childRender).toBeCalledTimes(1)
      expect(slotRender).toBeCalledTimes(1)
      expect(serializeInner(root)).toBe('<div></div>')

      // update from child
      update()

      await nextTick()
      expect(msg.value).toBe('bar')
      expect(foo.value).toBe('bar')
      expect(setValue).toBeCalledTimes(1)
      expect(childRender).toBeCalledTimes(2)
      expect(slotRender).toBeCalledTimes(2)
      expect(serializeInner(root)).toBe('<div>bar</div>')
    })

    test('with modifiers & transformers', async () => {
      let childMsg: Ref<string>
      let childModifiers: Record<string, true | undefined>

      const compRender = vi.fn()
      const Comp = defineComponent({
        props: ['msg', 'msgModifiers'],
        emits: ['update:msg'],
        setup(props) {
          ;[childMsg, childModifiers] = useModel(props, 'msg', {
            get(val) {
              return val.toLowerCase()
            },
            set(val) {
              if (childModifiers.upper) {
                return val.toUpperCase()
              }
            },
          })
          return () => {
            compRender()
            return childMsg.value
          }
        },
      })

      const msg = ref('HI')
      const Parent = defineComponent({
        setup() {
          return () =>
            h(Comp, {
              msg: msg.value,
              msgModifiers: { upper: true },
              'onUpdate:msg': val => {
                msg.value = val
              },
            })
        },
      })

      const root = nodeOps.createElement('div')
      render(h(Parent), root)

      // should be lowered
      expect(serializeInner(root)).toBe('hi')

      // child update
      childMsg!.value = 'Hmm'

      await nextTick()
      expect(childMsg!.value).toBe('hmm')
      expect(serializeInner(root)).toBe('hmm')
      // parent should get uppercase value
      expect(msg.value).toBe('HMM')

      // parent update
      msg.value = 'Ughh'
      await nextTick()
      expect(serializeInner(root)).toBe('ughh')
      expect(msg.value).toBe('Ughh')

      // child update again
      childMsg!.value = 'ughh'
      await nextTick()
      expect(msg.value).toBe('UGHH')
    })
  })

  test('createPropsRestProxy', () => {
    const original = shallowReactive({
      foo: 1,
      bar: 2,
      baz: 3,
    })
    const rest = createPropsRestProxy(original, ['foo', 'bar'])
    expect('foo' in rest).toBe(false)
    expect('bar' in rest).toBe(false)
    expect(rest.baz).toBe(3)
    expect(Object.keys(rest)).toEqual(['baz'])

    original.baz = 4
    expect(rest.baz).toBe(4)
  })

  describe('withAsyncContext', () => {
    // disable options API because applyOptions() also resets currentInstance
    // and we want to ensure the logic works even with Options API disabled.
    beforeEach(() => {
      __FEATURE_OPTIONS_API__ = false
    })

    afterEach(() => {
      __FEATURE_OPTIONS_API__ = true
    })

    test('basic', async () => {
      const spy = vi.fn()

      let beforeInstance: ComponentInternalInstance | null = null
      let afterInstance: ComponentInternalInstance | null = null
      let resolve: (msg: string) => void

      const Comp = defineComponent({
        async setup() {
          let __temp: any, __restore: any

          beforeInstance = getCurrentInstance()

          const msg =
            (([__temp, __restore] = withAsyncContext(
              () =>
                new Promise(r => {
                  resolve = r
                }),
            )),
            (__temp = await __temp),
            __restore(),
            __temp)

          // register the lifecycle after an await statement
          onMounted(spy)
          afterInstance = getCurrentInstance()
          return () => msg
        },
      })

      const root = nodeOps.createElement('div')
      render(
        h(() => h(Suspense, () => h(Comp))),
        root,
      )

      expect(spy).not.toHaveBeenCalled()
      resolve!('hello')
      // wait a macro task tick for all micro ticks to resolve
      await new Promise(r => setTimeout(r))
      // mount hook should have been called
      expect(spy).toHaveBeenCalled()
      // should retain same instance before/after the await call
      expect(beforeInstance).toBe(afterInstance)
      expect(serializeInner(root)).toBe('hello')
    })

    test('error handling', async () => {
      const spy = vi.fn()

      let beforeInstance: ComponentInternalInstance | null = null
      let afterInstance: ComponentInternalInstance | null = null
      let reject: () => void

      const Comp = defineComponent({
        async setup() {
          let __temp: any, __restore: any

          beforeInstance = getCurrentInstance()
          try {
            ;[__temp, __restore] = withAsyncContext(
              () =>
                new Promise((_, rj) => {
                  reject = rj
                }),
            )
            __temp = await __temp
            __restore()
          } catch (e: any) {
            // ignore
          }
          // register the lifecycle after an await statement
          onMounted(spy)
          afterInstance = getCurrentInstance()
          return () => ''
        },
      })

      const root = nodeOps.createElement('div')
      render(
        h(() => h(Suspense, () => h(Comp))),
        root,
      )

      expect(spy).not.toHaveBeenCalled()
      reject!()
      // wait a macro task tick for all micro ticks to resolve
      await new Promise(r => setTimeout(r))
      // mount hook should have been called
      expect(spy).toHaveBeenCalled()
      // should retain same instance before/after the await call
      expect(beforeInstance).toBe(afterInstance)
    })

    test('should not leak instance on multiple awaits', async () => {
      let resolve: (val?: any) => void
      let beforeInstance: ComponentInternalInstance | null = null
      let afterInstance: ComponentInternalInstance | null = null
      let inBandInstance: ComponentInternalInstance | null = null
      let outOfBandInstance: ComponentInternalInstance | null = null

      const ready = new Promise(r => {
        resolve = r
      })

      async function doAsyncWork() {
        // should still have instance
        inBandInstance = getCurrentInstance()
        await Promise.resolve()
        // should not leak instance
        outOfBandInstance = getCurrentInstance()
      }

      const Comp = defineComponent({
        async setup() {
          let __temp: any, __restore: any

          beforeInstance = getCurrentInstance()

          // first await
          ;[__temp, __restore] = withAsyncContext(() => Promise.resolve())
          __temp = await __temp
          __restore()

          // setup exit, instance set to null, then resumed
          ;[__temp, __restore] = withAsyncContext(() => doAsyncWork())
          __temp = await __temp
          __restore()

          afterInstance = getCurrentInstance()
          return () => {
            resolve()
            return ''
          }
        },
      })

      const root = nodeOps.createElement('div')
      render(
        h(() => h(Suspense, () => h(Comp))),
        root,
      )

      await ready
      expect(inBandInstance).toBe(beforeInstance)
      expect(outOfBandInstance).toBeNull()
      expect(afterInstance).toBe(beforeInstance)
      expect(getCurrentInstance()).toBeNull()
    })

    test('should not leak on multiple awaits + error', async () => {
      let resolve: (val?: any) => void
      const ready = new Promise(r => {
        resolve = r
      })

      const Comp = defineComponent({
        async setup() {
          let __temp: any, __restore: any
          ;[__temp, __restore] = withAsyncContext(() => Promise.resolve())
          __temp = await __temp
          __restore()
          ;[__temp, __restore] = withAsyncContext(() => Promise.reject())
          __temp = await __temp
          __restore()
        },
        render() {},
      })

      const app = createApp(() => h(Suspense, () => h(Comp)))
      app.config.errorHandler = () => {
        resolve()
        return false
      }

      const root = nodeOps.createElement('div')
      app.mount(root)

      await ready
      expect(getCurrentInstance()).toBeNull()
    })

    // #4050
    test('race conditions', async () => {
      const uids = {
        one: { before: NaN, after: NaN },
        two: { before: NaN, after: NaN },
      }

      const Comp = defineComponent({
        props: ['name'],
        async setup(props: { name: 'one' | 'two' }) {
          let __temp: any, __restore: any

          uids[props.name].before = getCurrentInstance()!.uid
          ;[__temp, __restore] = withAsyncContext(() => Promise.resolve())
          __temp = await __temp
          __restore()

          uids[props.name].after = getCurrentInstance()!.uid
          return () => ''
        },
      })

      const app = createApp(() =>
        h(Suspense, () =>
          h('div', [h(Comp, { name: 'one' }), h(Comp, { name: 'two' })]),
        ),
      )
      const root = nodeOps.createElement('div')
      app.mount(root)

      await new Promise(r => setTimeout(r))
      expect(uids.one.before).not.toBe(uids.two.before)
      expect(uids.one.before).toBe(uids.one.after)
      expect(uids.two.before).toBe(uids.two.after)
    })

    test('should teardown in-scope effects', async () => {
      let resolve: (val?: any) => void
      const ready = new Promise(r => {
        resolve = r
      })

      let c: ComputedRef

      const Comp = defineComponent({
        async setup() {
          let __temp: any, __restore: any
          ;[__temp, __restore] = withAsyncContext(() => Promise.resolve())
          __temp = await __temp
          __restore()

          c = computed(() => {})
          // register the lifecycle after an await statement
          onMounted(resolve)
          return () => ''
        },
      })

      const app = createApp(() => h(Suspense, () => h(Comp)))
      const root = nodeOps.createElement('div')
      app.mount(root)

      await ready
      expect(c!.effect.active).toBe(true)

      app.unmount()
      expect(c!.effect.active).toBe(false)
    })
  })
})