Update AddToHomescreenDataFetcher to accept badge icon for WebAPK.

chrome/browser/android/webapps/add_to_homescreen_data_fetcher_unittest.cc

 // Copyright 2016 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/android/webapps/add_to_homescreen_data_fetcher.h"
 
 #include <memory>
 #include <string>
 
 #include "base/callback_forward.h"
@@ skipping 113 matching lines @@
   }
 
   SkBitmap FinalizeLauncherIconInBackground(const SkBitmap& icon,
                                             const GURL& url,
                                             bool* is_generated) override {
     *is_generated = false;
     return icon;
   }
 
   void OnDataAvailable(const ShortcutInfo& info,
-                       const SkBitmap& icon) override {
+                       const SkBitmap& primary_icon,
+                       const SkBitmap& badge_icon) override {
     data_available_ = true;
     if (!quit_closure_.is_null())
       quit_closure_.Run();
   }
 
   bool is_webapk_compatible() const { return is_webapk_compatible_; }
   bool determined_webapk_compatibility() const {
     return determined_webapk_compatibility_;
   }
   bool title_available() const { return title_available_; }
@@ skipping 51 matching lines @@
   }
 
   void TearDown() override {
     embedded_worker_test_helper_.reset();
     ChromeRenderViewHostTestHarness::TearDown();
   }
 
   scoped_refptr<AddToHomescreenDataFetcher> BuildFetcher(
       bool check_webapk_compatible,
       AddToHomescreenDataFetcher::Observer* observer) {
-    return new AddToHomescreenDataFetcher(web_contents(), 1, 1, 1, 1,
+    return new AddToHomescreenDataFetcher(web_contents(), 1, 1, 1, 1, 1,
                                           check_webapk_compatible, observer);
   }
 
   // Set the manifest to be returned as a result of WebContents::GetManifest().
   void SetManifest(const GURL& manifest_url,
                    const content::Manifest& manifest,
                    int fetch_delay_ms) {
     MockWebContents* mock_web_contents =
         static_cast<MockWebContents*>(web_contents());
     mock_web_contents->SetManifest(manifest_url, manifest, fetch_delay_ms);
@@ skipping 141 matching lines @@
     }
     // This callback enables the text field in the add-to-homescreen dialog.
     EXPECT_TRUE(waiter.title_available());
 
     fetcher->set_weak_observer(nullptr);
 }
 
 INSTANTIATE_TEST_CASE_P(CheckWebApkCompatibility,
                         AddToHomescreenDataFetcherTestCommon,
                         ::testing::Values(false, true));