Browse Source

Add some async message hooks in the standard tests.

Jason Grout 6 years ago
parent
commit
bdcf02f1b4

+ 3 - 3
packages/services/test/src/kernel/ifuture.spec.ts

@@ -68,8 +68,8 @@ describe('Kernel.IFuture', () => {
           tester.send(msgDone);
         };
 
-        future.registerMessageHook((msg) => {
-          calls.push('last');
+        future.registerMessageHook(async (msg) => {
+          await calls.push('last');
           return true;
         });
 
@@ -136,7 +136,7 @@ describe('Kernel.IFuture', () => {
           return true;
         });
 
-        future.registerMessageHook((msg) => {
+        future.registerMessageHook(async (msg) => {
           calls.push('first');
           return false;
         });

+ 5 - 4
packages/services/test/src/kernel/ikernel.spec.ts

@@ -996,8 +996,9 @@ describe('Kernel.IKernel', () => {
           tester.send(msgDone);
         };
 
-        kernel.registerMessageHook(parentHeader.msg_id, (msg) => {
-          calls.push('last');
+        kernel.registerMessageHook(parentHeader.msg_id, async (msg) => {
+          // Make this hook call asynchronous
+          await calls.push('last');
           return true;
         });
 
@@ -1065,8 +1066,8 @@ describe('Kernel.IKernel', () => {
           return false;
         });
 
-        future.onIOPub = () => {
-          calls.push('iopub');
+        future.onIOPub = async () => {
+          await calls.push('iopub');
         };
       });