feat: add useLifecycles

src/useLifecycles.ts

@@ -0,0 +1,12 @@
+import { getCurrentInstance, onMounted, onUnmounted } from 'vue'
+
+interface Callback {
+  (): any
+}
+
+export function useLifecycles(mountedCb: Callback, unmountCb: Callback): void {
+  if (getCurrentInstance()) {
+    mountedCb && onMounted(mountedCb)
+    unmountCb && onUnmounted(unmountCb)
+  }
+}

tests/useLifecycles.test.ts

@@ -0,0 +1,18 @@
+import { mount } from '@vue/test-utils'
+import { useLifecycles } from '../src/useLifecycles'
+
+describe('test useLifecycles', () => {
+  test('callback should be called when mounted or unmounted', () => {
+    const onMounted = jest.fn()
+    const onUnmounted = jest.fn()
+    const wrapper = mount({
+      template: '<div>test</div>',
+      setup () {
+        useLifecycles(onMounted, onUnmounted)
+      }
+    })
+    expect(onMounted).toHaveBeenCalledTimes(1)
+    wrapper.unmount()
+    expect(onUnmounted).toHaveBeenCalledTimes(1)
+  })
+})