Add a triggered profile reset mechanism.

chrome/browser/profile_resetter/triggered_profile_resetter_win_unittest.cc

+// Copyright 2015 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/profile_resetter/triggered_profile_resetter.h"
+
+#include <stdint.h>
+
+#include "base/memory/scoped_ptr.h"
+#include "base/metrics/field_trial.h"
+#include "base/prefs/pref_service.h"
+#include "base/test/test_reg_util_win.h"
+#include "base/win/registry.h"
+#include "chrome/browser/prefs/browser_prefs.h"
+#include "chrome/browser/profile_resetter/triggered_profile_resetter_factory.h"
+#include "chrome/common/pref_names.h"
+#include "chrome/test/base/testing_profile.h"
+#include "components/variations/entropy_provider.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::win::RegKey;
+
+class TriggeredProfileResetterTest : public testing::Test {
+ protected:
+  void SetUp() {
+    override_manager_.OverrideRegistry(HKEY_CURRENT_USER);
+
+    // Activate the triggered reset field trial for these tests.
+    field_trial_list_.reset(
+        new base::FieldTrialList(new metrics::SHA1EntropyProvider("foo")));
+    base::FieldTrial* trial = base::FieldTrialList::CreateFieldTrial(
+        "TriggeredResetFieldTrial", "On");
+    trial->group();
+  }
+
+  content::TestBrowserThreadBundle thread_bundle_;
+  TestingProfile profile_;
+
+  void SetRegTimestampAndToolName(const base::string16& toolname,
+                                  FILETIME* file_time) {
+    RegKey trigger_key(HKEY_CURRENT_USER, kTriggeredResetRegistryPath,
+                       KEY_ALL_ACCESS);
+    ASSERT_TRUE(trigger_key.Valid());
+    FILETIME ft;
+    ::GetSystemTimeAsFileTime(&ft);
+    ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetTimestamp, &ft,
+                                       sizeof(ft), REG_QWORD) == ERROR_SUCCESS);
+    ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetToolName,
+                                       toolname.c_str()) == ERROR_SUCCESS);
+    if (file_time)
+      *file_time = ft;
+  }
+
+ private:
+  registry_util::RegistryOverrideManager override_manager_;
+  scoped_ptr<base::FieldTrialList> field_trial_list_;
+};
+
+TEST_F(TriggeredProfileResetterTest, HasResetTriggerAndClear) {
+  SetRegTimestampAndToolName(base::string16(), nullptr);
+  TriggeredProfileResetter triggered_profile_resetter(&profile_);
+  triggered_profile_resetter.Activate();
+  EXPECT_TRUE(triggered_profile_resetter.HasResetTrigger());
+  triggered_profile_resetter.ClearResetTrigger();
+  EXPECT_FALSE(triggered_profile_resetter.HasResetTrigger());
+}
+
+TEST_F(TriggeredProfileResetterTest, HasDuplicateResetTrigger) {
+  FILETIME ft = {};
+  SetRegTimestampAndToolName(base::string16(), &ft);
+  profile_.GetPrefs()->SetInt64(prefs::kLastProfileResetTimestamp,
+                                bit_cast<int64_t, FILETIME>(ft));
+
+  TriggeredProfileResetter triggered_profile_resetter(&profile_);
+  triggered_profile_resetter.Activate();
+  EXPECT_FALSE(triggered_profile_resetter.HasResetTrigger());
+}
+
+TEST_F(TriggeredProfileResetterTest, HasToolName) {
+  const wchar_t kToolName[] = L"ToolyMcTool";
+  SetRegTimestampAndToolName(kToolName, nullptr);
+  TriggeredProfileResetter triggered_profile_resetter(&profile_);
+  triggered_profile_resetter.Activate();
+  EXPECT_TRUE(triggered_profile_resetter.HasResetTrigger());
+  EXPECT_STREQ(kToolName,
+               triggered_profile_resetter.GetResetToolName().c_str());
+}
+
+TEST_F(TriggeredProfileResetterTest, HasLongToolName) {
+  const wchar_t kLongToolName[] =
+      L"ToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcTool"
+      L"ToolMcToolToolMcToolToolMcToolThisIsTheToolThatNeverEndsYesItGoesOnAnd"
+      L"OnMyFriend";
+  const wchar_t kExpectedToolName[] =
+      L"ToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcTool"
+      L"ToolMcToolToolMcToolToolMcTool";
+  SetRegTimestampAndToolName(kLongToolName, nullptr);
+  TriggeredProfileResetter triggered_profile_resetter(&profile_);
+  triggered_profile_resetter.Activate();
+  EXPECT_TRUE(triggered_profile_resetter.HasResetTrigger());
+  EXPECT_STREQ(kExpectedToolName,
+               triggered_profile_resetter.GetResetToolName().c_str());
+}