front_end/entrypoints/main/ExecutionContextSelector.test.ts

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import * as SDK from '../../core/sdk/sdk.js';
import {
  createTarget,
} from '../../testing/EnvironmentHelpers.js';
import {
  describeWithMockConnection,
} from '../../testing/MockConnection.js';
import {
  getMainFrame,
} from '../../testing/ResourceTreeHelpers.js';
import * as UI from '../../ui/legacy/legacy.js';

import * as Main from './main.js';

describeWithMockConnection('ExecutionContextSelector', () => {
  it('switches to the default context once available', () => {
    new Main.ExecutionContextSelector.ExecutionContextSelector(
        SDK.TargetManager.TargetManager.instance(), UI.Context.Context.instance());

    const tabTarget = createTarget({type: SDK.Target.Type.TAB});
    const mainFrameTarget = createTarget({type: SDK.Target.Type.FRAME, parentTarget: tabTarget});
    const subframeTarget = createTarget({type: SDK.Target.Type.FRAME, parentTarget: mainFrameTarget});
    const prerenderTarget = createTarget({type: SDK.Target.Type.FRAME, parentTarget: tabTarget, subtype: 'prerender'});
    const serviceWorkerTarget = createTarget({type: SDK.Target.Type.ServiceWorker});
    const workerTarget = createTarget({type: SDK.Target.Type.Worker});

    const contextSetFlavor = sinon.spy(UI.Context.Context.instance(), 'setFlavor');

    const sentExecutionContextCreated = (target: SDK.Target.Target, includeFrameId = true) => {
      const frameId = includeFrameId ? getMainFrame(target).id : undefined;

      const runtimeModel = target.model(SDK.RuntimeModel.RuntimeModel);
      runtimeModel!.dispatchEventToListeners(
          SDK.RuntimeModel.Events.ExecutionContextCreated,
          {isDefault: true, frameId, target: () => target} as SDK.RuntimeModel.ExecutionContext);
    };

    sentExecutionContextCreated(subframeTarget);
    sinon.assert.called(contextSetFlavor);

    contextSetFlavor.resetHistory();
    sentExecutionContextCreated(subframeTarget);
    sinon.assert.notCalled(contextSetFlavor);

    sentExecutionContextCreated(mainFrameTarget);
    sinon.assert.called(contextSetFlavor);

    contextSetFlavor.resetHistory();
    sentExecutionContextCreated(prerenderTarget);
    sinon.assert.notCalled(contextSetFlavor);

    contextSetFlavor.resetHistory();
    sentExecutionContextCreated(serviceWorkerTarget, /* includeFrameId */ false);
    sinon.assert.notCalled(contextSetFlavor);

    contextSetFlavor.resetHistory();
    sentExecutionContextCreated(workerTarget, /* includeFrameId */ false);
    sinon.assert.notCalled(contextSetFlavor);
  });
});