@juzi/wechaty/src/user-mixins/poolify.spec.ts

Wechaty is a RPA SDK for Chatbot Makers.

#!/usr/bin/env -S node --no-warnings --loader ts-node/esm
import {
  test,
  AssertEqual,
}               from 'tstest'

import {
  poolifyMixin,
}                       from './poolify.js'
import {
  wechatifyUserModule,
  wechatifyMixin,
}                       from './wechatify.js'

test('poolifyMixin()', async t => {
  class Life {

    meaning = 42
    constructor () {}

  }

  class UserClassTest extends poolifyMixin(Life)<UserClassTest>() {

    id = 0
    constructor () { super() }

  }

  const meaningType: AssertEqual<number, typeof UserClassTest.prototype.meaning> = true
  t.ok(meaningType, 'should get meaning type for class instance')

  type PoolGetReturnType = Exclude<
    ReturnType<typeof UserClassTest.pool.get>,
    undefined
  >
  const poolGetType: AssertEqual<
    PoolGetReturnType,
    UserClassTest
  > = true
  t.ok(poolGetType, 'should get Life type for pool.get()')

  type LoadReturnType = ReturnType<typeof UserClassTest.load>
  const loadType: AssertEqual<
    LoadReturnType,
    UserClassTest
  > = true
  t.ok(loadType, 'should get UserClassTest type for load()')
})

test('poolifyMixin() pool', async t => {
  class Life {

    meaning = 42

  }

  class UserClassTest extends wechatifyMixin(
    poolifyMixin(Life)<UserClassTest>(),
  ) {

    id = 0
    constructor () { super() }

  }

  const UserClass1 = wechatifyUserModule(UserClassTest)({} as any)
  const UserClass2 = wechatifyUserModule(UserClassTest)({} as any)

  t.equal(UserClass1.pool.size, 0, 'should be size 0 for class 1')
  t.equal(UserClass2.pool.size, 0, 'should be size 0 for class 2')

  UserClass1.load('id1')
  t.equal(UserClass1.pool.size, 1, 'should be size 1 for class 1 after class1.load()')
  t.equal(UserClass2.pool.size, 0, 'should be size 0 for class 2 after class1.load()')

  UserClass2.load('id2')
  t.equal(UserClass1.pool.size, 1, 'should be size 1 for class 1 after class2.load()')
  t.equal(UserClass2.pool.size, 1, 'should be size 1 for class 2 after class2.load()')
})

test('static findAll()', async t => {

  class UserClassTest extends wechatifyMixin(
    poolifyMixin(Object)<UserClassTest>(),
  ) {

    static async findAll<T extends typeof UserClassTest> (
      this : T,
      id   : string,
    ) {
      return this.load(id)
    }

    id = 0
    constructor () { super() }

  }

  /* eslint-disable no-undef */
  type findAllReturnType = ReturnType<
    typeof UserClassTest.findAll
  >
  const typeTest: AssertEqual<
    findAllReturnType,
    Promise<UserClassTest>
  > = true

  t.ok(typeTest, 'should return the UserClassTest type')
})