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

import type * as Trace from '../../models/trace/trace.js';
import {describeWithEnvironment} from '../../testing/EnvironmentHelpers.js';
import {TraceLoader} from '../../testing/TraceLoader.js';
import * as UI from '../../ui/legacy/legacy.js';

import * as Timeline from './timeline.js';
import * as Utils from './utils/utils.js';

describeWithEnvironment('TimelineTreeView', function() {
  describe('Third party tree', function() {
    it('does not select the first row by default', async function() {
      const {parsedTrace} = await TraceLoader.traceEngine(this, 'web-dev-with-commit.json.gz');
      const treeView = new Timeline.ThirdPartyTreeView.ThirdPartyTreeViewWidget();
      const mapper = new Utils.EntityMapper.EntityMapper(parsedTrace);
      const events = [...mapper.mappings().eventsByEntity.values()].flat();
      treeView.setModelWithEvents(events, parsedTrace, mapper);
      const sel: Timeline.TimelineSelection.TimeRangeSelection = {
        bounds: parsedTrace.Meta.traceBounds,
      };
      const box = new UI.Widget.VBox();
      treeView.show(box.element);
      treeView.updateContents(sel);
      assert.isNull(treeView.dataGrid.selectedNode);
    });
    it('includes 1p and extension badges', async function() {
      // This trace creates 2 nodes in the tree. One representing the first party entity, and one for
      // a chrome extension.
      const {parsedTrace} = await TraceLoader.traceEngine(this, 'extension-tracks-and-marks.json.gz');
      const treeView = new Timeline.ThirdPartyTreeView.ThirdPartyTreeViewWidget();
      const mapper = new Utils.EntityMapper.EntityMapper(parsedTrace);
      const events = [...mapper.mappings().eventsByEntity.values()].flat();

      treeView.setModelWithEvents(events, parsedTrace, mapper);
      const sel: Timeline.TimelineSelection.TimeRangeSelection = {
        bounds: parsedTrace.Meta.traceBounds,
      };
      treeView.updateContents(sel);
      const tree = treeView.buildTree() as Trace.Extras.TraceTree.BottomUpRootNode;
      const topNodesIterator = [...tree.children().values()].flat();

      // Node with ext
      const firstNode = topNodesIterator[0];
      assert.strictEqual(firstNode.id.toString(), 'ienfalfjdbdpebioblfackkekamfmbnh');

      const extensionNode = new Timeline.TimelineTreeView.TreeGridNode(
          firstNode, firstNode.totalTime, firstNode.selfTime, firstNode.totalTime, treeView);
      const extEntity = extensionNode?.createCell('site');

      let gotBadgeName = extEntity.querySelector<HTMLTableRowElement>('.entity-badge')?.textContent || '';
      assert.strictEqual(gotBadgeName, 'Extension');

      // Node with first party
      const secondNode = topNodesIterator[1] as Trace.Extras.TraceTree.Node;
      assert.strictEqual(secondNode.id.toString(), 'localhost');

      const firstPartyNode = new Timeline.TimelineTreeView.TreeGridNode(
          secondNode, secondNode.totalTime, secondNode.selfTime, secondNode.totalTime, treeView);
      const firstPartyEntity = firstPartyNode?.createCell('site');

      gotBadgeName = firstPartyEntity.querySelector<HTMLTableRowElement>('.entity-badge')?.textContent || '';
      assert.strictEqual(gotBadgeName, '1st party');
    });
  });
});