fix: use correct type

arnoson · Feb 21, 2023 · 66d8049 · 66d8049
1 parent 3803a96
commit 66d8049
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/index.test.ts b/tests/index.test.ts
@@ -3,7 +3,8 @@ import {
   registerComponent,
   mountComponent,
   mountComponents,
-  defineProps
+  defineProps,
+  SimpleElement
 } from '../src'
 
 it('mounts a component', () => {
@@ -162,11 +163,13 @@ it('interferes prop types from default values', () => {
 
 it(`exposes the component function's return value`, () => {
   const exposed = { test: () => {} }
-  registerComponent('test', () => exposed)
+  const component = registerComponent('test', () => exposed)
 
   document.body.innerHTML = `
     <div data-simple-component="test" id="my-id"></div>
   `
+
   mountComponents(document.body)
-  expect(document.getElementById('my-id')?.$component).toBe(exposed)
+  const el = document.getElementById('my-id') as SimpleElement<typeof component>
+  expect(el.$component).toBe(exposed)
 })