1 // Copyright 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "chrome/browser/diagnostics/diagnostics_controller.h" 6 7 #include "base/command_line.h" 8 #include "base/file_util.h" 9 #include "base/files/scoped_temp_dir.h" 10 #include "base/memory/scoped_ptr.h" 11 #include "base/path_service.h" 12 #include "chrome/browser/diagnostics/diagnostics_model.h" 13 #include "chrome/browser/diagnostics/diagnostics_writer.h" 14 #include "chrome/browser/diagnostics/sqlite_diagnostics.h" 15 #include "chrome/common/chrome_paths.h" 16 #include "chrome/common/chrome_switches.h" 17 #include "chromeos/chromeos_constants.h" 18 #include "testing/gtest/include/gtest/gtest.h" 19 20 namespace diagnostics { 21 22 // Basic harness to acquire and release the required temporary environment to 23 // run a test in. 24 class DiagnosticsControllerTest : public testing::Test { 25 protected: 26 DiagnosticsControllerTest() : cmdline_(CommandLine::NO_PROGRAM) {} 27 28 virtual ~DiagnosticsControllerTest() {} 29 30 virtual void SetUp() { 31 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 32 base::FilePath test_data; 33 PathService::Get(chrome::DIR_TEST_DATA, &test_data); 34 test_data = test_data.Append(FILE_PATH_LITERAL("diagnostics")); 35 test_data = test_data.Append(FILE_PATH_LITERAL("user")); 36 base::CopyDirectory(test_data, temp_dir_.path(), true); 37 profile_dir_ = temp_dir_.path().Append(FILE_PATH_LITERAL("user")); 38 39 #if defined(OS_CHROMEOS) 40 // Redirect the home dir to the profile directory. We have to do this 41 // because NSS uses the HOME directory to find where to store it's database, 42 // so that's where the diagnostics and recovery code looks for it. 43 44 // Preserve existing home directory setting, if any. 45 const char* old_home_dir = ::getenv("HOME"); 46 if (old_home_dir) 47 old_home_dir_ = old_home_dir; 48 else 49 old_home_dir_.clear(); 50 ::setenv("HOME", profile_dir_.value().c_str(), 1); 51 #endif 52 53 cmdline_ = CommandLine(CommandLine::NO_PROGRAM); 54 cmdline_.AppendSwitchPath(switches::kUserDataDir, profile_dir_); 55 cmdline_.AppendSwitch(switches::kDiagnostics); 56 cmdline_.AppendSwitch(switches::kDiagnosticsRecovery); 57 writer_.reset(); 58 // Use this writer instead, if you want to see the diagnostics output. 59 // writer_.reset(new DiagnosticsWriter(DiagnosticsWriter::MACHINE)); 60 } 61 62 virtual void TearDown() { 63 DiagnosticsController::GetInstance()->ClearResults(); 64 #if defined(OS_CHROMEOS) 65 if (!old_home_dir_.empty()) { 66 ::setenv("HOME", old_home_dir_.c_str(), 1); 67 } else { 68 ::unsetenv("HOME"); 69 } 70 old_home_dir_.clear(); 71 #endif 72 } 73 74 void CorruptDataFile(const base::FilePath& path) { 75 // Just write some random characters into the file tInvaludUsero "corrupt" 76 // it. 77 const char bogus_data[] = "wwZ2uNYNuyUVzFbDm3DL"; 78 file_util::WriteFile(path, bogus_data, arraysize(bogus_data)); 79 } 80 81 scoped_ptr<DiagnosticsModel> model_; 82 CommandLine cmdline_; 83 base::ScopedTempDir temp_dir_; 84 scoped_ptr<DiagnosticsWriter> writer_; 85 base::FilePath profile_dir_; 86 87 #if defined(OS_CHROMEOS) 88 std::string old_home_dir_; 89 #endif 90 91 DISALLOW_COPY_AND_ASSIGN(DiagnosticsControllerTest); 92 }; 93 94 TEST_F(DiagnosticsControllerTest, Diagnostics) { 95 DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); 96 EXPECT_TRUE(DiagnosticsController::GetInstance()->HasResults()); 97 const DiagnosticsModel& results = 98 DiagnosticsController::GetInstance()->GetResults(); 99 EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); 100 EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); 101 for (int i = 0; i < results.GetTestRunCount(); ++i) { 102 const DiagnosticsModel::TestInfo& info(results.GetTest(i)); 103 EXPECT_EQ(DiagnosticsModel::TEST_OK, info.GetResult()) << "Test: " 104 << info.GetId(); 105 } 106 } 107 108 TEST_F(DiagnosticsControllerTest, RecoverAllOK) { 109 DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); 110 DiagnosticsController::GetInstance()->RunRecovery(cmdline_, writer_.get()); 111 EXPECT_TRUE(DiagnosticsController::GetInstance()->HasResults()); 112 const DiagnosticsModel& results = 113 DiagnosticsController::GetInstance()->GetResults(); 114 EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); 115 EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); 116 for (int i = 0; i < results.GetTestRunCount(); ++i) { 117 const DiagnosticsModel::TestInfo& info(results.GetTest(i)); 118 EXPECT_EQ(DiagnosticsModel::RECOVERY_OK, info.GetResult()) << "Test: " 119 << info.GetId(); 120 } 121 } 122 123 #if defined(OS_CHROMEOS) 124 TEST_F(DiagnosticsControllerTest, RecoverFromNssCertDbFailure) { 125 base::FilePath db_path = profile_dir_.Append(chromeos::kNssCertDbPath); 126 EXPECT_TRUE(base::PathExists(db_path)); 127 CorruptDataFile(db_path); 128 DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); 129 ASSERT_TRUE(DiagnosticsController::GetInstance()->HasResults()); 130 const DiagnosticsModel& results = 131 DiagnosticsController::GetInstance()->GetResults(); 132 EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); 133 EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); 134 135 const DiagnosticsModel::TestInfo* info = NULL; 136 EXPECT_TRUE(results.GetTestInfo(kSQLiteIntegrityNSSCertTest, &info)); 137 EXPECT_EQ(DiagnosticsModel::TEST_FAIL_CONTINUE, info->GetResult()); 138 EXPECT_EQ(DIAG_SQLITE_ERROR_HANDLER_CALLED, info->GetOutcomeCode()); 139 140 DiagnosticsController::GetInstance()->RunRecovery(cmdline_, writer_.get()); 141 EXPECT_EQ(DiagnosticsModel::RECOVERY_OK, info->GetResult()); 142 EXPECT_FALSE(base::PathExists(db_path)); 143 } 144 145 TEST_F(DiagnosticsControllerTest, RecoverFromNssKeyDbFailure) { 146 base::FilePath db_path = profile_dir_.Append(chromeos::kNssKeyDbPath); 147 EXPECT_TRUE(base::PathExists(db_path)); 148 CorruptDataFile(db_path); 149 DiagnosticsController::GetInstance()->Run(cmdline_, writer_.get()); 150 ASSERT_TRUE(DiagnosticsController::GetInstance()->HasResults()); 151 const DiagnosticsModel& results = 152 DiagnosticsController::GetInstance()->GetResults(); 153 EXPECT_EQ(results.GetTestRunCount(), results.GetTestAvailableCount()); 154 EXPECT_EQ(DiagnosticsModel::kDiagnosticsTestCount, results.GetTestRunCount()); 155 156 const DiagnosticsModel::TestInfo* info = NULL; 157 EXPECT_TRUE(results.GetTestInfo(kSQLiteIntegrityNSSKeyTest, &info)); 158 EXPECT_EQ(DiagnosticsModel::TEST_FAIL_CONTINUE, info->GetResult()); 159 EXPECT_EQ(DIAG_SQLITE_ERROR_HANDLER_CALLED, info->GetOutcomeCode()); 160 161 DiagnosticsController::GetInstance()->RunRecovery(cmdline_, writer_.get()); 162 EXPECT_EQ(DiagnosticsModel::RECOVERY_OK, info->GetResult()); 163 EXPECT_FALSE(base::PathExists(db_path)); 164 } 165 #endif 166 167 } // namespace diagnostics 168
