Blob support for IDB [Chromium]

content/child/indexed_db/proxy_webidbcursor_impl_unittest.cc

 // Copyright 2013 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/memory/scoped_ptr.h"
 #include "base/values.h"
 #include "content/child/indexed_db/indexed_db_dispatcher.h"
 #include "content/child/indexed_db/indexed_db_key_builders.h"
 #include "content/child/indexed_db/proxy_webidbcursor_impl.h"
 #include "content/child/thread_safe_sender.h"
 #include "content/common/indexed_db/indexed_db_key.h"
 #include "ipc/ipc_sync_message_filter.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/platform/WebData.h"
 #include "third_party/WebKit/public/platform/WebIDBCallbacks.h"
 
+using blink::WebBlobInfo;
 using blink::WebData;
 using blink::WebIDBCallbacks;
 using blink::WebIDBDatabase;
 using blink::WebIDBKey;
 using blink::WebIDBKeyTypeNumber;
+using blink::WebVector;
 
 namespace content {
 
 namespace {
 
 class MockDispatcher : public IndexedDBDispatcher {
  public:
   MockDispatcher(ThreadSafeSender* thread_safe_sender)
       : IndexedDBDispatcher(thread_safe_sender),
         prefetch_calls_(0),
@@ skipping 29 matching lines @@
  private:
   int prefetch_calls_;
   int last_prefetch_count_;
   int continue_calls_;
   int32 destroyed_cursor_id_;
   scoped_ptr<WebIDBCallbacks> callbacks_;
 };
 
 class MockContinueCallbacks : public WebIDBCallbacks {
  public:
-  MockContinueCallbacks(IndexedDBKey* key = 0) : key_(key) {}
+  MockContinueCallbacks(
+      IndexedDBKey* key = 0, WebVector<WebBlobInfo>* webBlobInfo = 0)
+    : key_(key),
+      webBlobInfo_(webBlobInfo) {
+  }
 
   virtual void onSuccess(const WebIDBKey& key,
                          const WebIDBKey& primaryKey,
-                         const WebData& value) {
+                         const WebData& value,
+                         const WebVector<WebBlobInfo>& webBlobInfo) OVERRIDE {
 
     if (key_)
       *key_ = IndexedDBKeyBuilder::Build(key);
+    if (webBlobInfo_)
+      *webBlobInfo_ = webBlobInfo;
   }
 
  private:
   IndexedDBKey* key_;
+  WebVector<WebBlobInfo>* webBlobInfo_;
 };
 
 }  // namespace
 
 TEST(RendererWebIDBCursorImplTest, PrefetchTest) {
 
   WebIDBKey null_key;
   null_key.assignNull();
 
   scoped_refptr<base::MessageLoopProxy> message_loop_proxy(
@@ skipping 35 matching lines @@
 
       // Verify that the requested count has increased since last time.
       int prefetch_count = dispatcher.last_prefetch_count();
       EXPECT_GT(prefetch_count, last_prefetch_count);
       last_prefetch_count = prefetch_count;
 
       // Fill the prefetch cache as requested.
       std::vector<IndexedDBKey> keys;
       std::vector<IndexedDBKey> primary_keys(prefetch_count);
       std::vector<WebData> values(prefetch_count);
+      std::vector<WebVector<WebBlobInfo> > blob_info;
       for (int i = 0; i < prefetch_count; ++i) {
         keys.push_back(IndexedDBKey(expected_key + i, WebIDBKeyTypeNumber));
+        blob_info.push_back(WebVector<WebBlobInfo>(
+            static_cast<size_t>(expected_key + i)));
       }
-      cursor.SetPrefetchData(keys, primary_keys, values);
+      cursor.SetPrefetchData(keys, primary_keys, values, blob_info);
 
       // Note that the real dispatcher would call cursor->CachedContinue()
       // immediately after cursor->SetPrefetchData() to service the request
       // that initiated the prefetch.
 
       // Verify that the cache is used for subsequent continue() calls.
       for (int i = 0; i < prefetch_count; ++i) {
         IndexedDBKey key;
-        cursor.continueFunction(null_key, new MockContinueCallbacks(&key));
+        WebVector<WebBlobInfo> web_blob_info;
+        cursor.continueFunction(null_key,
+            new MockContinueCallbacks(&key, &web_blob_info));
         EXPECT_EQ(continue_calls, dispatcher.continue_calls());
         EXPECT_EQ(repetitions + 1, dispatcher.prefetch_calls());
 
         EXPECT_EQ(WebIDBKeyTypeNumber, key.type());
+        EXPECT_EQ(expected_key, static_cast<int>(web_blob_info.size()));
         EXPECT_EQ(expected_key++, key.number());
       }
     }
   }
 
   EXPECT_EQ(dispatcher.destroyed_cursor_id(),
             RendererWebIDBCursorImpl::kInvalidCursorId);
 }
 
 }  // namespace content