ui/file_manager/image_loader/image_loader_client_unittest.js

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

'use strict';

/** @suppress {const|checkTypes} */
function setUp() {
  chrome.metricsPrivate = {
    MetricTypeType:
        {HISTOGRAM_LOG: 'histogram-log', HISTOGRAM_LINEAR: 'histogram-linear'},
    recordPercentage: function() {},
    recordValue: function() {}
  };

  chrome.i18n = {
    getMessage: function() {},
  };
}

/**
 * Lets the client to load URL and returns the local cache (not caches in the
 * image loader extension) is used or not.
 *
 * @param {ImageLoaderClient} client
 * @param {string} url URL
 * @param {boolean} cache Whether to request caching on the request.
 * @return {Promise<boolean>} True if the local cache is used.
 */
function loadAndCheckCacheUsed(client, url, cache) {
  let cacheUsed = true;

  /** @suppress {accessControls} */
  ImageLoaderClient.sendMessage_ = function(message, callback) {
    cacheUsed = false;
    if (callback) {
      callback(new LoadImageResponse(
          LoadImageResponseStatus.SUCCESS, message.taskId || -1,
          {data: 'ImageData', width: 100, height: 100}));
    }
  };

  let request = LoadImageRequest.createForUrl(url);
  request.cache = cache;

  return new Promise(function(fulfill) {
    client.load(request, function() {
      fulfill(cacheUsed);
    });
  });
}

function testCache(callback) {
  var client = new ImageLoaderClient();
  reportPromise(
      loadAndCheckCacheUsed(client, 'http://example.com/image.jpg', true)
          .then(function(cacheUsed) {
            assertFalse(cacheUsed);
            return loadAndCheckCacheUsed(
                client, 'http://example.com/image.jpg', true);
          })
          .then(function(cacheUsed) {
            assertTrue(cacheUsed);
          }),
      callback);
}

function testNoCache(callback) {
  var client = new ImageLoaderClient();
  reportPromise(
      loadAndCheckCacheUsed(client, 'http://example.com/image.jpg', false)
          .then(function(cacheUsed) {
            assertFalse(cacheUsed);
            return loadAndCheckCacheUsed(
                client, 'http://example.com/image.jpg', false);
          })
          .then(function(cacheUsed) {
            assertFalse(cacheUsed);
          }),
      callback);
}

function testDataURLCache(callback) {
  var client = new ImageLoaderClient();
  reportPromise(
      loadAndCheckCacheUsed(client, 'data:URI', true)
          .then(function(cacheUsed) {
            assertFalse(cacheUsed);
            return loadAndCheckCacheUsed(client, 'data:URI', true);
          })
          .then(function(cacheUsed) {
            assertFalse(cacheUsed);
          }),
      callback);
}