The chromium-side backchannel plumbing for blobs in IDB.

content/child/indexed_db/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/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 54 matching lines @@
   int reset_calls_;
   int last_used_count_;
   int advance_calls_;
   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
 
 class WebIDBCursorImplTest : public testing::Test {
  public:
   WebIDBCursorImplTest() {
     null_key_.assignNull();
     sync_message_filter_ = new IPC::SyncMessageFilter(NULL);
     thread_safe_sender_ = new ThreadSafeSender(
@@ skipping 46 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(),
             WebIDBCursorImpl::kInvalidCursorId);
 }
 
 TEST_F(WebIDBCursorImplTest, AdvancePrefetchTest) {
@@ skipping 18 matching lines @@
             dispatcher_->continue_calls());
   EXPECT_EQ(0, dispatcher_->advance_calls());
 
   const int prefetch_count = dispatcher_->last_prefetch_count();
 
   // Fill the prefetch cache as requested.
   int expected_key = 0;
   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.
 
   // Need at least this many in the cache for the test steps.
   ASSERT_GE(prefetch_count, 5);
 
   // IDBCursor.continue()
   IndexedDBKey key;
@@ skipping 48 matching lines @@
   cursor.ResetPrefetchCache();
 
   // No reset should have been sent since nothing has been received yet.
   EXPECT_EQ(0, dispatcher_->reset_calls());
 
   // Fill the prefetch cache as requested.
   int prefetch_count = dispatcher_->last_prefetch_count();
   std::vector<IndexedDBKey> keys(prefetch_count);
   std::vector<IndexedDBKey> primary_keys(prefetch_count);
   std::vector<WebData> values(prefetch_count);
-  cursor.SetPrefetchData(keys, primary_keys, values);
+  std::vector<WebVector<WebBlobInfo> > blob_info(prefetch_count);
+  cursor.SetPrefetchData(keys, primary_keys, values, blob_info);
 
   // No reset should have been sent since prefetch data hasn't been used.
   EXPECT_EQ(0, dispatcher_->reset_calls());
 
   // The real dispatcher would call cursor->CachedContinue(), so do that:
   scoped_ptr<WebIDBCallbacks> callbacks(new MockContinueCallbacks());
   cursor.CachedContinue(callbacks.get());
 
   // Now the cursor should have reset the rest of the cache.
   EXPECT_EQ(1, dispatcher_->reset_calls());
   EXPECT_EQ(1, dispatcher_->last_used_count());
 }
 
 }  // namespace content