1 files changed, 6 insertions, 5 deletions
diff --git a/content/browser/shareable_file_reference_unittest.cc b/content/browser/shareable_file_reference_unittest.cc
index 84d035f..c2dae30 100644
--- a/content/browser/shareable_file_reference_unittest.cc
+++ b/content/browser/shareable_file_reference_unittest.cc
@@ -6,8 +6,9 @@
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
-#include "base/message_loop/message_loop_proxy.h"
 #include "base/run_loop.h"
+#include "base/single_thread_task_runner.h"
+#include "base/thread_task_runner_handle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using storage::ShareableFileReference;
@@ -16,8 +17,8 @@ namespace content {
 
 TEST(ShareableFileReferenceTest, TestReferences) {
   base::MessageLoop message_loop;
-  scoped_refptr<base::MessageLoopProxy> loop_proxy =
-      base::MessageLoopProxy::current();
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner =
+      base::ThreadTaskRunnerHandle::Get();
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
 
@@ -31,7 +32,7 @@ TEST(ShareableFileReferenceTest, TestReferences) {
   reference1 = ShareableFileReference::Get(file);
   EXPECT_FALSE(reference1.get());
   reference1 = ShareableFileReference::GetOrCreate(
-      file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, loop_proxy.get());
+      file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, task_runner.get());
   EXPECT_TRUE(reference1.get());
   EXPECT_TRUE(file == reference1->path());
 
@@ -40,7 +41,7 @@ TEST(ShareableFileReferenceTest, TestReferences) {
   reference2 = ShareableFileReference::Get(file);
   EXPECT_EQ(reference1.get(), reference2.get());
   reference2 = ShareableFileReference::GetOrCreate(
-      file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, loop_proxy.get());
+      file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, task_runner.get());
   EXPECT_EQ(reference1.get(), reference2.get());
 
   // Drop the first reference, the file and reference should still be there.