@@ -22,17 +22,19 @@ import { assert, beforeEach, describe, expect, test, vi } from 'vitest';
2222import { DashboardExtension } from './dashboard-extension' ;
2323import { vol } from 'memfs' ;
2424
25- import type { ContextsManager } from './manager/contexts-manager' ;
26- import type { ContextsStatesDispatcher } from './manager/contexts-states-dispatcher' ;
25+ import { ContextsManager } from './manager/contexts-manager' ;
26+ import { ContextsStatesDispatcher } from './manager/contexts-states-dispatcher' ;
2727
2828let extensionContextMock : ExtensionContext ;
2929let dashboardExtension : DashboardExtension ;
3030let contextsManagerMock : ContextsManager ;
31- let contextsStatesDispatcher : ContextsStatesDispatcher ;
31+ let contextsStatesDispatcherMock : ContextsStatesDispatcher ;
3232
3333vi . mock ( import ( 'node:fs' ) ) ;
3434vi . mock ( import ( 'node:fs/promises' ) ) ;
3535vi . mock ( import ( '@kubernetes/client-node' ) ) ;
36+ vi . mock ( import ( './manager/contexts-manager' ) ) ;
37+ vi . mock ( import ( './manager/contexts-states-dispatcher' ) ) ;
3638
3739beforeEach ( ( ) => {
3840 vi . restoreAllMocks ( ) ;
@@ -50,14 +52,19 @@ beforeEach(() => {
5052 extensionContextMock = {
5153 subscriptions : [ ] ,
5254 } as unknown as ExtensionContext ;
55+
5356 // Create a mock for the contextsManager
5457 contextsManagerMock = {
5558 update : vi . fn ( ) ,
5659 } as unknown as ContextsManager ;
57- contextsStatesDispatcher = {
60+ vi . mocked ( ContextsManager ) . mockReturnValue ( contextsManagerMock ) ;
61+
62+ contextsStatesDispatcherMock = {
5863 init : vi . fn ( ) ,
5964 } as unknown as ContextsStatesDispatcher ;
60- dashboardExtension = new DashboardExtension ( extensionContextMock , contextsManagerMock , contextsStatesDispatcher ) ;
65+ vi . mocked ( ContextsStatesDispatcher ) . mockReturnValue ( contextsStatesDispatcherMock ) ;
66+
67+ dashboardExtension = new DashboardExtension ( extensionContextMock ) ;
6168 vi . mocked ( kubernetes . getKubeconfig ) . mockReturnValue ( {
6269 path : '/path/to/kube/config' ,
6370 } as Uri ) ;
@@ -80,7 +87,7 @@ describe('a kubeconfig file is not present', () => {
8087 callback ( { type : 'UPDATE' , location : { path : '/path/to/kube/config' } as Uri } ) ;
8188 expect ( contextsManagerMock . update ) . toHaveBeenCalledOnce ( ) ;
8289
83- expect ( contextsStatesDispatcher . init ) . toHaveBeenCalledOnce ( ) ;
90+ expect ( contextsStatesDispatcherMock . init ) . toHaveBeenCalledOnce ( ) ;
8491 } ) ;
8592
8693 test ( 'should deactivate correctly' , async ( ) => {
@@ -107,7 +114,7 @@ describe('a kubeconfig file is present', () => {
107114 callback ( { type : 'UPDATE' , location : { path : '/path/to/kube/config' } as Uri } ) ;
108115 expect ( contextsManagerMock . update ) . toHaveBeenCalledOnce ( ) ;
109116
110- expect ( contextsStatesDispatcher . init ) . toHaveBeenCalledOnce ( ) ;
117+ expect ( contextsStatesDispatcherMock . init ) . toHaveBeenCalledOnce ( ) ;
111118 } ) ;
112119
113120 test ( 'should deactivate correctly' , async ( ) => {
0 commit comments