Add browser test for CustomizationWallpaperDownloader.

chrome/browser/chromeos/login/wallpaper_manager_test_utils.cc

+// 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.
+
+#include "chrome/browser/chromeos/login/wallpaper_manager_test_utils.h"
+
+#include "ash/ash_switches.h"
+#include "base/command_line.h"
+#include "base/file_util.h"
+#include "base/files/file_path.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/logging.h"
+#include "base/run_loop.h"
+#include "base/time/time.h"
+#include "chrome/browser/chromeos/login/wallpaper_manager.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/gfx/codec/jpeg_codec.h"
+#include "ui/gfx/point.h"
+#include "ui/gfx/rect.h"
+
+namespace chromeos {
+
+namespace {
+
+class TestWallpaperObserverPendingListEmpty
+    : public WallpaperManager::Observer {
+ public:
+  explicit TestWallpaperObserverPendingListEmpty(
+      WallpaperManager* wallpaper_manager)
+      : empty_(false), wallpaper_manager_(wallpaper_manager) {
+    DCHECK(wallpaper_manager_);
+    wallpaper_manager_->AddObserver(this);
+  }
+
+  virtual ~TestWallpaperObserverPendingListEmpty() {
+    wallpaper_manager_->RemoveObserver(this);
+  }
+
+  virtual void OnWallpaperAnimationFinished(
+      const std::string& user_id) OVERRIDE {}
+
+  virtual void OnPendingListEmptyForTesting() OVERRIDE {
+    empty_ = true;
+    base::MessageLoop::current()->Quit();
+  }
+
+  void WaitForPendingListEmpty() {
+    if (wallpaper_manager_->GetPendingListSizeForTesting() == 0) {
+      empty_ = true;
+      return;
+    }
+    while (!empty_)
+      base::RunLoop().Run();
+  }
+
+ private:
+  bool empty_;
+  WallpaperManager* wallpaper_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestWallpaperObserverPendingListEmpty);
+};
+
+}  // namespace
+
+namespace wallpaper_manager_test_utils {
+
+const SkColor kLargeDefaultWallpaperColor = SK_ColorRED;
+const SkColor kSmallDefaultWallpaperColor = SK_ColorGREEN;
+const SkColor kLargeGuestWallpaperColor = SK_ColorBLUE;
+const SkColor kSmallGuestWallpaperColor = SK_ColorYELLOW;
+
+const SkColor kCustomWallpaperColor = SK_ColorMAGENTA;
+
+const int kWallpaperSize = 2;
+
+bool CreateJPEGImage(int width,
+                     int height,
+                     SkColor color,
+                     std::vector<unsigned char>* output) {
+  SkBitmap bitmap;
+  bitmap.setConfig(SkBitmap::kARGB_8888_Config, width, height, 0);
+  bitmap.allocPixels();
+  bitmap.eraseColor(color);
+
+  const int kQuality = 80;
+  if (!gfx::JPEGCodec::Encode(
+          static_cast<const unsigned char*>(bitmap.getPixels()),
+          gfx::JPEGCodec::FORMAT_SkBitmap,
+          width,
+          height,
+          bitmap.rowBytes(),
+          kQuality,
+          output)) {
+    LOG(ERROR) << "Unable to encode " << width << "x" << height << " bitmap";
+    return false;
+  }
+  return true;
+}
+
+gfx::ImageSkia CreateTestImage(int width, int height, SkColor color) {
+  SkBitmap bitmap;
+  bitmap.setConfig(SkBitmap::kARGB_8888_Config, width, height);
+  bitmap.allocPixels();
+  bitmap.eraseColor(color);
+  return gfx::ImageSkia::CreateFrom1xBitmap(bitmap);
+}
+
+bool WriteJPEGFile(const base::FilePath& path,
+                   int width,
+                   int height,
+                   SkColor color) {
+  std::vector<unsigned char> output;
+  if (!CreateJPEGImage(width, height, color, &output))
+    return false;
+
+  size_t bytes_written = base::WriteFile(
+      path, reinterpret_cast<const char*>(&output[0]), output.size());
+  if (bytes_written != output.size()) {
+    LOG(ERROR) << "Wrote " << bytes_written << " byte(s) instead of "
+               << output.size() << " to " << path.value();
+    return false;
+  }
+  return true;
+}
+
+bool ImageIsNearColor(gfx::ImageSkia image, SkColor expected_color) {
+  if (image.size().IsEmpty()) {
+    LOG(ERROR) << "Image is empty";
+    return false;
+  }
+
+  const SkBitmap* bitmap = image.bitmap();
+  if (!bitmap) {
+    LOG(ERROR) << "Unable to get bitmap from image";
+    return false;
+  }
+
+  bitmap->lockPixels();
+  gfx::Point center = gfx::Rect(image.size()).CenterPoint();
+  SkColor image_color = bitmap->getColor(center.x(), center.y());
+  bitmap->unlockPixels();
+
+  const int kDiff = 3;
+  if (std::abs(static_cast<int>(SkColorGetA(image_color)) -
+               static_cast<int>(SkColorGetA(expected_color))) > kDiff ||
+      std::abs(static_cast<int>(SkColorGetR(image_color)) -
+               static_cast<int>(SkColorGetR(expected_color))) > kDiff ||
+      std::abs(static_cast<int>(SkColorGetG(image_color)) -
+               static_cast<int>(SkColorGetG(expected_color))) > kDiff ||
+      std::abs(static_cast<int>(SkColorGetB(image_color)) -
+               static_cast<int>(SkColorGetB(expected_color))) > kDiff) {
+    LOG(ERROR) << "Expected color near 0x" << std::hex << expected_color
+               << " but got 0x" << image_color;
+    return false;
+  }
+
+  return true;
+}
+
+void WaitAsyncWallpaperLoadFinished() {
+  TestWallpaperObserverPendingListEmpty observer(WallpaperManager::Get());
+  observer.WaitForPendingListEmpty();
+}
+
+void CreateCmdlineWallpapers(const base::ScopedTempDir& dir,
+                             scoped_ptr<base::CommandLine>* command_line) {
+  std::vector<std::string> options;
+  options.push_back(std::string("WM_Test_cmdline"));
+  const base::FilePath small_file =
+      dir.path().Append(FILE_PATH_LITERAL("small.jpg"));
+  options.push_back(std::string("--") +
+                    ash::switches::kAshDefaultWallpaperSmall + "=" +
+                    small_file.value());
+  const base::FilePath large_file =
+      dir.path().Append(FILE_PATH_LITERAL("large.jpg"));
+  options.push_back(std::string("--") +
+                    ash::switches::kAshDefaultWallpaperLarge + "=" +
+                    large_file.value());
+
+  const base::FilePath guest_small_file =
+      dir.path().Append(FILE_PATH_LITERAL("guest_small.jpg"));
+  options.push_back(std::string("--") + ash::switches::kAshGuestWallpaperSmall +
+                    "=" + guest_small_file.value());
+  const base::FilePath guest_large_file =
+      dir.path().Append(FILE_PATH_LITERAL("guest_large.jpg"));
+  options.push_back(std::string("--") + ash::switches::kAshGuestWallpaperLarge +
+                    "=" + guest_large_file.value());
+
+  ASSERT_TRUE(WriteJPEGFile(
+      small_file, kWallpaperSize, kWallpaperSize, kSmallDefaultWallpaperColor));
+  ASSERT_TRUE(WriteJPEGFile(
+      large_file, kWallpaperSize, kWallpaperSize, kLargeDefaultWallpaperColor));
+
+  ASSERT_TRUE(WriteJPEGFile(guest_small_file,
+                            kWallpaperSize,
+                            kWallpaperSize,
+                            kSmallGuestWallpaperColor));
+  ASSERT_TRUE(WriteJPEGFile(guest_large_file,
+                            kWallpaperSize,
+                            kWallpaperSize,
+                            kLargeGuestWallpaperColor));
+
+  command_line->reset(new base::CommandLine(options));
+  WallpaperManager::Get()->SetCommandLineForTesting(command_line->get());
+}
+
+}  // namespace wallpaper_manager_test_utils
+
+}  // namespace chromeos