Merge 65840 - Provide sync integration tests with a way to set a favicon for ...

chrome/test/live_sync/live_bookmarks_sync_test.h

 // Copyright (c) 2010 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.
 
 #ifndef CHROME_TEST_LIVE_SYNC_LIVE_BOOKMARKS_SYNC_TEST_H_
 #define CHROME_TEST_LIVE_SYNC_LIVE_BOOKMARKS_SYNC_TEST_H_
 #pragma once
 
 #include <string>
 #include <vector>
 
 #include "base/compiler_specific.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/test/live_sync/bookmark_model_verifier.h"
 #include "chrome/test/live_sync/live_sync_test.h"
 #include "chrome/test/ui_test_utils.h"
+#include "gfx/codec/png_codec.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/skia/include/core/SkBitmap.h"
 
 class LiveBookmarksSyncTest : public LiveSyncTest {
  public:
   explicit LiveBookmarksSyncTest(TestType test_type)
       : LiveSyncTest(test_type) {}
 
   virtual ~LiveBookmarksSyncTest() {}
 
   // Sets up sync profiles and clients and initializes the bookmark verifier.
   virtual bool SetupClients() WARN_UNUSED_RESULT {
@@ skipping 104 matching lines @@
   void SetTitle(int profile,
                 const BookmarkNode* node,
                 const std::wstring& new_title) {
     ASSERT_EQ(GetBookmarkModel(profile)->GetNodeByID(node->id()), node)
         << "Node " << node->GetTitle() << " does not belong to "
         << "Profile " << profile;
     verifier_helper_->SetTitle(
         GetBookmarkModel(profile), node, WideToUTF16(new_title));
   }
 
+  // Sets the favicon of the node |node| (of type BookmarkNode::URL) in the
+  // bookmark model of profile |profile| using the data in |icon_bytes_vector|.
+  void SetFavicon(int profile,
+                  const BookmarkNode* node,
+                  const std::vector<unsigned char>& icon_bytes_vector) {
+    ASSERT_EQ(GetBookmarkModel(profile)->GetNodeByID(node->id()), node)
+        << "Node " << node->GetTitle() << " does not belong to "
+        << "Profile " << profile;
+    ASSERT_EQ(BookmarkNode::URL, node->type())
+        << "Node " << node->GetTitle() << " must be a url.";
+    verifier_helper_->SetFavicon(
+        GetBookmarkModel(profile), node, icon_bytes_vector);
+  }
+
   // Changes the url of the node |node| in the bookmark model of profile
   // |profile| to |new_url|. Returns a pointer to the node with the changed url.
   const BookmarkNode* SetURL(int profile,
                              const BookmarkNode* node,
                              const GURL& new_url) WARN_UNUSED_RESULT {
     if (GetBookmarkModel(profile)->GetNodeByID(node->id()) != node) {
       LOG(ERROR) << "Node " << node->GetTitle() << " does not belong to "
                  << "Profile " << profile;
       return NULL;
     }
@@ skipping 102 matching lines @@
   const BookmarkNode* GetUniqueNodeByURL(int profile,
                                          const GURL& url) WARN_UNUSED_RESULT {
     std::vector<const BookmarkNode*> nodes;
     GetBookmarkModel(profile)->GetNodesByURL(url, &nodes);
     EXPECT_EQ(1U, nodes.size());
     if (nodes.empty())
       return NULL;
     return nodes[0];
   }
 
+  // Creates a unique favicon using |seed|.
+  static std::vector<unsigned char> CreateFavicon(int seed) {
+    const int w = 16;
+    const int h = 16;
+    SkBitmap bmp;
+    bmp.setConfig(SkBitmap::kARGB_8888_Config, w, h);
+    bmp.allocPixels();
+    uint32_t* src_data = bmp.getAddr32(0, 0);
+    for (int i = 0; i < w * h; ++i) {
+      src_data[i] = SkPreMultiplyARGB((seed + i) % 255,
+                                      (seed + i) % 250,
+                                      (seed + i) % 245,
+                                      (seed + i) % 240);
+    }
+    std::vector<unsigned char> favicon;
+    gfx::PNGCodec::EncodeBGRASkBitmap(bmp, false, &favicon);
+    return favicon;
+  }
+
  private:
   // Helper object that has the functionality to verify changes made to the
   // bookmarks of individual profiles.
   scoped_ptr<BookmarkModelVerifier> verifier_helper_;
 
   DISALLOW_COPY_AND_ASSIGN(LiveBookmarksSyncTest);
 };
 
 class SingleClientLiveBookmarksSyncTest : public LiveBookmarksSyncTest {
  public:
@@ skipping 29 matching lines @@
  public:
   ManyClientLiveBookmarksSyncTest()
       : LiveBookmarksSyncTest(MANY_CLIENT) {}
   ~ManyClientLiveBookmarksSyncTest() {}
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ManyClientLiveBookmarksSyncTest);
 };
 
 #endif  // CHROME_TEST_LIVE_SYNC_LIVE_BOOKMARKS_SYNC_TEST_H_