| Index: chrome/browser/net/sqlite_server_bound_cert_store_unittest.cc
|
| diff --git a/chrome/browser/net/sqlite_server_bound_cert_store_unittest.cc b/chrome/browser/net/sqlite_server_bound_cert_store_unittest.cc
|
| index eeedf778442c0fe50aee76640bd862650b4d2b87..83db52a7528e8f832fa7e03058ee1eadf97151eb 100644
|
| --- a/chrome/browser/net/sqlite_server_bound_cert_store_unittest.cc
|
| +++ b/chrome/browser/net/sqlite_server_bound_cert_store_unittest.cc
|
| @@ -10,10 +10,10 @@
|
| #include "base/message_loop.h"
|
| #include "base/run_loop.h"
|
| #include "base/stl_util.h"
|
| -#include "base/test/thread_test_helper.h"
|
| #include "chrome/browser/net/sqlite_server_bound_cert_store.h"
|
| #include "chrome/common/chrome_constants.h"
|
| -#include "content/public/test/test_browser_thread.h"
|
| +#include "content/public/browser/browser_thread.h"
|
| +#include "content/public/test/test_browser_thread_bundle.h"
|
| #include "net/base/test_data_directory.h"
|
| #include "net/test/cert_test_util.h"
|
| #include "sql/statement.h"
|
| @@ -25,8 +25,7 @@ using content::BrowserThread;
|
| class SQLiteServerBoundCertStoreTest : public testing::Test {
|
| public:
|
| SQLiteServerBoundCertStoreTest()
|
| - : db_thread_(BrowserThread::DB),
|
| - io_thread_(BrowserThread::IO, &message_loop_) {}
|
| + : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {}
|
|
|
| void Load(
|
| ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert>* certs) {
|
| @@ -80,7 +79,6 @@ class SQLiteServerBoundCertStoreTest : public testing::Test {
|
| }
|
|
|
| virtual void SetUp() {
|
| - db_thread_.Start();
|
| ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
|
| store_ = new SQLiteServerBoundCertStore(
|
| temp_dir_.path().Append(chrome::kOBCertFilename), NULL);
|
| @@ -97,9 +95,7 @@ class SQLiteServerBoundCertStoreTest : public testing::Test {
|
| "a", "b"));
|
| }
|
|
|
| - base::MessageLoopForIO message_loop_;
|
| - content::TestBrowserThread db_thread_;
|
| - content::TestBrowserThread io_thread_;
|
| + content::TestBrowserThreadBundle thread_bundle_;
|
| base::ScopedTempDir temp_dir_;
|
| scoped_refptr<SQLiteServerBoundCertStore> store_;
|
| ScopedVector<net::DefaultServerBoundCertStore::ServerBoundCert> certs_;
|
| @@ -120,11 +116,8 @@ TEST_F(SQLiteServerBoundCertStoreTest, TestPersistence) {
|
| // to write its data to disk. Then we can see if after loading it again it
|
| // is still there.
|
| store_ = NULL;
|
| - scoped_refptr<base::ThreadTestHelper> helper(
|
| - new base::ThreadTestHelper(
|
| - BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
|
| // Make sure we wait until the destructor has run.
|
| - ASSERT_TRUE(helper->Run());
|
| + base::RunLoop().RunUntilIdle();
|
| store_ = new SQLiteServerBoundCertStore(
|
| temp_dir_.path().Append(chrome::kOBCertFilename), NULL);
|
|
|
| @@ -158,7 +151,7 @@ TEST_F(SQLiteServerBoundCertStoreTest, TestPersistence) {
|
| store_->DeleteServerBoundCert(*certs[1]);
|
| store_ = NULL;
|
| // Make sure we wait until the destructor has run.
|
| - ASSERT_TRUE(helper->Run());
|
| + base::RunLoop().RunUntilIdle();
|
| certs.clear();
|
| store_ = new SQLiteServerBoundCertStore(
|
| temp_dir_.path().Append(chrome::kOBCertFilename), NULL);
|
| @@ -233,11 +226,7 @@ TEST_F(SQLiteServerBoundCertStoreTest, TestUpgradeV1) {
|
| ASSERT_STREQ("\xbb", certs[1]->cert().c_str());
|
|
|
| store_ = NULL;
|
| - // Make sure we wait until the destructor has run.
|
| - scoped_refptr<base::ThreadTestHelper> helper(
|
| - new base::ThreadTestHelper(
|
| - BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
|
| - ASSERT_TRUE(helper->Run());
|
| + base::RunLoop().RunUntilIdle();
|
|
|
| // Verify the database version is updated.
|
| {
|
| @@ -322,10 +311,7 @@ TEST_F(SQLiteServerBoundCertStoreTest, TestUpgradeV2) {
|
|
|
| store_ = NULL;
|
| // Make sure we wait until the destructor has run.
|
| - scoped_refptr<base::ThreadTestHelper> helper(
|
| - new base::ThreadTestHelper(
|
| - BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
|
| - ASSERT_TRUE(helper->Run());
|
| + base::RunLoop().RunUntilIdle();
|
|
|
| // Verify the database version is updated.
|
| {
|
| @@ -414,10 +400,7 @@ TEST_F(SQLiteServerBoundCertStoreTest, TestUpgradeV3) {
|
|
|
| store_ = NULL;
|
| // Make sure we wait until the destructor has run.
|
| - scoped_refptr<base::ThreadTestHelper> helper(
|
| - new base::ThreadTestHelper(
|
| - BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
|
| - ASSERT_TRUE(helper->Run());
|
| + base::RunLoop().RunUntilIdle();
|
|
|
| // Verify the database version is updated.
|
| {
|
|
|