Chromium Code Reviews| Index: chrome/browser/conflicts/module_database_impl_win_unittest.cc |
| diff --git a/chrome/browser/conflicts/module_database_impl_win_unittest.cc b/chrome/browser/conflicts/module_database_impl_win_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..aa6bcdf5847b8b3833b155ced0aa64000307f587 |
| --- /dev/null |
| +++ b/chrome/browser/conflicts/module_database_impl_win_unittest.cc |
| @@ -0,0 +1,68 @@ |
| +// Copyright 2016 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/conflicts/module_database_impl_win.h" |
| + |
| +#include "base/message_loop/message_loop.h" |
| +#include "chrome/browser/conflicts/module_database_win.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +namespace { |
| + |
| +// Factory for mojom::ModuleEvents. |
| +mojom::ModuleEventPtr ModuleEvent(mojom::ModuleEventType type, |
| + const char* path, |
| + uint64_t address, |
| + uint32_t size) { |
| + mojom::ModuleEvent event; |
| + event.event_type = type; |
| + event.module_path = std::string(path); |
| + event.load_address = address; |
| + event.size = size; |
| + return event.Clone(); |
| +} |
| + |
| +} // namespace |
| + |
| +class ModuleDatabaseImplTest : public testing::Test { |
| + public: |
| + ModuleDatabaseImplTest() |
| + : message_loop_(new base::MessageLoop), |
| + module_database_(new ::ModuleDatabase(message_loop_->task_runner())), |
| + module_database_impl_(new ModuleDatabaseImpl(module_database_.get())) {} |
| + |
| + const ModuleDatabase::ModuleSet& modules() { |
| + return module_database_->modules_; |
| + } |
| + |
| + const ModuleDatabase::ProcessSet& processes() { |
| + return module_database_->processes_; |
| + } |
| + |
| + std::unique_ptr<base::MessageLoop> message_loop_; |
|
sky
2016/12/14 21:58:08
protected:
chrisha
2016/12/19 20:15:36
Done.
|
| + std::unique_ptr<ModuleDatabase> module_database_; |
| + std::unique_ptr<ModuleDatabaseImpl> module_database_impl_; |
| +}; |
|
sky
2016/12/14 21:58:08
private: DISALLOW...
chrisha
2016/12/19 20:15:36
Done.
|
| + |
| +TEST_F(ModuleDatabaseImplTest, CallsForwardedAsExpected) { |
| + EXPECT_EQ(module_database_.get(), module_database_impl_->module_database()); |
| + EXPECT_EQ(0u, module_database_impl_->process_id()); |
| + EXPECT_EQ(0u, modules().size()); |
| + EXPECT_EQ(0u, processes().size()); |
| + |
| + module_database_impl_->OnProcessStarted(1234, content::PROCESS_TYPE_BROWSER); |
| + EXPECT_EQ(1234u, module_database_impl_->process_id()); |
| + EXPECT_EQ(0u, modules().size()); |
| + EXPECT_EQ(1u, processes().size()); |
| + |
| + module_database_impl_->OnModuleEvent( |
| + ModuleEvent(mojom::ModuleEventType::MODULE_LOADED, "dummy.dll", |
| + 0xBAADF00D, 20 * 4096)); |
| + EXPECT_EQ(1u, modules().size()); |
| + EXPECT_EQ(1u, processes().size()); |
| + |
| + module_database_impl_->OnProcessEnded(); |
| + EXPECT_EQ(1u, modules().size()); |
| + EXPECT_EQ(0u, processes().size()); |
| +} |