chrome-devtools-frontend 1.0.1662289 → 1.0.1664496

package/front_end/generated/protocol.ts

@@ -943,7 +943,6 @@ export namespace Audits {
   }
 
   export const enum MixedContentResourceType {
-    AttributionSrc = 'AttributionSrc',
     Audio = 'Audio',
     Beacon = 'Beacon',
     CSPReport = 'CSPReport',
@@ -2507,11 +2506,6 @@ export namespace Browser {
     buckets: Bucket[];
   }
 
-  export const enum PrivacySandboxAPI {
-    BiddingAndAuctionServices = 'BiddingAndAuctionServices',
-    TrustedKeyValue = 'TrustedKeyValue',
-  }
-
   export interface SetPermissionRequest {
     /**
      * Descriptor of permission to override.
@@ -2743,17 +2737,6 @@ export namespace Browser {
     url: string;
   }
 
-  export interface AddPrivacySandboxCoordinatorKeyConfigRequest {
-    api: PrivacySandboxAPI;
-    coordinatorOrigin: string;
-    keyConfig: string;
-    /**
-     * BrowserContext to perform the action in. When omitted, default browser
-     * context is used.
-     */
-    browserContextId?: BrowserContextID;
-  }
-
   /**
    * Fired when page is about to start a download.
    */
@@ -18317,54 +18300,12 @@ export namespace Storage {
     count: number;
   }
 
-  /**
-   * Protected audience interest group auction identifier.
-   */
-  export type InterestGroupAuctionId = OpaqueIdentifier<string, 'Protocol.Storage.InterestGroupAuctionId'>;
-
-  /**
-   * Enum of interest group access types.
-   */
-  export const enum InterestGroupAccessType {
-    Join = 'join',
-    Leave = 'leave',
-    Update = 'update',
-    Loaded = 'loaded',
-    Bid = 'bid',
-    Win = 'win',
-    AdditionalBid = 'additionalBid',
-    AdditionalBidWin = 'additionalBidWin',
-    TopLevelBid = 'topLevelBid',
-    TopLevelAdditionalBid = 'topLevelAdditionalBid',
-    Clear = 'clear',
-  }
-
-  /**
-   * Enum of auction events.
-   */
-  export const enum InterestGroupAuctionEventType {
-    Started = 'started',
-    ConfigResolved = 'configResolved',
-  }
-
-  /**
-   * Enum of network fetches auctions can do.
-   */
-  export const enum InterestGroupAuctionFetchType {
-    BidderJs = 'bidderJs',
-    BidderWasm = 'bidderWasm',
-    SellerJs = 'sellerJs',
-    BidderTrustedSignals = 'bidderTrustedSignals',
-    SellerTrustedSignals = 'sellerTrustedSignals',
-  }
-
   /**
    * Enum of shared storage access scopes.
    */
   export const enum SharedStorageAccessScope {
     Window = 'window',
     SharedStorageWorklet = 'sharedStorageWorklet',
-    ProtectedAudienceWorklet = 'protectedAudienceWorklet',
     Header = 'header',
   }
 
@@ -18797,29 +18738,6 @@ export namespace Storage {
     didDeleteTokens: boolean;
   }
 
-  export interface GetInterestGroupDetailsRequest {
-    ownerOrigin: string;
-    name: string;
-  }
-
-  export interface GetInterestGroupDetailsResponse extends ProtocolResponseWithError {
-    /**
-     * This largely corresponds to:
-     * https://wicg.github.io/turtledove/#dictdef-generatebidinterestgroup
-     * but has absolute expirationTime instead of relative lifetimeMs and
-     * also adds joiningOrigin.
-     */
-    details: any;
-  }
-
-  export interface SetInterestGroupTrackingRequest {
-    enable: boolean;
-  }
-
-  export interface SetInterestGroupAuctionTrackingRequest {
-    enable: boolean;
-  }
-
   export interface GetSharedStorageMetadataRequest {
     ownerOrigin: string;
   }
@@ -18881,12 +18799,6 @@ export namespace Storage {
     sets: RelatedWebsiteSet[];
   }
 
-  export interface SetProtectedAudienceKAnonymityRequest {
-    owner: string;
-    name: string;
-    hashes: binary[];
-  }
-
   /**
    * A cache's contents have been modified.
    */
@@ -18971,66 +18883,6 @@ export namespace Storage {
     bucketId: string;
   }
 
-  /**
-   * One of the interest groups was accessed. Note that these events are global
-   * to all targets sharing an interest group store.
-   */
-  export interface InterestGroupAccessedEvent {
-    accessTime: Network.TimeSinceEpoch;
-    type: InterestGroupAccessType;
-    ownerOrigin: string;
-    name: string;
-    /**
-     * For topLevelBid/topLevelAdditionalBid, and when appropriate,
-     * win and additionalBidWin
-     */
-    componentSellerOrigin?: string;
-    /**
-     * For bid or somethingBid event, if done locally and not on a server.
-     */
-    bid?: number;
-    bidCurrency?: string;
-    /**
-     * For non-global events --- links to interestGroupAuctionEvent
-     */
-    uniqueAuctionId?: InterestGroupAuctionId;
-  }
-
-  /**
-   * An auction involving interest groups is taking place. These events are
-   * target-specific.
-   */
-  export interface InterestGroupAuctionEventOccurredEvent {
-    eventTime: Network.TimeSinceEpoch;
-    type: InterestGroupAuctionEventType;
-    uniqueAuctionId: InterestGroupAuctionId;
-    /**
-     * Set for child auctions.
-     */
-    parentAuctionId?: InterestGroupAuctionId;
-    /**
-     * Set for started and configResolved
-     */
-    auctionConfig?: any;
-  }
-
-  /**
-   * Specifies which auctions a particular network fetch may be related to, and
-   * in what role. Note that it is not ordered with respect to
-   * Network.requestWillBeSent (but will happen before loadingFinished
-   * loadingFailed).
-   */
-  export interface InterestGroupAuctionNetworkRequestCreatedEvent {
-    type: InterestGroupAuctionFetchType;
-    requestId: Network.RequestId;
-    /**
-     * This is the set of the auctions using the worklet that issued this
-     * request.  In the case of trusted signals, it's possible that only some of
-     * them actually care about the keys being queried.
-     */
-    auctions: InterestGroupAuctionId[];
-  }
-
   /**
    * Shared storage was accessed by the associated page.
    * The following parameters are included in all events.

package/front_end/models/ai_assistance/AgentProject.ts

@@ -2,9 +2,9 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+import * as TextUtils from '../../core/text_utils/text_utils.js';
 import * as Diff from '../../third_party/diff/diff.js';
 import * as Persistence from '../persistence/persistence.js';
-import * as TextUtils from '../text_utils/text_utils.js';
 import type * as Workspace from '../workspace/workspace.js';
 
 import {debugLog} from './debug.js';

package/front_end/models/ai_assistance/AiAgent2.ts

@@ -3,7 +3,6 @@
 // found in the LICENSE file.
 
 import * as Host from '../../core/host/host.js';
-import * as SDK from '../../core/sdk/sdk.js';
 import type * as LHModel from '../lighthouse/lighthouse.js';
 
 import {
@@ -68,7 +67,7 @@ export class AiAgent2 extends AiAgent<unknown> {
   readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_DEVTOOLS_V2_AGENT;
   readonly userTier = 'TESTERS';
 
-  #changes = new ChangeManager();
+  #changes: ChangeManager;
   #execJs: typeof executeJsCode;
   readonly #allowedOrigin?: () => AllowedOriginResult;
   readonly #lighthouseRecording?:
@@ -83,6 +82,7 @@ export class AiAgent2 extends AiAgent<unknown> {
 
   constructor(opts: AiAgent2Options) {
     super(opts);
+    this.#changes = new ChangeManager(opts.targetManager);
     this.#lighthouseRecording = opts.lighthouseRecording;
     this.#execJs = opts.execJs ?? executeJsCode;
     this.#allowedOrigin = opts.allowedOrigin;
@@ -231,7 +231,7 @@ User query: ${enhancedQuery}`;
           createExtensionScope: this.#createExtensionScope.bind(this),
           execJs: this.#execJs,
           getExecutionContextNode: () => this.context instanceof DOMNodeContext ? this.context.getItem() : null,
-          getTarget: () => SDK.TargetManager.TargetManager.instance().primaryPageTarget(),
+          getTarget: () => this.targetManager.primaryPageTarget(),
           getEstablishedOrigin: () => this.#getConversationOrigin(),
           lighthouseRecording: this.#lighthouseRecording,
         };

package/front_end/models/ai_assistance/AiConversation.ts

@@ -72,6 +72,8 @@ export interface AiConversationOptions {
   onInspectElement?: () => Promise<SDK.DOMModel.DOMNode|null>;
   networkTimeCalculator?: NetworkTimeCalculator.NetworkTransferTimeCalculator;
   lighthouseRecording?: (overrides?: LHModel.RunTypes.RunOverrides) => Promise<LHModel.ReporterTypes.ReportJSON|null>;
+  aiHistoryStorage?: AiHistoryStorage;
+  targetManager?: SDK.TargetManager.TargetManager;
 }
 
 export class AiConversation {
@@ -110,6 +112,8 @@ export class AiConversation {
   #lighthouseRecording?: (overrides?: LHModel.RunTypes.RunOverrides) => Promise<LHModel.ReporterTypes.ReportJSON|null>;
   #onInspectElement?: () => Promise<SDK.DOMModel.DOMNode|null>;
   #networkTimeCalculator?: NetworkTimeCalculator.NetworkTransferTimeCalculator;
+  readonly #aiHistoryStorage: AiHistoryStorage;
+  readonly #targetManager: SDK.TargetManager.TargetManager;
 
   constructor(options: AiConversationOptions) {
     const {
@@ -123,6 +127,8 @@ export class AiConversation {
       onInspectElement,
       networkTimeCalculator,
       lighthouseRecording,
+      aiHistoryStorage = AiHistoryStorage.instance(),
+      targetManager = SDK.TargetManager.TargetManager.instance(),
     } = options;
     this.#changeManager = changeManager;
     this.#aidaClient = aidaClient;
@@ -130,6 +136,8 @@ export class AiConversation {
     this.#onInspectElement = onInspectElement;
     this.#networkTimeCalculator = networkTimeCalculator;
     this.#lighthouseRecording = lighthouseRecording;
+    this.#aiHistoryStorage = aiHistoryStorage;
+    this.#targetManager = targetManager;
 
     this.id = id;
     this.#isReadOnly = isReadOnly;
@@ -208,7 +216,7 @@ export class AiConversation {
   }
 
   #reconstructHistory(historyWithoutImages: ResponseData[]): ResponseData[] {
-    const imageHistory = AiHistoryStorage.instance().getImageHistory();
+    const imageHistory = this.#aiHistoryStorage.getImageHistory();
     if (imageHistory && imageHistory.length > 0) {
       const history: ResponseData[] = [];
       for (const data of historyWithoutImages) {
@@ -286,12 +294,12 @@ export class AiConversation {
 
   async addHistoryItem(item: ResponseData): Promise<void> {
     this.history.push(item);
-    await AiHistoryStorage.instance().upsertHistoryEntry(this.serialize());
+    await this.#aiHistoryStorage.upsertHistoryEntry(this.serialize());
     if (item.type === ResponseType.USER_QUERY) {
-      void AiHistoryStorage.instance().addRecentPrompt(item.query);
+      void this.#aiHistoryStorage.addRecentPrompt(item.query);
       if (item.imageId && item.imageInput && 'inlineData' in item.imageInput) {
         const inlineData = item.imageInput.inlineData;
-        await AiHistoryStorage.instance().upsertImage({
+        await this.#aiHistoryStorage.upsertImage({
           id: item.imageId,
           data: inlineData.data,
           mimeType: inlineData.mimeType,
@@ -361,6 +369,7 @@ export class AiConversation {
       lighthouseRecording: this.#lighthouseRecording,
       allowedOrigin: this.allowedOrigin,
       history,
+      targetManager: this.#targetManager,
     };
 
     this.#agent = Root.Runtime.hostConfig.devToolsAiV2Architecture?.enabled ? new AiAgent2(options) :
@@ -397,18 +406,18 @@ export class AiConversation {
           } = {},
           ): AsyncGenerator<ResponseData, void, void> {
     this.#navigationOccurredDuringRun = false;
-    const originAtRunStart = getPrimaryPageOrigin();
+    const originAtRunStart = getPrimaryPageOrigin(this.#targetManager);
     const listener = (): void => {
       // If an unexpected navigation to a different origin occurred
       // during processing the user's request, we don't want to allow
       // the agent to run any function calls and retrieve data from the new origin.
       // Performance agent and accessibility agent navigate to 'about://' or 'chrome://terms'
-      const newOrigin = getPrimaryPageOrigin();
+      const newOrigin = getPrimaryPageOrigin(this.#targetManager);
       if (originAtRunStart !== newOrigin && newOrigin && !ALLOWED_PAGE_NAVIGATIONS.includes(newOrigin)) {
         this.#navigationOccurredDuringRun = true;
       }
     };
-    const targetManager = SDK.TargetManager.TargetManager.instance();
+    const targetManager = this.#targetManager;
     targetManager.addModelListener(SDK.ResourceTreeModel.ResourceTreeModel,
                                    SDK.ResourceTreeModel.Events.PrimaryPageChanged, listener, this);
 
@@ -524,7 +533,7 @@ export class AiConversation {
     if (this.#origin) {
       return {origin: this.#origin};
     }
-    this.#origin = getPrimaryPageOrigin();
+    this.#origin = getPrimaryPageOrigin(this.#targetManager);
 
     return {origin: this.#origin};
   };
@@ -538,8 +547,10 @@ function isAiAssistanceContextSelectionAgentEnabled(): boolean {
   return Boolean(Root.Runtime.hostConfig.devToolsAiAssistanceContextSelectionAgent?.enabled);
 }
 
-function getPrimaryPageOrigin(): Platform.DevToolsPath.UrlString|undefined {
-  const target = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
+function getPrimaryPageOrigin(
+    targetManager: SDK.TargetManager.TargetManager,
+    ): Platform.DevToolsPath.UrlString|undefined {
+  const target = targetManager.primaryPageTarget();
   const inspectedURL = target?.inspectedURL();
   return inspectedURL ? new Common.ParsedURL.ParsedURL(inspectedURL).securityOrigin() : undefined;
 }

package/front_end/models/ai_assistance/AiHistoryStorage.ts

@@ -4,6 +4,7 @@
 // found in the LICENSE file.
 
 import * as Common from '../../core/common/common.js';
+import * as Root from '../../core/root/root.js';
 
 import {ResponseType, type SerializedResponseData} from './agents/AiAgent.js';
 
@@ -33,8 +34,6 @@ export interface SerializedImage {
   data: string;
 }
 
-let instance: AiHistoryStorage|null = null;
-
 const DEFAULT_MAX_STORAGE_SIZE = 50 * 1024 * 1024;
 export const MAX_RECENT_PROMPTS_COUNT = 20;
 export const MAX_CONVERSATIONS_COUNT = 50;
@@ -55,14 +54,17 @@ export class AiHistoryStorage extends Common.ObjectWrapper.ObjectWrapper<EventTy
   #mutex = new Common.Mutex.Mutex();
   #maxStorageSize: number;
 
-  constructor(maxStorageSize = DEFAULT_MAX_STORAGE_SIZE) {
+  constructor(
+      settings: Common.Settings.Settings = Common.Settings.Settings.instance(),
+      maxStorageSize = DEFAULT_MAX_STORAGE_SIZE,
+  ) {
     super();
-    this.#historySetting = Common.Settings.Settings.instance().createSetting('ai-assistance-history-entries', []);
-    this.#imageHistorySettings = Common.Settings.Settings.instance().createSetting(
+    this.#historySetting = settings.createSetting('ai-assistance-history-entries', []);
+    this.#imageHistorySettings = settings.createSetting(
         'ai-assistance-history-images',
         [],
     );
-    this.#recentPromptsSetting = Common.Settings.Settings.instance().createSetting('ai-assistance-recent-prompts', []);
+    this.#recentPromptsSetting = settings.createSetting('ai-assistance-recent-prompts', []);
     this.#maxStorageSize = maxStorageSize;
   }
 
@@ -225,12 +227,21 @@ export class AiHistoryStorage extends Common.ObjectWrapper.ObjectWrapper<EventTy
       opts: {
         forceNew: boolean,
         maxStorageSize?: number,
+        settings?: Common.Settings.Settings,
       } = {forceNew: false, maxStorageSize: DEFAULT_MAX_STORAGE_SIZE},
       ): AiHistoryStorage {
-    const {forceNew, maxStorageSize} = opts;
-    if (!instance || forceNew) {
-      instance = new AiHistoryStorage(maxStorageSize);
+    const {forceNew, maxStorageSize, settings} = opts;
+    if (!Root.DevToolsContext.globalInstance().has(AiHistoryStorage) || forceNew) {
+      Root.DevToolsContext.globalInstance().set(AiHistoryStorage,
+                                                new AiHistoryStorage(
+                                                    settings ?? Common.Settings.Settings.instance(),
+                                                    maxStorageSize,
+                                                    ));
     }
-    return instance;
+    return Root.DevToolsContext.globalInstance().get(AiHistoryStorage);
+  }
+
+  static removeInstance(): void {
+    Root.DevToolsContext.globalInstance().delete(AiHistoryStorage);
   }
 }

package/front_end/models/ai_assistance/ChangeManager.ts

@@ -36,8 +36,8 @@ export class ChangeManager {
   readonly #stylesheetChanges = new Map<Protocol.DOM.StyleSheetId, Change[]>();
   readonly #backupStylesheetChanges = new Map<Protocol.DOM.StyleSheetId, Change[]>();
 
-  constructor() {
-    SDK.TargetManager.TargetManager.instance().addModelListener(
+  constructor(targetManager: SDK.TargetManager.TargetManager = SDK.TargetManager.TargetManager.instance()) {
+    targetManager.addModelListener(
         SDK.ResourceTreeModel.ResourceTreeModel,
         SDK.ResourceTreeModel.Events.PrimaryPageChanged,
         this.clear,

package/front_end/models/ai_assistance/ExtensionScope.ts

@@ -237,7 +237,11 @@ export class ExtensionScope {
     return node.localName() || node.nodeName().toLowerCase();
   }
 
-  static getSourceLocation(styleRule: SDK.CSSRule.CSSStyleRule): string|undefined {
+  static getSourceLocation(
+      styleRule: SDK.CSSRule.CSSStyleRule,
+      cssWorkspaceBinding: Bindings.CSSWorkspaceBinding.CSSWorkspaceBinding =
+          Bindings.CSSWorkspaceBinding.CSSWorkspaceBinding.instance(),
+      ): string|undefined {
     const styleSheetHeader = styleRule.header;
     if (!styleSheetHeader) {
       return;
@@ -250,7 +254,7 @@ export class ExtensionScope {
     const lineNumber = styleSheetHeader.lineNumberInSource(range.startLine);
     const columnNumber = styleSheetHeader.columnNumberInSource(range.startLine, range.startColumn);
     const location = new SDK.CSSModel.CSSLocation(styleSheetHeader, lineNumber, columnNumber);
-    const uiLocation = Bindings.CSSWorkspaceBinding.CSSWorkspaceBinding.instance().rawLocationToUILocation(location);
+    const uiLocation = cssWorkspaceBinding.rawLocationToUILocation(location);
     return uiLocation?.linkText(/* skipTrim= */ true, /* showColumnNumber= */ true);
   }
 

package/front_end/models/ai_assistance/agents/AccessibilityAgent.ts

@@ -97,7 +97,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
   constructor(opts: ExecuteJsAgentOptions) {
     super(opts);
     this.#lighthouseRecording = opts.lighthouseRecording;
-    this.#changes = opts.changeManager || new ChangeManager();
+    this.#changes = opts.changeManager || new ChangeManager(opts.targetManager);
     this.#execJs = opts.execJs ?? executeJsCode;
     this.#createExtensionScope = opts.createExtensionScope ?? ((changes: ChangeManager) => {
                                    return new ExtensionScope(changes, this.sessionId, this.#getDocumentBodyNode());
@@ -125,7 +125,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
   }
 
   protected override async preRun(): Promise<void> {
-    const target = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
+    const target = this.targetManager.primaryPageTarget();
     const domModel = target?.model(SDK.DOMModel.DOMModel);
     // We need to ensure the document is requested so that #getDocumentBodyNode()
     // can return a valid node for the JavaScript execution context.
@@ -144,10 +144,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
    * so that the AI has a valid $0 to start with.
    */
   #getDocumentBodyNode(): SDK.DOMModel.DOMNode|null {
-    const document = SDK.TargetManager.TargetManager.instance()
-                         .primaryPageTarget()
-                         ?.model(SDK.DOMModel.DOMModel)
-                         ?.existingDocument();
+    const document = this.targetManager.primaryPageTarget()?.model(SDK.DOMModel.DOMModel)?.existingDocument();
     return document?.body ?? document ?? null;
   }
 
@@ -168,7 +165,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
   }
 
   async #resolvePathToNode(path: string): Promise<SDK.DOMModel.DOMNode|null> {
-    const target = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
+    const target = this.targetManager.primaryPageTarget();
     if (!target) {
       return null;
     }

package/front_end/models/ai_assistance/agents/AiAgent.ts

@@ -6,10 +6,10 @@ import * as Host from '../../../core/host/host.js';
 import type {UrlString} from '../../../core/platform/DevToolsPath.js';
 import type * as Platform from '../../../core/platform/platform.js';
 import * as Root from '../../../core/root/root.js';
-import type * as SDK from '../../../core/sdk/sdk.js';
+import * as SDK from '../../../core/sdk/sdk.js';
+import type * as TextUtils from '../../../core/text_utils/text_utils.js';
 import type * as Protocol from '../../../generated/protocol.js';
 import type * as LHModel from '../../lighthouse/lighthouse.js';
-import type * as TextUtils from '../../text_utils/text_utils.js';
 import type * as Trace from '../../trace/trace.js';
 import type * as Workspace from '../../workspace/workspace.js';
 import {areOriginsEquivalent, extractContextOrigin, isOpaqueOrigin} from '../AiOrigins.js';
@@ -166,6 +166,7 @@ export interface AgentOptions {
   history?: Host.AidaClient.Content[];
   allowedOrigin?: () => AllowedOriginResult;
   lighthouseRecording?: (overrides?: LHModel.RunTypes.RunOverrides) => Promise<LHModel.ReporterTypes.ReportJSON|null>;
+  targetManager?: SDK.TargetManager.TargetManager;
 }
 
 export interface ParsedAnswer {
@@ -485,6 +486,7 @@ export abstract class AiAgent<T> {
   readonly confirmSideEffect: typeof Promise.withResolvers;
   readonly #functionDeclarations = new Map<string, FunctionDeclaration<Record<string, unknown>, unknown>>();
   readonly #allowedOrigin?: () => AllowedOriginResult;
+  readonly #targetManager: SDK.TargetManager.TargetManager;
 
   /**
    * Used in the debug mode and evals.
@@ -518,6 +520,7 @@ export abstract class AiAgent<T> {
     this.confirmSideEffect = opts.confirmSideEffectForTest ?? (() => Promise.withResolvers());
     this.#history = opts.history ?? [];
     this.#allowedOrigin = opts.allowedOrigin;
+    this.#targetManager = opts.targetManager ?? SDK.TargetManager.TargetManager.instance();
   }
 
   async enhanceQuery(query: string, selected: ConversationContext<T>|null,
@@ -534,6 +537,10 @@ export abstract class AiAgent<T> {
     return [...this.#history];
   }
 
+  get targetManager(): SDK.TargetManager.TargetManager {
+    return this.#targetManager;
+  }
+
   /**
    * Add a fact which will be sent for any subsequent requests.
    * Returns the new list of all facts.

package/front_end/models/ai_assistance/agents/ContextSelectionAgent.ts

@@ -74,6 +74,14 @@ Your role is to understand the user's query, identify the appropriate specialize
 * The only available types are \`#req\` for network request and \`#file\` for source files. Only use ID inside the link, never ask about user selecting by ID.
 `;
 
+export interface ContextSelectionAgentOptions extends AgentOptions {
+  performanceRecordAndReload?: () => Promise<Trace.TraceModel.ParsedTrace>;
+  onInspectElement?: () => Promise<SDK.DOMModel.DOMNode|null>;
+  networkTimeCalculator?: NetworkTimeCalculator.NetworkTransferTimeCalculator;
+  networkLog?: Logs.NetworkLog.NetworkLog;
+  workspace?: Workspace.Workspace.WorkspaceImpl;
+}
+
 /**
  * One agent instance handles one conversation. Create a new agent
  * instance for a new conversation.
@@ -101,13 +109,13 @@ export class ContextSelectionAgent extends AiAgent<never> {
   readonly #lighthouseRecording?:
       (overrides?: LHModel.RunTypes.RunOverrides) => Promise<LHModel.ReporterTypes.ReportJSON|null>;
   #allowedOrigin: () => AllowedOriginResult;
+  readonly #networkLog: Logs.NetworkLog.NetworkLog;
+  readonly #workspace: Workspace.Workspace.WorkspaceImpl;
 
-  constructor(opts: AgentOptions&{
-    performanceRecordAndReload?: () => Promise<Trace.TraceModel.ParsedTrace>,
-    onInspectElement?: () => Promise<SDK.DOMModel.DOMNode|null>,
-    networkTimeCalculator?: NetworkTimeCalculator.NetworkTransferTimeCalculator,
-  }) {
+  constructor(opts: ContextSelectionAgentOptions) {
     super(opts);
+    this.#networkLog = opts.networkLog ?? Logs.NetworkLog.NetworkLog.instance();
+    this.#workspace = opts.workspace ?? Workspace.Workspace.WorkspaceImpl.instance();
     this.#performanceRecordAndReload = opts.performanceRecordAndReload;
     this.#lighthouseRecording = opts.lighthouseRecording;
     this.#onInspectElement = opts.onInspectElement;
@@ -146,7 +154,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
 
         let hasCrossOriginRequest = false;
         const requestsToShow: NetworkRequest[] = [];
-        for (const request of Logs.NetworkLog.NetworkLog.instance().requests()) {
+        for (const request of this.#networkLog.requests()) {
           const requestOrigin = getRequestContextOrigin(request);
           /**
            * NOTE: this origin check does not ensure that all the requests are
@@ -226,7 +234,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
             error: 'No request found',
           };
         }
-        const request = Logs.NetworkLog.NetworkLog.instance().requests().find(req => {
+        const request = this.#networkLog.requests().find(req => {
           if (req.requestId() !== id) {
             return false;
           }
@@ -281,7 +289,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
 
         const files: Array<{file: string, id: number | undefined}> = [];
         const uiSourceCodes: Workspace.UISourceCode.UISourceCode[] = [];
-        for (const file of ContextSelectionAgent.getUISourceCodes()) {
+        for (const file of ContextSelectionAgent.getUISourceCodes(this.#workspace)) {
           const fileUrl = file.url();
           const fileOrigin = Common.ParsedURL.ParsedURL.extractOrigin(fileUrl);
 
@@ -339,7 +347,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
         }
         const origin = allowedOriginResult.origin;
 
-        const file = ContextSelectionAgent.getUISourceCodes().find(file => {
+        const file = ContextSelectionAgent.getUISourceCodes(this.#workspace).find(file => {
           if (ContextSelectionAgent.uiSourceCodeId.get(file) !== params.id) {
             return false;
           }
@@ -554,8 +562,8 @@ export class ContextSelectionAgent extends AiAgent<never> {
    * coming from SourceMaps (usually only one) as that has simple code and
    * usually is what the user authored.
    */
-  static getUISourceCodes(): Workspace.UISourceCode.UISourceCode[] {
-    const workspace = Workspace.Workspace.WorkspaceImpl.instance();
+  static getUISourceCodes(workspace: Workspace.Workspace.WorkspaceImpl = Workspace.Workspace.WorkspaceImpl.instance()):
+      Workspace.UISourceCode.UISourceCode[] {
     const projects =
         workspace.projects().filter(project => project.type() === Workspace.Workspace.projectTypes.Network);
     const uiSourceCodes = new Map<string, Workspace.UISourceCode.UISourceCode>();