11import { vi } from "vitest" ;
22import $ from "jquery" ;
3- import { ElementWithUtils } from "../src/ts/utils/dom" ;
3+ import { ElementsWithUtils , ElementWithUtils } from "../src/ts/utils/dom" ;
44
55//@ts -expect-error add to global
66global [ "$" ] = $ ;
@@ -20,7 +20,7 @@ vi.mock("../src/ts/firebase", () => ({
2020 isAuthenticated : ( ) => false ,
2121} ) ) ;
2222
23- vi . mock ( "../src/ts/utils/dom" , ( ) => {
23+ vi . mock ( "../src/ts/utils/dom" , async ( importOriginal ) => {
2424 const createMockElement = ( ) : ElementWithUtils => {
2525 return {
2626 disable : vi . fn ( ) . mockReturnThis ( ) ,
@@ -45,9 +45,9 @@ vi.mock("../src/ts/utils/dom", () => {
4545 setStyle : vi . fn ( ) . mockReturnThis ( ) ,
4646 getStyle : vi . fn ( ) . mockReturnValue ( { } ) ,
4747 isFocused : vi . fn ( ) . mockReturnValue ( false ) ,
48- qs : vi . fn ( ) . mockReturnValue ( null ) ,
48+ qs : vi . fn ( ) . mockImplementation ( ( ) => createMockElement ( ) ) ,
4949 qsr : vi . fn ( ) . mockImplementation ( ( ) => createMockElement ( ) ) ,
50- qsa : vi . fn ( ) . mockReturnValue ( [ ] ) ,
50+ qsa : vi . fn ( ) . mockImplementation ( ( ) => new ElementsWithUtils ( ) ) ,
5151 empty : vi . fn ( ) . mockReturnThis ( ) ,
5252 appendHtml : vi . fn ( ) . mockReturnThis ( ) ,
5353 append : vi . fn ( ) . mockReturnThis ( ) ,
@@ -71,10 +71,14 @@ vi.mock("../src/ts/utils/dom", () => {
7171 } ;
7272 } ;
7373
74+ const actual = await importOriginal ( ) ;
75+
7476 return {
77+ //@ts -expect-error - mocking private method
78+ ...actual ,
7579 qsr : vi . fn ( ) . mockImplementation ( ( ) => createMockElement ( ) ) ,
7680 qs : vi . fn ( ) . mockImplementation ( ( ) => createMockElement ( ) ) ,
77- qsa : vi . fn ( ) . mockReturnValue ( [ ] ) ,
81+ qsa : vi . fn ( ) . mockImplementation ( ( ) => new ElementsWithUtils ( ) ) ,
7882 } ;
7983} ) ;
8084
0 commit comments