diff --git a/CHANGELOG.md b/CHANGELOG.md
index 444195c..9908176 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,6 @@
+## 6.0.2
+* Added missing mock functions
+
 ## 6.0.2
 * Packages update
 * Vulnerabilities patched
diff --git a/package-lock.json b/package-lock.json
index a57f03c..91cd8bd 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "powerbi-visuals-utils-testutils",
-  "version": "6.0.2",
+  "version": "6.0.3",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "powerbi-visuals-utils-testutils",
-      "version": "6.0.2",
+      "version": "6.0.3",
       "license": "MIT",
       "dependencies": {
         "d3-array": "3.2.4",
diff --git a/package.json b/package.json
index b341591..d5e8455 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "powerbi-visuals-utils-testutils",
-  "version": "6.0.2",
+  "version": "6.0.3",
   "description": "powerbi-visuals-utils-testutils",
   "main": "lib/index.js",
   "module": "lib/index.js",
diff --git a/src/mocks/mockVisualHost.ts b/src/mocks/mockVisualHost.ts
index ab487a6..39e5de5 100644
--- a/src/mocks/mockVisualHost.ts
+++ b/src/mocks/mockVisualHost.ts
@@ -1,3 +1,4 @@
+/* eslint-disable @typescript-eslint/no-empty-function, @typescript-eslint/no-unused-vars*/
 /*
  *  Power BI Visualizations
  *
@@ -48,6 +49,8 @@ import IWebAccessService = powerbi.extensibility.IWebAccessService;
 
 // powerbi.extensibility.visual
 import IVisualHost = powerbi.extensibility.visual.IVisualHost;
+import ModalDialogResult = powerbi.extensibility.visual.ModalDialogResult
+import DialogOpenOptions = powerbi.extensibility.visual.DialogOpenOptions
 import CustomVisualApplyCustomSortArgs = powerbi.extensibility.visual.CustomVisualApplyCustomSortArgs;
 import IAcquireAADTokenService = powerbi.extensibility.IAcquireAADTokenService;
 
@@ -66,9 +69,8 @@ export class MockIVisualHost implements IVisualHost {
     public licenseManager: IVisualLicenseManager;
     public webAccessService: IWebAccessService;
     public acquireAADTokenService: IAcquireAADTokenService;
-    public drill: (args: DrillArgs) => void;
-    public setCanDrill: (drillAllowed: boolean) => void;
-    public applyCustomSort: (args: CustomVisualApplyCustomSortArgs) => void;
+    public modalDialogResult: ModalDialogResult;
+    public hostEnv: powerbi.common.CustomVisualHostEnv = 1;
 
     constructor(
         colorPalette?: IColorPalette,
@@ -84,8 +86,9 @@ export class MockIVisualHost implements IVisualHost {
         downloadService?: IDownloadService,
         licenseManager?: IVisualLicenseManager,
         webAccessService?: IWebAccessService,
-        acquireAADTokenService?: IAcquireAADTokenService
-        ) {
+        acquireAADTokenService?: IAcquireAADTokenService,
+        modalDialogResult?: powerbi.extensibility.visual.ModalDialogResult
+    ) {
 
         this.colorPaletteInstance = colorPalette;
         this.selectionManager = selectionManager;
@@ -101,6 +104,7 @@ export class MockIVisualHost implements IVisualHost {
         this.licenseManager = licenseManager;
         this.webAccessService = webAccessService;
         this.acquireAADTokenService = acquireAADTokenService;
+        this.modalDialogResult = modalDialogResult;
     }
 
     public createSelectionIdBuilder(): ISelectionIdBuilder {
@@ -123,10 +127,7 @@ export class MockIVisualHost implements IVisualHost {
         this.localeInstance.locale = language;
     }
 
-    // eslint-disable-next-line @typescript-eslint/no-empty-function
-    public applyJsonFilter(filter: powerbi.IFilter, objectName: string, propertyName: string, action: powerbi.FilterAction) {
-
-    }
+    public applyJsonFilter(filter: powerbi.IFilter, objectName: string, propertyName: string, action: powerbi.FilterAction) {}
 
     public get telemetry() {
         return this.telemetryService;
@@ -136,7 +137,6 @@ export class MockIVisualHost implements IVisualHost {
         return this.authenticationService;
     }
 
-    // eslint-disable-next-line @typescript-eslint/no-empty-function
     public persistProperties(changes: VisualObjectInstancesToPersist) { }
 
     public get tooltipService(): ITooltipService {
@@ -163,21 +163,27 @@ export class MockIVisualHost implements IVisualHost {
         return true;
     }
 
-    // eslint-disable-next-line @typescript-eslint/no-empty-function
-    public refreshHostData() {
-    }
+    public refreshHostData() {}
 
     public createLocalizationManager(): powerbi.extensibility.ILocalizationManager {
         return {
             getDisplayName: (key: string) => ""
         };
     }
+    
+    public drill(args: DrillArgs): void {}
 
-    public switchFocusModeState: (on: boolean) => void;
+    public setCanDrill(drillAllowed: boolean): void {}
 
-    public hostEnv: powerbi.common.CustomVisualHostEnv = 1;
+    public applyCustomSort(args: CustomVisualApplyCustomSortArgs): void {}
+
+    public switchFocusModeState(on: boolean): void {}
 
-    public displayWarningIcon: (hoverText: string, detailedText: string) => void;
+    public displayWarningIcon(hoverText: string, detailedText: string): void {}
 
-    public openModalDialog: (dialogId: string, options?: powerbi.extensibility.visual.DialogOpenOptions, initialState?: object) => powerbi.IPromise<powerbi.extensibility.visual.ModalDialogResult>;
+    public openModalDialog(dialogId: string, options?: DialogOpenOptions, initialState?: object): powerbi.IPromise<ModalDialogResult> {
+        return new Promise<ModalDialogResult>((resolve, rejects) => {
+            resolve(this.modalDialogResult)
+        }) as any;
+    }
 }
diff --git a/src/mocks/mocks.ts b/src/mocks/mocks.ts
index c9cdad3..50dc064 100644
--- a/src/mocks/mocks.ts
+++ b/src/mocks/mocks.ts
@@ -59,10 +59,11 @@ import HostCapabilities = powerbi.extensibility.HostCapabilities;
 import IVisualLicenseManager = powerbi.extensibility.IVisualLicenseManager;
 import IWebAccessService = powerbi.extensibility.IWebAccessService;
 import IAcquireAADTokenService = powerbi.extensibility.IAcquireAADTokenService;
+import ModalDialogResult = powerbi.extensibility.visual.ModalDialogResult
 import {MockIAcquireAADTokenService} from "./mockIAcquireAADTokenService";
 
 // eslint-disable-next-line @typescript-eslint/ban-types
-export function createVisualHost(locale?: Object, allowInteractions?: boolean, colors?: IColorInfo[], isEnabled?: boolean, displayNames?: any, token?: string): IVisualHost {
+export function createVisualHost(locale?: Object, allowInteractions?: boolean, colors?: IColorInfo[], isEnabled?: boolean, displayNames?: any, token?: string, modalDialogResult?: ModalDialogResult): IVisualHost {
     return new MockIVisualHost(
         createColorPalette(colors),
         createSelectionManager(),
@@ -77,7 +78,8 @@ export function createVisualHost(locale?: Object, allowInteractions?: boolean, c
         createDownloadService(),
         licenseManager(),
         webAccessService(),
-        acquireAADTokenService()
+        acquireAADTokenService(),
+        modalDialogResult
     )
 }