Load extension icons for more scale factors.

chrome/browser/extensions/extension_icon_manager_unittest.cc

 // Copyright (c) 2012 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 "base/command_line.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
+#include "base/strings/stringprintf.h"
+#include "base/test/scoped_command_line.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "chrome/browser/extensions/extension_icon_manager.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/crx_file/id_util.h"
 #include "content/public/test/test_browser_thread.h"
 #include "extensions/common/extension.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/layout.h"
+#include "ui/display/display_list.h"
+#include "ui/display/display_switches.h"
+#include "ui/display/test/test_screen.h"
 #include "ui/gfx/favicon_size.h"
 #include "ui/gfx/image/image.h"
 #include "ui/gfx/image/image_skia.h"
 #include "ui/gfx/image/image_unittest_util.h"
 #include "ui/gfx/skia_util.h"
 
 namespace extensions {
 namespace {
 
+class ScopedSetDeviceScaleFactor {
+ public:
+  explicit ScopedSetDeviceScaleFactor(float scale) {
+    display::Display::ResetForceDeviceScaleFactorForTesting();
+    // It should be enough just to call Display::SetScaleAndBounds, but on Mac
+    // that rounds the scale unless there's a forced device scale factor.
+    command_line_.GetProcessCommandLine()->AppendSwitchASCII(
+        switches::kForceDeviceScaleFactor, base::StringPrintf("%3.2f", scale));
+    // This has to be inited after fiddling with the command line.
+    test_screen_ = base::MakeUnique<display::test::TestScreen>();
+    display::Screen::SetScreenInstance(test_screen_.get());
+  }
+
+  ~ScopedSetDeviceScaleFactor() {
+    display::Display::ResetForceDeviceScaleFactorForTesting();
+    display::Screen::SetScreenInstance(nullptr);
+  }
+
+ private:
+  std::unique_ptr<display::test::TestScreen> test_screen_;
+  base::test::ScopedCommandLine command_line_;
+
+  DISALLOW_COPY_AND_ASSIGN(ScopedSetDeviceScaleFactor);
+};
+
 using content::BrowserThread;
 
 // Our test class that takes care of managing the necessary threads for loading
 // extension icons, and waiting for those loads to happen.
 class ExtensionIconManagerTest : public testing::Test {
  public:
   ExtensionIconManagerTest() :
       unwaited_image_loads_(0),
       waiting_(false),
       ui_thread_(BrowserThread::UI, &ui_loop_),
@@ skipping 185 matching lines @@
       {ui::SCALE_FACTOR_125P},
       // One scale factor for which we have an icon, one scale factor for which
       // we don't.
       {ui::SCALE_FACTOR_100P, ui::SCALE_FACTOR_300P},
       // Just a scale factor where we don't have any icon. This falls back to
       // the default icon.
       {ui::SCALE_FACTOR_300P}};
 
   for (size_t i = 0; i < supported_scales.size(); ++i) {
     SCOPED_TRACE(testing::Message() << "Test case: " << i);
+    // Since active Displays' scale factors are also taken into account, to make
+    // the logic in this test work, we need to set the scale factor to one of
+    // the "supported" scales.
+    ScopedSetDeviceScaleFactor scoped_dsf(
+        ui::GetScaleForScaleFactor(supported_scales[i][0]));
     ui::test::ScopedSetSupportedScaleFactors scoped(supported_scales[i]);
     TestIconManager icon_manager(this);
 
     icon_manager.LoadIcon(profile.get(), extension.get());
     WaitForImageLoad();
     gfx::Image icon = icon_manager.GetIcon(extension->id());
-
     // Determine if the default icon fallback will be used. We'll use the
     // default when none of the supported scale factors can find an appropriate
     // icon.
     bool should_fall_back_to_default = true;
     for (auto supported_scale : supported_scales[i]) {
       if (gfx::kFaviconSize * ui::GetScaleForScaleFactor(supported_scale) <=
           kMaxIconSizeInManifest) {
         should_fall_back_to_default = false;
         break;
       }
@@ skipping 13 matching lines @@
 
       const bool has_representation = image_skia.HasRepresentation(scale);
       // We shouldn't have a representation if the extension didn't provide a
       // big enough icon.
       if (gfx::kFaviconSize * scale > kMaxIconSizeInManifest)
         EXPECT_FALSE(has_representation);
       else
         EXPECT_EQ(ui::IsSupportedScale(scale), has_representation);
     }
   }
+
+  // Now check that the scale factors for active displays are respected, even
+  // when it's not a supported scale.
+  EXPECT_FALSE(ui::IsSupportedScale(ui::SCALE_FACTOR_150P));
+  ScopedSetDeviceScaleFactor scoped_dsf(1.5f);
+  TestIconManager icon_manager(this);
+  icon_manager.LoadIcon(profile.get(), extension.get());
+  WaitForImageLoad();
+  gfx::ImageSkia icon = icon_manager.GetIcon(extension->id()).AsImageSkia();
+  EXPECT_TRUE(icon.HasRepresentation(1.5f));
 }
 
 }  // namespace
 }  // namespace extensions