Index: components/previews/core/previews_io_data_unittest.cc |
diff --git a/components/previews/core/previews_io_data_unittest.cc b/components/previews/core/previews_io_data_unittest.cc |
index c2907d9ec691e33f42fdbd677d3c765bf337b024..1475e9590f8bca4d257a030242336447dd23fa1f 100644 |
--- a/components/previews/core/previews_io_data_unittest.cc |
+++ b/components/previews/core/previews_io_data_unittest.cc |
@@ -2,21 +2,20 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
#include "components/previews/core/previews_io_data.h" |
#include <memory> |
#include "base/memory/ptr_util.h" |
#include "base/memory/ref_counted.h" |
#include "base/message_loop/message_loop.h" |
-#include "base/run_loop.h" |
#include "base/single_thread_task_runner.h" |
#include "components/previews/core/previews_ui_service.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace previews { |
namespace { |
class TestPreviewsIOData : public PreviewsIOData { |
public: |
@@ -24,23 +23,24 @@ class TestPreviewsIOData : public PreviewsIOData { |
const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner, |
const scoped_refptr<base::SingleThreadTaskRunner>& ui_task_runner) |
: PreviewsIOData(io_task_runner, ui_task_runner), initialized_(false) {} |
~TestPreviewsIOData() override {} |
// Whether Initialize was called. |
bool initialized() { return initialized_; } |
private: |
// Set |initialized_| to true and use base class functionality. |
- void InitializeOnIOThread() override { |
+ void InitializeOnIOThread( |
+ std::unique_ptr<PreviewsOptOutStore> previews_opt_out_store) override { |
initialized_ = true; |
- PreviewsIOData::InitializeOnIOThread(); |
+ PreviewsIOData::InitializeOnIOThread(std::move(previews_opt_out_store)); |
} |
// Whether Initialize was called. |
bool initialized_; |
}; |
class PreviewsIODataTest : public testing::Test { |
public: |
PreviewsIODataTest() {} |
@@ -63,19 +63,19 @@ class PreviewsIODataTest : public testing::Test { |
private: |
std::unique_ptr<TestPreviewsIOData> io_data_; |
std::unique_ptr<PreviewsUIService> ui_service_; |
}; |
} // namespace |
TEST_F(PreviewsIODataTest, TestInitialization) { |
set_io_data(base::WrapUnique( |
new TestPreviewsIOData(loop_.task_runner(), loop_.task_runner()))); |
- set_ui_service( |
- base::WrapUnique(new PreviewsUIService(io_data(), loop_.task_runner()))); |
- base::RunLoop().RunUntilIdle(); |
+ set_ui_service(base::WrapUnique( |
+ new PreviewsUIService(io_data(), loop_.task_runner(), nullptr))); |
+ loop_.RunUntilIdle(); |
// After the outstanding posted tasks have run, |io_data_| should be fully |
// initialized. |
EXPECT_TRUE(io_data()->initialized()); |
} |
} // namespace previews |