| // Copyright 2017 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_inspector_win.h" |
| |
| #include <memory> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/bind.h" |
| #include "base/environment.h" |
| #include "base/files/file_path.h" |
| #include "base/macros.h" |
| #include "base/test/scoped_task_environment.h" |
| #include "base/test/test_simple_task_runner.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace { |
| |
| base::FilePath GetKernel32DllFilePath() { |
| std::unique_ptr<base::Environment> env = base::Environment::Create(); |
| std::string sysroot; |
| EXPECT_TRUE(env->GetVar("SYSTEMROOT", &sysroot)); |
| |
| base::FilePath path = |
| base::FilePath::FromUTF8Unsafe(sysroot).Append(L"system32\\kernel32.dll"); |
| |
| return path; |
| } |
| |
| class ModuleInspectorTest : public testing::Test { |
| protected: |
| ModuleInspectorTest() |
| : module_inspector_(base::Bind(&ModuleInspectorTest::OnModuleInspected, |
| base::Unretained(this))) {} |
| |
| void AddModules(const std::vector<ModuleInfoKey>& modules) { |
| for (const auto& module : modules) |
| module_inspector_.AddModule(module); |
| } |
| |
| // Callback for ModuleInspector. |
| void OnModuleInspected( |
| const ModuleInfoKey& module_key, |
| std::unique_ptr<ModuleInspectionResult> inspection_result) { |
| inspected_modules_.push_back(std::move(inspection_result)); |
| } |
| |
| const std::vector<std::unique_ptr<ModuleInspectionResult>>& |
| inspected_modules() { |
| return inspected_modules_; |
| } |
| |
| protected: |
| // Must be before the ModuleInspector. |
| base::test::ScopedTaskEnvironment scoped_task_environment_; |
| |
| private: |
| ModuleInspector module_inspector_; |
| |
| std::vector<std::unique_ptr<ModuleInspectionResult>> inspected_modules_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ModuleInspectorTest); |
| }; |
| |
| } // namespace |
| |
| TEST_F(ModuleInspectorTest, OneModule) { |
| AddModules({ |
| {GetKernel32DllFilePath(), 0, 0, 1}, |
| }); |
| |
| scoped_task_environment_.RunUntilIdle(); |
| |
| EXPECT_EQ(1u, inspected_modules().size()); |
| |
| EXPECT_TRUE(inspected_modules().front()); |
| } |
| |
| TEST_F(ModuleInspectorTest, MultipleModules) { |
| AddModules({ |
| {base::FilePath(), 0, 0, 1}, |
| {base::FilePath(), 0, 0, 2}, |
| {base::FilePath(), 0, 0, 3}, |
| {base::FilePath(), 0, 0, 4}, |
| {base::FilePath(), 0, 0, 5}, |
| }); |
| |
| scoped_task_environment_.RunUntilIdle(); |
| |
| EXPECT_EQ(5u, inspected_modules().size()); |
| for (const auto& inspection_result : inspected_modules()) |
| EXPECT_TRUE(inspection_result); |
| } |