Merge 23285bb into 1b1c286

stonelasley · Oct 29, 2018 · 84164e6 · 84164e6
2 parents 1b1c286 + 23285bb
commit 84164e6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/angular/modal.ts b/src/angular/modal.ts
@@ -1,7 +1,7 @@
 export class ModalMock {
     public static instance(): any {
         let _dismissCallback: Function;
-        let instance = jasmine.createSpyObj('Modal', ['present', 'dismiss', 'onDidDismiss']);
+        let instance = jasmine.createSpyObj('Modal', ['present', 'dismiss', 'onDidDismiss', 'onWillDismiss']);
         instance.present.and.returnValue(Promise.resolve());
 
         instance.dismiss.and.callFake(() => {
@@ -12,6 +12,10 @@ export class ModalMock {
         instance.onDidDismiss.and.callFake((callback: Function) => {
             _dismissCallback = callback;
         });
+
+        instance.onWillDismiss.and.callFake((callback: Function) => {
+            _dismissCallback = callback;
+        });
 
         return instance;
     }