Cleanup mojo Wallpaper interfaces for mash.

chrome/browser/chromeos/customization/customization_wallpaper_downloader_browsertest.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 <stddef.h>
 
 #include <vector>
 
-#include "ash/common/wallpaper/wallpaper_controller.h"
 #include "ash/common/wm_shell.h"
 #include "base/command_line.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/time/time.h"
 #include "chrome/browser/chromeos/customization/customization_document.h"
 #include "chrome/browser/chromeos/customization/customization_wallpaper_downloader.h"
 #include "chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager.h"
 #include "chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager_test_utils.h"
+#include "chrome/browser/ui/ash/ash_util.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chromeos/chromeos_switches.h"
+#include "content/public/common/service_manager_connection.h"
 #include "net/http/http_response_headers.h"
 #include "net/http/http_status_code.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_fetcher_impl.h"
+#include "services/shell/public/cpp/connector.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chromeos {
 
 namespace {
 
 const char kOEMWallpaperURL[] = "http://somedomain.com/image.png";
 
 const char kServicesManifest[] =
     "{"
@@ skipping 42 matching lines @@
       base::RunLoop().Run();
   }
 
  private:
   bool finished_;
   WallpaperManager* wallpaper_manager_;
 
   DISALLOW_COPY_AND_ASSIGN(TestWallpaperObserver);
 };
 
+// The callback used to retrieve the wallpaper from ash.
+void GetWallpaperCallback(gfx::ImageSkia* image_out,
+                          wallpaper::WallpaperLayout* layout_out,
+                          base::Closure callback,
+                          const SkBitmap& wallpaper,
+                          wallpaper::WallpaperLayout layout) {
+  *image_out = gfx::ImageSkia::CreateFrom1xBitmap(wallpaper);
+  *layout_out = layout;
+  callback.Run();
+}
+
+// A helper to synchronously get the wallpaper from ash.
+void GetWallpaper(gfx::ImageSkia* image_out,
+                  wallpaper::WallpaperLayout* layout_out) {
+  auto connection = content::ServiceManagerConnection::GetForProcess();
+  auto connector = connection ? connection->GetConnector() : nullptr;

[James Cook, 2016/10/14 21:58:02]

As before, I don't think these types are complex enough to justify "auto". It
probably avoids line wrapping, but I don't think that's enough benefit. I would
rather have explicit types.

[msw, 2016/10/17 22:22:47]

I reverted these test changes.

+  if (!connector)
+    return;
+
+  ash::mojom::WallpaperControllerPtr wallpaper_controller;
+  // Under mash the WallpaperController interface is in the ash process. In

[James Cook, 2016/10/14 21:58:02]

So does this test no longer exercise the classic ash code path?

[msw, 2016/10/17 22:22:48]

I restored the classic ash code path here.

+  // classic ash we provide it to ourself.
+  if (chrome::IsRunningInMash()) {
+    connector->ConnectToInterface("service:ash", &wallpaper_controller);
+  } else {
+    connector->ConnectToInterface("service:content_browser",
+                                  &wallpaper_controller);
+  }
+  base::RunLoop run_loop;
+  wallpaper_controller->GetWallpaper(base::Bind(
+      &GetWallpaperCallback, image_out, layout_out, run_loop.QuitClosure()));
+  run_loop.Run();
+}
+
 }  // namespace
 
 // This is helper class for net::FakeURLFetcherFactory.
 class TestWallpaperImageURLFetcherCallback {
  public:
   TestWallpaperImageURLFetcherCallback(
       const GURL& url,
       const size_t require_retries,
       const std::vector<unsigned char>& jpeg_data_raw)
       : url_(url),
@@ skipping 114 matching lines @@
   // requests.
   std::unique_ptr<net::TestURLFetcherFactory> fallback_fetcher_factory_;
   std::unique_ptr<net::FakeURLFetcherFactory> fetcher_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(WallpaperImageFetcherFactory);
 };
 
 class CustomizationWallpaperDownloaderBrowserTest
     : public InProcessBrowserTest {
  public:
-  CustomizationWallpaperDownloaderBrowserTest()
-      : controller_(NULL),
-        local_state_(NULL) {
-  }
-
+  CustomizationWallpaperDownloaderBrowserTest() {}
   ~CustomizationWallpaperDownloaderBrowserTest() override {}
 
-  void SetUpOnMainThread() override {
-    controller_ = ash::WmShell::Get()->wallpaper_controller();
-    local_state_ = g_browser_process->local_state();
-  }
-
   void SetUpCommandLine(base::CommandLine* command_line) override {
     command_line->AppendSwitch(chromeos::switches::kLoginManager);
     command_line->AppendSwitchASCII(chromeos::switches::kLoginProfile, "user");
   }
 
-  void TearDownOnMainThread() override { controller_ = NULL; }
-
  protected:
   void CreateCmdlineWallpapers() {
     cmdline_wallpaper_dir_.reset(new base::ScopedTempDir);
     ASSERT_TRUE(cmdline_wallpaper_dir_->CreateUniqueTempDir());
     wallpaper_manager_test_utils::CreateCmdlineWallpapers(
         *cmdline_wallpaper_dir_, &wallpaper_manager_command_line_);
   }
 
-  ash::WallpaperController* controller_;
-  PrefService* local_state_;
   std::unique_ptr<base::CommandLine> wallpaper_manager_command_line_;
 
   // Directory created by CreateCmdlineWallpapersAndSetFlags() to store default
   // wallpaper images.
   std::unique_ptr<base::ScopedTempDir> cmdline_wallpaper_dir_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(CustomizationWallpaperDownloaderBrowserTest);
 };
 
 IN_PROC_BROWSER_TEST_F(CustomizationWallpaperDownloaderBrowserTest,
                        OEMWallpaperIsPresent) {
   CreateCmdlineWallpapers();
   WallpaperManager::Get()->SetDefaultWallpaperNow(EmptyAccountId());
   wallpaper_manager_test_utils::WaitAsyncWallpaperLoadFinished();
+
+  gfx::ImageSkia image;
+  wallpaper::WallpaperLayout layout = wallpaper::NUM_WALLPAPER_LAYOUT;
+  GetWallpaper(&image, &layout);
   EXPECT_TRUE(wallpaper_manager_test_utils::ImageIsNearColor(
-      controller_->GetWallpaper(),
-      wallpaper_manager_test_utils::kSmallDefaultWallpaperColor));
-
+      image, wallpaper_manager_test_utils::kSmallDefaultWallpaperColor));
   WallpaperImageFetcherFactory url_factory(
       GURL(kOEMWallpaperURL),
       wallpaper_manager_test_utils::kWallpaperSize,
       wallpaper_manager_test_utils::kWallpaperSize,
       wallpaper_manager_test_utils::kCustomWallpaperColor,
       0 /* require_retries */);
 
   TestWallpaperObserver observer(WallpaperManager::Get());
   chromeos::ServicesCustomizationDocument* customization =
       chromeos::ServicesCustomizationDocument::GetInstance();
   EXPECT_TRUE(
       customization->LoadManifestFromString(std::string(kServicesManifest)));
 
   observer.WaitForWallpaperAnimationFinished();
+  GetWallpaper(&image, &layout);
   EXPECT_TRUE(wallpaper_manager_test_utils::ImageIsNearColor(
-      controller_->GetWallpaper(),
-      wallpaper_manager_test_utils::kCustomWallpaperColor));
+      image, wallpaper_manager_test_utils::kCustomWallpaperColor));
   EXPECT_EQ(1U, url_factory.num_attempts());
 }
 
 IN_PROC_BROWSER_TEST_F(CustomizationWallpaperDownloaderBrowserTest,
                        OEMWallpaperRetryFetch) {
   CreateCmdlineWallpapers();
   WallpaperManager::Get()->SetDefaultWallpaperNow(EmptyAccountId());
   wallpaper_manager_test_utils::WaitAsyncWallpaperLoadFinished();
+
+  gfx::ImageSkia image;
+  wallpaper::WallpaperLayout layout = wallpaper::NUM_WALLPAPER_LAYOUT;
+  GetWallpaper(&image, &layout);
   EXPECT_TRUE(wallpaper_manager_test_utils::ImageIsNearColor(
-      controller_->GetWallpaper(),
-      wallpaper_manager_test_utils::kSmallDefaultWallpaperColor));
+      image, wallpaper_manager_test_utils::kSmallDefaultWallpaperColor));
 
   WallpaperImageFetcherFactory url_factory(
       GURL(kOEMWallpaperURL),
       wallpaper_manager_test_utils::kWallpaperSize,
       wallpaper_manager_test_utils::kWallpaperSize,
       wallpaper_manager_test_utils::kCustomWallpaperColor,
       1 /* require_retries */);
 
   TestWallpaperObserver observer(WallpaperManager::Get());
   chromeos::ServicesCustomizationDocument* customization =
       chromeos::ServicesCustomizationDocument::GetInstance();
   EXPECT_TRUE(
       customization->LoadManifestFromString(std::string(kServicesManifest)));
 
   observer.WaitForWallpaperAnimationFinished();
+  GetWallpaper(&image, &layout);
   EXPECT_TRUE(wallpaper_manager_test_utils::ImageIsNearColor(
-      controller_->GetWallpaper(),
-      wallpaper_manager_test_utils::kCustomWallpaperColor));
+      image, wallpaper_manager_test_utils::kCustomWallpaperColor));
 
   EXPECT_EQ(2U, url_factory.num_attempts());
 }
 
 }  // namespace chromeos