chrome/test/data/webui/engagement/site_engagement_browsertest.js

// Copyright 2017 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.

/**
 * @fileoverview Test suite for the Site Engagement WebUI.
 */
var ROOT_PATH = '../../../../../';
var EXAMPLE_URL_1 = 'http://example.com/';
var EXAMPLE_URL_2 = 'http://shmlexample.com/';

GEN('#include "chrome/browser/engagement/site_engagement_service.h"');
GEN('#include "chrome/browser/engagement/site_engagement_service_factory.h"');
GEN('#include "chrome/browser/ui/browser.h"');

function SiteEngagementBrowserTest() {}

SiteEngagementBrowserTest.prototype = {
  __proto__: testing.Test.prototype,

  browsePreload: 'chrome://site-engagement',

  runAccessibilityChecks: false,

  isAsync: true,

  testGenPreamble: function() {
    GEN('SiteEngagementService* service =');
    GEN('  SiteEngagementServiceFactory::GetForProfile(browser()->profile());');
    GEN('service->ResetBaseScoreForURL(GURL("' + EXAMPLE_URL_1 + '"), 10);');
    GEN('service->ResetBaseScoreForURL(GURL("' + EXAMPLE_URL_2 +
        '"), 3.14159);');
  },

  extraLibraries: [
    ROOT_PATH + 'third_party/mocha/mocha.js',
    ROOT_PATH + 'chrome/test/data/webui/mocha_adapter.js',
  ],

  /** @override */
  setUp: function() {
    testing.Test.prototype.setUp.call(this);
    suiteSetup(function() {
      return whenPageIsPopulatedForTest().then(disableAutoupdateForTests);
    });
  },
};

// This test is flaky on Windows. See https://crbug.com/734716.
GEN('#if defined(OS_WIN)');
GEN('#define MAYBE_All DISABLED_All');
GEN('#else');
GEN('#define MAYBE_All All');
GEN('#endif');
TEST_F('SiteEngagementBrowserTest', 'MAYBE_All', function() {
  var cells;

  function getCells() {
    var originCells =
        Array.from(document.getElementsByClassName('origin-cell'));
    var scoreInputs =
        Array.from(document.getElementsByClassName('base-score-input'));
    var bonusScoreCells =
        Array.from(document.getElementsByClassName('bonus-score-cell'));
    var totalScoreCells =
        Array.from(document.getElementsByClassName('total-score-cell'));
    return originCells.map((c, i) => {
      return {
        origin: c,
        scoreInput: scoreInputs[i],
        bonusScore: bonusScoreCells[i],
        totalScore: totalScoreCells[i],
      };
    });
  }

  setup(function() {
    cells = getCells();
  });

  test('check engagement values are loaded', function() {
    assertDeepEquals(
        [EXAMPLE_URL_1, EXAMPLE_URL_2], cells.map((c) => c.origin.textContent));
  });

  test('scores rounded to 2 decimal places', function() {
    assertDeepEquals(['10', '3.14'], cells.map((x) => x.scoreInput.value));
    assertDeepEquals(['0', '0'], cells.map((x) => x.bonusScore.textContent));
    assertDeepEquals(
        ['10', '3.14'], cells.map((x) => x.totalScore.textContent));
  });

  test('change score', function() {
    var firstRow = cells[0];
    firstRow.scoreInput.value = 50;
    firstRow.scoreInput.dispatchEvent(new Event('change'));

    return uiHandler.getSiteEngagementDetails().then((response) => {
      assertEquals(firstRow.origin.textContent, response.info[0].origin.url);
      assertEquals(50, response.info[0].baseScore);
    });
  });
  mocha.run();
});