fix tests

Azure · Jun 24, 2022 · 2aa5cfe · 2aa5cfe
1 parent 43da21a
commit 2aa5cfe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/test/eventHandlers/InvocationHandler.test.ts b/test/eventHandlers/InvocationHandler.test.ts
@@ -11,6 +11,8 @@ import * as sinon from 'sinon';
 import { AzureFunctionsRpcMessages as rpc } from '../../azure-functions-language-worker-protobuf/src/rpc';
 import { FunctionInfo } from '../../src/FunctionInfo';
 import { FunctionLoader } from '../../src/FunctionLoader';
+import { WorkerChannel } from '../../src/WorkerChannel';
+import { Msg as AppStartMsg } from '../startApp.test';
 import { beforeEventHandlerSuite } from './beforeEventHandlerSuite';
 import { TestEventStream } from './TestEventStream';
 import { Msg as WorkerInitMsg } from './WorkerInitHandler.test';
@@ -323,16 +325,19 @@ namespace InputData {
 describe('InvocationHandler', () => {
     let stream: TestEventStream;
     let loader: sinon.SinonStubbedInstance<FunctionLoader>;
+    let channel: WorkerChannel;
     let coreApi: typeof coreTypes;
     let testDisposables: coreTypes.Disposable[] = [];
 
     before(async () => {
-        ({ stream, loader } = beforeEventHandlerSuite());
+        ({ stream, loader, channel } = beforeEventHandlerSuite());
         coreApi = await import('@azure/functions-core');
     });
 
     beforeEach(async () => {
         hookData = '';
+        channel.appHookData = {};
+        channel.appLevelOnlyHookData = {};
     });
 
     afterEach(async () => {
@@ -779,8 +784,8 @@ describe('InvocationHandler', () => {
         await stream.assertCalledWith(
             WorkerInitMsg.receivedInitLog,
             WorkerInitMsg.warning('Worker failed to load package.json: file does not exist'),
-            Msg.executingHooksLog(1, 'appStart'),
-            Msg.executedHooksLog('appStart'),
+            AppStartMsg.executingHooksLog(1, 'appStart'),
+            AppStartMsg.executedHooksLog('appStart'),
             WorkerInitMsg.response
         );
         expect(startFunc.callCount).to.be.equal(1);
@@ -832,8 +837,8 @@ describe('InvocationHandler', () => {
         await stream.assertCalledWith(
             WorkerInitMsg.receivedInitLog,
             WorkerInitMsg.warning('Worker failed to load package.json: file does not exist'),
-            Msg.executingHooksLog(1, 'appStart'),
-            Msg.executedHooksLog('appStart'),
+            AppStartMsg.executingHooksLog(1, 'appStart'),
+            AppStartMsg.executedHooksLog('appStart'),
             WorkerInitMsg.response
         );
         expect(startFunc.callCount).to.be.equal(1);

diff --git a/test/startApp.test.ts b/test/startApp.test.ts
@@ -13,7 +13,7 @@ import { Msg as WorkerInitMsg } from './eventHandlers/WorkerInitHandler.test';
 import LogCategory = rpc.RpcLog.RpcLogCategory;
 import LogLevel = rpc.RpcLog.Level;
 
-namespace Msg {
+export namespace Msg {
     export function executingHooksLog(count: number, hookName: string): rpc.IStreamingMessage {
         return {
             rpcLog: {
@@ -62,6 +62,8 @@ describe('startApp', () => {
         coreApi.Disposable.from(...testDisposables).dispose();
         testDisposables = [];
         process.chdir(originalCwd);
+        channel.appHookData = {};
+        channel.appLevelOnlyHookData = {};
     });
 
     it('runs app start hooks in non-specialization scenario', async () => {