| Index: chrome/browser/user_style_sheet_watcher_unittest.cc
|
| diff --git a/chrome/browser/user_style_sheet_watcher_unittest.cc b/chrome/browser/user_style_sheet_watcher_unittest.cc
|
| index e3f0d069256c42b89754d7434e003672ac668de6..c6fb59f40e57cc01a94c5f209aad76198f2907a9 100644
|
| --- a/chrome/browser/user_style_sheet_watcher_unittest.cc
|
| +++ b/chrome/browser/user_style_sheet_watcher_unittest.cc
|
| @@ -1,14 +1,16 @@
|
| -// Copyright (c) 2010 The Chromium Authors. All rights reserved.
|
| +// 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/user_style_sheet_watcher.h"
|
|
|
| +#include "base/basictypes.h"
|
| #include "base/base64.h"
|
| #include "base/file_util.h"
|
| #include "base/message_loop.h"
|
| #include "base/scoped_temp_dir.h"
|
| #include "base/string_util.h"
|
| +#include "base/threading/thread.h"
|
| #include "content/browser/browser_thread.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| @@ -23,22 +25,27 @@ TEST(UserStyleSheetWatcherTest, StyleLoad) {
|
| ASSERT_TRUE(file_util::WriteFile(style_sheet_file,
|
| css_file_contents.data(), css_file_contents.length()));
|
|
|
| + MessageLoop loop(MessageLoop::TYPE_UI);
|
| + base::Thread io_thread("UserStyleSheetWatcherTestIOThread");
|
| + base::Thread::Options options(MessageLoop::TYPE_IO, 0);
|
| + io_thread.StartWithOptions(options);
|
| + BrowserThread browser_ui_thread(BrowserThread::UI, &loop);
|
| + BrowserThread browser_file_thread(BrowserThread::FILE,
|
| + io_thread.message_loop());
|
| +
|
| scoped_refptr<UserStyleSheetWatcher> style_sheet_watcher(
|
| new UserStyleSheetWatcher(dir.path()));
|
| - MessageLoop loop;
|
| - BrowserThread ui_thread(BrowserThread::UI, &loop);
|
| - BrowserThread file_thread(BrowserThread::FILE, &loop);
|
| style_sheet_watcher->Init();
|
|
|
| + io_thread.Stop();
|
| loop.RunAllPending();
|
|
|
| GURL result_url = style_sheet_watcher->user_style_sheet();
|
| std::string result = result_url.spec();
|
| std::string prefix = "data:text/css;charset=utf-8;base64,";
|
| - EXPECT_TRUE(StartsWithASCII(result, prefix, true));
|
| + ASSERT_TRUE(StartsWithASCII(result, prefix, true));
|
| result = result.substr(prefix.length());
|
| std::string decoded;
|
| - EXPECT_TRUE(base::Base64Decode(result, &decoded));
|
| - EXPECT_EQ(css_file_contents, decoded);
|
| - style_sheet_watcher = NULL;
|
| + ASSERT_TRUE(base::Base64Decode(result, &decoded));
|
| + ASSERT_EQ(css_file_contents, decoded);
|
| }
|
|
|