core/packages/runtime-core/__tests__/apiCreateComponent.spec.tsx

import { createComponent } from '../src/apiCreateComponent'
import { ref } from '@vue/reactivity'
import { PropType } from '../src/componentProps'
import { h } from '../src/h'

// mock React just for TSX testing purposes
const React = {
  createElement: () => {}
}

test('createComponent type inference', () => {
  const MyComponent = createComponent({
    props: {
      a: Number,
      // required should make property non-void
      b: {
        type: String,
        required: true
      },
      // default value should infer type and make it non-void
      bb: {
        default: 'hello'
      },
      // explicit type casting
      cc: Array as PropType<string[]>,
      // required + type casting
      dd: {
        type: Array as PropType<string[]>,
        required: true
      },
      // explicit type casting with constructor
      ccc: Array as () => string[],
      // required + contructor type casting
      ddd: {
        type: Array as () => string[],
        required: true
      }
    },
    setup(props) {
      props.a && props.a * 2
      props.b.slice()
      props.bb.slice()
      props.cc && props.cc.push('hoo')
      props.dd.push('dd')
      return {
        c: ref(1),
        d: {
          e: ref('hi')
        }
      }
    },
    render() {
      const props = this.$props
      props.a && props.a * 2
      props.b.slice()
      props.bb.slice()
      props.cc && props.cc.push('hoo')
      props.dd.push('dd')
      this.a && this.a * 2
      this.b.slice()
      this.bb.slice()
      this.c * 2
      this.d.e.slice()
      this.cc && this.cc.push('hoo')
      this.dd.push('dd')
      return h('div', this.bb)
    }
  })
  // test TSX props inference
  ;<MyComponent
    a={1}
    b="foo"
    dd={['foo']}
    ddd={['foo']}
    // should allow extraneous as attrs
    class="bar"
  />
})

test('type inference w/ optional props declaration', () => {
  const Comp = createComponent({
    setup(props: { msg: string }) {
      props.msg
      return {
        a: 1
      }
    },
    render() {
      this.$props.msg
      this.msg
      this.a * 2
      return h('div', this.msg)
    }
  })
  ;<Comp msg="hello" />
})

test('type inference w/ direct setup function', () => {
  const Comp = createComponent((props: { msg: string }) => {
    return () => <div>{props.msg}</div>
  })
  ;<Comp msg="hello" />
})

test('type inference w/ array props declaration', () => {
  const Comp = createComponent({
    props: ['a', 'b'],
    setup(props) {
      props.a
      props.b
      return {
        c: 1
      }
    },
    render() {
      this.$props.a
      this.$props.b
      this.a
      this.b
      this.c
    }
  })
  ;<Comp a={1} b={2} />
})

test('with legacy options', () => {
  createComponent({
    props: { a: Number },
    setup() {
      return {
        b: 123
      }
    },
    data() {
      // Limitation: we cannot expose the return result of setup() on `this`
      // here in data() - somehow that would mess up the inference
      return {
        c: this.a || 123
      }
    },
    computed: {
      d(): number {
        return this.b + 1
      }
    },
    watch: {
      a() {
        this.b + 1
      }
    },
    created() {
      this.a && this.a * 2
      this.b * 2
      this.c * 2
      this.d * 2
    },
    methods: {
      doSomething() {
        this.a && this.a * 2
        this.b * 2
        this.c * 2
        this.d * 2
        return (this.a || 0) + this.b + this.c + this.d
      }
    },
    render() {
      this.a && this.a * 2
      this.b * 2
      this.c * 2
      this.d * 2
      return h('div', (this.a || 0) + this.b + this.c + this.d)
    }
  })
})