Apply the new asynchronous CookieMonster API to extension_service_unittest.cc.

chrome/browser/extensions/extension_service_unittest.cc

 // Copyright (c) 2011 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/extensions/extension_service_unittest.h"
 
 #include <algorithm>
 #include <set>
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/bind.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/json/json_reader.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/stl_util.h"
 #include "base/string16.h"
 #include "base/string_number_conversions.h"
@@ skipping 2634 matching lines @@
   UninstallExtension(good_crx, true);
 }
 
 TEST_F(ExtensionServiceTest, UninstallExtensionHelperTerminated) {
   InitializeEmptyExtensionService();
   InstallCrx(data_dir_.AppendASCII("good.crx"), true);
   TerminateExtension(good_crx);
   UninstallExtension(good_crx, true);
 }
 
+class Callback {

[jstritar, 2011/08/05 16:09:30]

Could you rename this something more specific? Like ExtensionCookieCallback?

[ycxiao, 2011/08/05 17:19:40]

Sure, I will rename it when I back to office(I am travelling now).

[jstritar, 2011/08/05 17:46:55]

Thanks!

[ycxiao, 2011/08/06 18:43:27]

Done.

+ public:
+  Callback()
+    : result_(false),
+      message_loop_factory_(MessageLoop::current()) {}
+
+  void SetCookieCallback(bool result) {
+    MessageLoop::current()->PostTask(
+        FROM_HERE,
+        message_loop_factory_.NewRunnableMethod(&MessageLoop::Quit));
+    result_ = result;
+  }
+
+  void GetAllCookiesCallback(const net::CookieList& list) {
+    MessageLoop::current()->PostTask(
+        FROM_HERE,
+        message_loop_factory_.NewRunnableMethod(&MessageLoop::Quit));
+    list_ = list;
+  }
+  net::CookieList list_;
+  bool result_;
+  ScopedRunnableMethodFactory<MessageLoop> message_loop_factory_;
+};
+
 // Verifies extension state is removed upon uninstall
 TEST_F(ExtensionServiceTest, ClearExtensionData) {
   InitializeEmptyExtensionService();
+  Callback callback;
 
   // Load a test extension.
   FilePath path = data_dir_;
   path = path.AppendASCII("good.crx");
   InstallCrx(path, true);
   const Extension* extension = service_->GetExtensionById(good_crx, false);
   ASSERT_TRUE(extension);
   GURL ext_url(extension->url());
   string16 origin_id =
       webkit_database::DatabaseUtil::GetOriginIdentifier(ext_url);
 
   // Set a cookie for the extension.
   net::CookieMonster* cookie_monster =
       profile_->GetRequestContextForExtensions()->GetURLRequestContext()->
       cookie_store()->GetCookieMonster();
   ASSERT_TRUE(cookie_monster);
   net::CookieOptions options;
-  cookie_monster->SetCookieWithOptions(ext_url, "dummy=value", options);
-  net::CookieList list = cookie_monster->GetAllCookiesForURL(ext_url);
-  EXPECT_EQ(1U, list.size());
+  cookie_monster->SetCookieWithOptionsAsync(
+       ext_url, "dummy=value", options,
+       base::Bind(&Callback::SetCookieCallback,
+                  base::Unretained(&callback)));
+  loop_.RunAllPending();
+  EXPECT_TRUE(callback.result_);
+  callback.message_loop_factory_.RevokeAll();
+  cookie_monster->GetAllCookiesForURLAsync(
+      ext_url,
+      base::Bind(&Callback::GetAllCookiesCallback,
+                 base::Unretained(&callback)));
+  loop_.RunAllPending();
+  EXPECT_EQ(1U, callback.list_.size());
+  callback.message_loop_factory_.RevokeAll();

[jstritar, 2011/08/05 16:05:06]

I'm not that familiar with the message loop code... why do you need the
RevokeAlls?

[ycxiao, 2011/08/05 17:19:40]

This RevokeAlls is a ScopedRunnableMethodFactory method which revoke any
outstanding runnable methods scheduled. It makes sure that the previous posted
tasks such as Quit() won't stay at the MessageLoop and have some further
effects.  

[jstritar, 2011/08/05 17:46:55]

Are they actually needed since you call loop_.RunAllPending()?

[ycxiao, 2011/08/06 18:43:27]

I think you are right. I removed the RevokeAll(). And resend a try job.

Thanks for notify!

 
   // Open a database.
   webkit_database::DatabaseTracker* db_tracker = profile_->GetDatabaseTracker();
   string16 db_name = UTF8ToUTF16("db");
   string16 description = UTF8ToUTF16("db_description");
   int64 size;
   db_tracker->DatabaseOpened(origin_id, db_name, description, 1, &size);
   db_tracker->DatabaseClosed(origin_id, db_name);
   std::vector<webkit_database::OriginInfo> origins;
   db_tracker->GetAllOriginsInfo(&origins);
@@ skipping 16 matching lines @@
   FilePath idb_path = idb_context->GetIndexedDBFilePath(origin_id);
   EXPECT_TRUE(file_util::CreateDirectory(idb_path.DirName()));
   EXPECT_EQ(0, file_util::WriteFile(idb_path, NULL, 0));
   EXPECT_TRUE(file_util::PathExists(idb_path));
 
   // Uninstall the extension.
   service_->UninstallExtension(good_crx, false, NULL);
   loop_.RunAllPending();
 
   // Check that the cookie is gone.
-  list = cookie_monster->GetAllCookiesForURL(ext_url);
-  EXPECT_EQ(0U, list.size());
+  cookie_monster->GetAllCookiesForURLAsync(
+       ext_url,
+       base::Bind(&Callback::GetAllCookiesCallback,
+                  base::Unretained(&callback)));
+  loop_.RunAllPending();
+  EXPECT_EQ(0U, callback.list_.size());
+  callback.message_loop_factory_.RevokeAll();
 
   // The database should have vanished as well.
   origins.clear();
   db_tracker->GetAllOriginsInfo(&origins);
   EXPECT_EQ(0U, origins.size());
 
   // Check that the LSO file has been removed.
   EXPECT_FALSE(file_util::PathExists(lso_path));
 
   // Check if the indexed db has disappeared too.
@@ skipping 1057 matching lines @@
   ASSERT_FALSE(AddPendingSyncInstall());
 
   // Wait for the external source to install.
   WaitForCrxInstall(crx_path_, true);
   ASSERT_TRUE(IsCrxInstalled());
 
   // Now that the extension is installed, sync request should fail
   // because the extension is already installed.
   ASSERT_FALSE(AddPendingSyncInstall());
 }