| // Copyright 2013 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/diagnostics/diagnostics_controller.h" |
| |
| #include <memory> |
| |
| #include "base/base_paths.h" |
| #include "base/command_line.h" |
| #include "base/files/file_util.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/macros.h" |
| #include "base/path_service.h" |
| #include "build/build_config.h" |
| #include "chrome/browser/diagnostics/diagnostics_model.h" |
| #include "chrome/browser/diagnostics/diagnostics_writer.h" |
| #include "chrome/browser/diagnostics/sqlite_diagnostics.h" |
| #include "chrome/common/chrome_paths.h" |
| #include "chrome/common/chrome_switches.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #if defined(OS_CHROMEOS) |
| #include "chromeos/chromeos_constants.h" |
| #endif // defined(OS_CHROMEOS) |
| |
| namespace diagnostics { |
| |
| // Basic harness to acquire and release the required temporary environment to |
| // run a test in. |
| class DiagnosticsControllerTest : public testing::Test { |
| protected: |
| DiagnosticsControllerTest() : cmdline_(base::CommandLine::NO_PROGRAM) {} |
| |
| ~DiagnosticsControllerTest() override {} |
| |
| void SetUp() override { |
| ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| base::FilePath test_data; |
| base::PathService::Get(chrome::DIR_TEST_DATA, &test_data); |
| test_data = test_data.Append(FILE_PATH_LITERAL("diagnostics")); |
| test_data = test_data.Append(FILE_PATH_LITERAL("user")); |
| base::CopyDirectory(test_data, temp_dir_.GetPath(), true); |
| profile_dir_ = temp_dir_.GetPath().Append(FILE_PATH_LITERAL("user")); |
| |
| #if defined(OS_CHROMEOS) |
| // Redirect the home dir to the profile directory. We have to do this |
| // because NSS uses the HOME directory to find where to store it's database, |
| // so that's where the diagnostics and recovery code looks for it. |
| base::PathService::Get(base::DIR_HOME, &old_home_dir_); |
| base::PathService::Override(base::DIR_HOME, profile_dir_); |
| #endif |
| |
| cmdline_ = base::CommandLine(base::CommandLine::NO_PROGRAM); |
| cmdline_.AppendSwitchPath(switches::kUserDataDir, profile_dir_); |
| cmdline_.AppendSwitch(switches::kDiagnostics); |
| cmdline_.AppendSwitch(switches::kDiagnosticsRecovery); |
| writer_.reset(); |
| // Use this writer instead, if you want to see the diagnostics output. |
| // writer_.reset(new DiagnosticsWriter(DiagnosticsWriter::MACHINE)); |
| } |
| |
| void TearDown() override { |
| DiagnosticsController::GetInstance()->ClearResults(); |
| #if defined(OS_CHROMEOS) |
| base::PathService::Override(base::DIR_HOME, old_home_dir_); |
| old_home_dir_.clear(); |
| #endif |
| } |
| |
| void CorruptDataFile(const base::FilePath& path) { |
| // Just write some random characters into the file tInvaludUsero "corrupt" |
| // it. |
| const char bogus_data[] = "wwZ2uNYNuyUVzFbDm3DL"; |
| base::WriteFile(path, bogus_data, arraysize(bogus_data)); |
| } |
| |
| std::unique_ptr<DiagnosticsModel> model_; |
| base::CommandLine cmdline_; |
| base::ScopedTempDir temp_dir_; |
| std::unique_ptr<DiagnosticsWriter> writer_; |
| base::FilePath profile_dir_; |
| |
| #if defined(OS_CHROMEOS) |
| base::FilePath old_home_dir_; |
| #endif |
| |
| DISALLOW_COPY_AND_ASSIGN(DiagnosticsControllerTest); |
| }; |
| |
| TEST_F(DiagnosticsControllerTest, Diagnostics) { |
| DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); |
| EXPECT_TRUE(DiagnosticsController::GetInstance()->HasResults()); |
| const DiagnosticsModel& results = |
| DiagnosticsController::GetInstance()->GetResults(); |
| EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); |
| EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); |
| for (int i = 0; i < results.GetTestRunCount(); ++i) { |
| const DiagnosticsModel::TestInfo& info(results.GetTest(i)); |
| EXPECT_EQ(DiagnosticsModel::TEST_OK, info.GetResult()) << "Test: " |
| << info.GetName(); |
| } |
| } |
| |
| TEST_F(DiagnosticsControllerTest, RecoverAllOK) { |
| DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); |
| DiagnosticsController::GetInstance()->RunRecovery(cmdline_, writer_.get()); |
| EXPECT_TRUE(DiagnosticsController::GetInstance()->HasResults()); |
| const DiagnosticsModel& results = |
| DiagnosticsController::GetInstance()->GetResults(); |
| EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); |
| EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); |
| for (int i = 0; i < results.GetTestRunCount(); ++i) { |
| const DiagnosticsModel::TestInfo& info(results.GetTest(i)); |
| EXPECT_EQ(DiagnosticsModel::RECOVERY_OK, info.GetResult()) |
| << "Test: " << info.GetName(); |
| } |
| } |
| |
| #if defined(OS_CHROMEOS) |
| TEST_F(DiagnosticsControllerTest, RecoverFromNssCertDbFailure) { |
| base::FilePath db_path = profile_dir_.Append(chromeos::kNssCertDbPath); |
| EXPECT_TRUE(base::PathExists(db_path)); |
| CorruptDataFile(db_path); |
| DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); |
| ASSERT_TRUE(DiagnosticsController::GetInstance()->HasResults()); |
| const DiagnosticsModel& results = |
| DiagnosticsController::GetInstance()->GetResults(); |
| EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); |
| EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); |
| |
| const DiagnosticsModel::TestInfo* info = NULL; |
| EXPECT_TRUE( |
| results.GetTestInfo(DIAGNOSTICS_SQLITE_INTEGRITY_NSS_CERT_TEST, &info)); |
| EXPECT_EQ(DiagnosticsModel::TEST_FAIL_CONTINUE, info->GetResult()); |
| EXPECT_EQ(DIAG_SQLITE_ERROR_HANDLER_CALLED, info->GetOutcomeCode()); |
| |
| DiagnosticsController::GetInstance()->RunRecovery(cmdline_, writer_.get()); |
| EXPECT_EQ(DiagnosticsModel::RECOVERY_OK, info->GetResult()); |
| EXPECT_FALSE(base::PathExists(db_path)); |
| } |
| |
| TEST_F(DiagnosticsControllerTest, RecoverFromNssKeyDbFailure) { |
| base::FilePath db_path = profile_dir_.Append(chromeos::kNssKeyDbPath); |
| EXPECT_TRUE(base::PathExists(db_path)); |
| CorruptDataFile(db_path); |
| DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); |
| ASSERT_TRUE(DiagnosticsController::GetInstance()->HasResults()); |
| const DiagnosticsModel& results = |
| DiagnosticsController::GetInstance()->GetResults(); |
| EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); |
| EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); |
| |
| const DiagnosticsModel::TestInfo* info = NULL; |
| EXPECT_TRUE( |
| results.GetTestInfo(DIAGNOSTICS_SQLITE_INTEGRITY_NSS_KEY_TEST, &info)); |
| EXPECT_EQ(DiagnosticsModel::TEST_FAIL_CONTINUE, info->GetResult()); |
| EXPECT_EQ(DIAG_SQLITE_ERROR_HANDLER_CALLED, info->GetOutcomeCode()); |
| |
| DiagnosticsController::GetInstance()->RunRecovery(cmdline_, writer_.get()); |
| EXPECT_EQ(DiagnosticsModel::RECOVERY_OK, info->GetResult()); |
| EXPECT_FALSE(base::PathExists(db_path)); |
| } |
| #endif |
| |
| } // namespace diagnostics |