Post-cleanup settings reset.

chrome/browser/safe_browsing/chrome_cleaner/chrome_cleaner_controller_win_unittest.cc

 // 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/safe_browsing/chrome_cleaner/chrome_cleaner_controller_win.h"
 
 #include <string>
 #include <tuple>
 #include <utility>
 
+#include "base/callback_helpers.h"
 #include "base/command_line.h"
 #include "base/run_loop.h"
 #include "base/task_scheduler/post_task.h"
 #include "base/test/multiprocess_test.h"
 #include "base/test/scoped_feature_list.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/metrics/chrome_metrics_service_accessor.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/safe_browsing/chrome_cleaner/mock_chrome_cleaner_process_win.h"
 #include "chrome/browser/safe_browsing/chrome_cleaner/reporter_runner_win.h"
 #include "chrome/browser/safe_browsing/chrome_cleaner/srt_field_trial_win.h"
+#include "chrome/test/base/testing_browser_process.h"
+#include "chrome/test/base/testing_profile.h"
+#include "chrome/test/base/testing_profile_manager.h"
 #include "components/chrome_cleaner/public/constants/constants.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/multiprocess_func_list.h"
 
 namespace safe_browsing {
 namespace {
 
 using ::testing::Combine;
 using ::testing::DoAll;
 using ::testing::InvokeWithoutArgs;
 using ::testing::SaveArg;
 using ::testing::StrictMock;
+using ::testing::UnorderedElementsAreArray;
 using ::testing::Values;
 using ::testing::_;
 using CrashPoint = MockChromeCleanerProcess::CrashPoint;
 using IdleReason = ChromeCleanerController::IdleReason;
 using State = ChromeCleanerController::State;
 using UserResponse = ChromeCleanerController::UserResponse;
 
 class MockChromeCleanerControllerObserver
     : public ChromeCleanerController::Observer {
  public:
@@ skipping 59 matching lines @@
     std::move(fetched_callback)
         .Run(base::FilePath(FILE_PATH_LITERAL("chrome_cleaner.exe")));
   }
 
   bool SafeBrowsingExtendedReportingScoutEnabled() override {
     return scout_enabled_;
   }
 
   bool IsMetricsAndCrashReportingEnabled() override { return metrics_enabled_; }
 
+  void TagForResetting(Profile* profile) override {
+    // This function should never be called by these tests.
+    EXPECT_TRUE(false);

[Nico, 2017/06/15 15:12:44]

Use ADD_FAILURE() (or FAIL() if you really wanted ASSERT_TRUE instead of
EXPECT_TRUE) instead of this

[ftirelo, 2017/06/15 17:28:05]

Done.

+  }
+
+  void ResetTaggedProfiles(std::vector<Profile*> profiles,
+                           base::OnceClosure continuation) override {
+    // This function should never be called by these tests.
+    EXPECT_TRUE(false);

[Nico, 2017/06/15 15:12:44]

likewise 

[ftirelo, 2017/06/15 17:28:05]

Done.

+  }
+
   // ChromeCleanerRunnerTestDelegate overrides.
 
   base::Process LaunchTestProcess(
       const base::CommandLine& command_line,
       const base::LaunchOptions& launch_options) override {
     command_line_ = command_line;
     // Return an invalid process.
     return base::Process();
   }
 
@@ skipping 71 matching lines @@
 // configurations and mocks the user's response.
 class ChromeCleanerControllerTest
     : public testing::TestWithParam<
           std::tuple<CleanerProcessStatus,
                      MockChromeCleanerProcess::CrashPoint,
                      UwsFoundStatus,
                      ChromeCleanerController::UserResponse>>,
       public ChromeCleanerRunnerTestDelegate,
       public ChromeCleanerControllerDelegate {
  public:
+  ChromeCleanerControllerTest() = default;
   ~ChromeCleanerControllerTest() override {}
 
   void SetUp() override {
     std::tie(process_status_, crash_point_, uws_found_status_, user_response_) =
         GetParam();
 
     cleaner_process_options_.SetDoFindUws(uws_found_status_ !=
                                           UwsFoundStatus::kNoUwsFound);
     cleaner_process_options_.set_reboot_required(
         uws_found_status_ == UwsFoundStatus::kUwsFoundRebootRequired);
@@ skipping 32 matching lines @@
     // fixture.
     return false;
   }
 
   bool IsMetricsAndCrashReportingEnabled() override {
     // Returning an arbitrary value since this is not being tested in this
     // fixture.
     return false;
   }
 
+  void TagForResetting(Profile* profile) override {
+    profiles_tagged_.push_back(profile);
+  }
+
+  void ResetTaggedProfiles(std::vector<Profile*> profiles,
+                           base::OnceClosure continuation) override {
+    for (Profile* profile : profiles)
+      profiles_to_reset_if_tagged_.push_back(profile);
+    std::move(continuation).Run();
+  }
+
   // ChromeCleanerRunnerTestDelegate overrides.
 
   base::Process LaunchTestProcess(
       const base::CommandLine& command_line,
       const base::LaunchOptions& launch_options) override {
     if (process_status_ != CleanerProcessStatus::kFetchSuccessValidProcess)
       return base::Process();
 
     // Add switches and program name that the test process needs for the multi
     // process tests.
@@ skipping 37 matching lines @@
            crash_point_ != CrashPoint::kAfterRequestSent &&
            user_response_ == UserResponse::kAccepted;
   }
 
   bool ExpectedOnRebootRequiredCalled() {
     return ExpectedFinalState() == State::kRebootRequired;
   }
 
   bool ExpectedUwsFound() { return ExpectedOnInfectedCalled(); }
 
+  bool ExpectedToTagProfile() {
+    return process_status_ == CleanerProcessStatus::kFetchSuccessValidProcess &&
+           (crash_point_ == CrashPoint::kNone ||
+            crash_point_ == CrashPoint::kAfterResponseReceived) &&
+           (uws_found_status_ == UwsFoundStatus::kUwsFoundNoRebootRequired ||
+            uws_found_status_ == UwsFoundStatus::kUwsFoundRebootRequired) &&
+           user_response_ == UserResponse::kAccepted;
+  }
+
+  bool ExpectedToResetSettings() {
+    return process_status_ == CleanerProcessStatus::kFetchSuccessValidProcess &&
+           crash_point_ == CrashPoint::kNone &&
+           uws_found_status_ == UwsFoundStatus::kUwsFoundNoRebootRequired &&
+           user_response_ == UserResponse::kAccepted;
+  }
+
   ChromeCleanerController::IdleReason ExpectedIdleReason() {
     EXPECT_EQ(ExpectedFinalState(), State::kIdle);
 
     if (process_status_ != CleanerProcessStatus::kFetchSuccessValidProcess ||
         crash_point_ == CrashPoint::kOnStartup ||
         crash_point_ == CrashPoint::kAfterConnection) {
       return IdleReason::kScanningFailed;
     }
 
     if (uws_found_status_ == UwsFoundStatus::kNoUwsFound)
@@ skipping 23 matching lines @@
 
   CleanerProcessStatus process_status_;
   MockChromeCleanerProcess::CrashPoint crash_point_;
   UwsFoundStatus uws_found_status_;
   ChromeCleanerController::UserResponse user_response_;
 
   MockChromeCleanerProcess::Options cleaner_process_options_;
 
   StrictMock<MockChromeCleanerControllerObserver> mock_observer_;
   ChromeCleanerController* controller_;
+
+  std::vector<Profile*> profiles_tagged_;
+  std::vector<Profile*> profiles_to_reset_if_tagged_;
 };
 
 MULTIPROCESS_TEST_MAIN(MockChromeCleanerProcessMain) {
   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
   MockChromeCleanerProcess::Options options;
   EXPECT_TRUE(MockChromeCleanerProcess::Options::FromCommandLine(*command_line,
                                                                  &options));
 
   std::string chrome_mojo_pipe_token = command_line->GetSwitchValueASCII(
       chrome_cleaner::kChromeMojoPipeTokenSwitch);
   EXPECT_FALSE(chrome_mojo_pipe_token.empty());
 
   // Since failures in any of the above calls to EXPECT_*() do not actually fail
   // the test, we need to ensure that we return an exit code to indicate test
   // failure in such cases.
   if (::testing::Test::HasFailure())
     return MockChromeCleanerProcess::kInternalTestFailureExitCode;
 
   MockChromeCleanerProcess mock_cleaner_process(options,
                                                 chrome_mojo_pipe_token);
   return mock_cleaner_process.Run();
 }
 
 TEST_P(ChromeCleanerControllerTest, WithMockCleanerProcess) {
+  TestingProfileManager profile_manager(TestingBrowserProcess::GetGlobal());
+  ASSERT_TRUE(profile_manager.SetUp());
+
+  constexpr char kTestProfileName1[] = "Test 1";
+  constexpr char kTestProfileName2[] = "Test 2";
+
+  Profile* profile1 = profile_manager.CreateTestingProfile(kTestProfileName1);
+  ASSERT_TRUE(profile1);
+  Profile* profile2 = profile_manager.CreateTestingProfile(kTestProfileName2);
+  ASSERT_TRUE(profile2);
+
+  const int num_profiles =
+      profile_manager.profile_manager()->GetNumberOfProfiles();
+  ASSERT_EQ(2, num_profiles);
+
   EXPECT_CALL(mock_observer_, OnIdle(_)).Times(1);
   controller_->AddObserver(&mock_observer_);
   EXPECT_EQ(controller_->state(), State::kIdle);
 
   EXPECT_CALL(mock_observer_, OnScanning()).Times(1);
   controller_->Scan(GetInvocationWithPromptTrigger());
   EXPECT_EQ(controller_->state(), State::kScanning);
 
   base::RunLoop run_loop;
 
   std::set<base::FilePath> files_to_delete_on_infected;
   std::set<base::FilePath> files_to_delete_on_cleaning;
 
   if (ExpectedOnIdleCalled()) {
     EXPECT_CALL(mock_observer_, OnIdle(ExpectedIdleReason()))
         .WillOnce(
             InvokeWithoutArgs([&run_loop]() { run_loop.QuitWhenIdle(); }));
   }
 
   if (ExpectedOnInfectedCalled()) {
     EXPECT_CALL(mock_observer_, OnInfected(_))
         .WillOnce(DoAll(SaveArg<0>(&files_to_delete_on_infected),
-                        InvokeWithoutArgs([this]() {
-                          controller_->ReplyWithUserResponse(user_response_);
+                        InvokeWithoutArgs([this, profile1]() {
+                          controller_->ReplyWithUserResponse(profile1,
+                                                             user_response_);
                         })));
   }
 
   if (ExpectedOnCleaningCalled()) {
     EXPECT_CALL(mock_observer_, OnCleaning(_))
         .WillOnce(SaveArg<0>(&files_to_delete_on_cleaning));
   }
 
   if (ExpectedOnRebootRequiredCalled()) {
     EXPECT_CALL(mock_observer_, OnRebootRequired())
         .WillOnce(
             InvokeWithoutArgs([&run_loop]() { run_loop.QuitWhenIdle(); }));
   }
 
   // Assert here that we expect at least one of OnIdle or OnRebootRequired to be
   // called, since otherwise, the test is set up incorrectly and is expected to
   // never stop.
   ASSERT_TRUE(ExpectedOnIdleCalled() || ExpectedOnRebootRequiredCalled());
   run_loop.Run();
 
   EXPECT_EQ(controller_->state(), ExpectedFinalState());
   EXPECT_EQ(!files_to_delete_on_infected.empty(), ExpectedUwsFound());
   EXPECT_EQ(!files_to_delete_on_cleaning.empty(),
             ExpectedUwsFound() && ExpectedOnCleaningCalled());
   if (!files_to_delete_on_infected.empty() &&
       !files_to_delete_on_cleaning.empty()) {
     EXPECT_EQ(files_to_delete_on_infected, files_to_delete_on_cleaning);
   }
 
+  std::vector<Profile*> expected_tagged;
+  if (ExpectedToTagProfile())
+    expected_tagged.push_back(profile1);
+  EXPECT_THAT(expected_tagged, UnorderedElementsAreArray(profiles_tagged_));
+
+  std::vector<Profile*> expected_reset_if_tagged;
+  if (ExpectedToResetSettings()) {
+    expected_reset_if_tagged.push_back(profile1);
+    expected_reset_if_tagged.push_back(profile2);
+  }
+  EXPECT_THAT(expected_reset_if_tagged,
+              UnorderedElementsAreArray(profiles_to_reset_if_tagged_));
+
   controller_->RemoveObserver(&mock_observer_);
 }
 
 INSTANTIATE_TEST_CASE_P(
     All,
     ChromeCleanerControllerTest,
     Combine(Values(CleanerProcessStatus::kFetchFailure,
                    CleanerProcessStatus::kFetchSuccessInvalidProcess,
                    CleanerProcessStatus::kFetchSuccessValidProcess),
             Values(CrashPoint::kNone,
                    CrashPoint::kOnStartup,
                    CrashPoint::kAfterConnection,
                    // CrashPoint::kAfterRequestSent is not used because we
                    // cannot ensure the order between the Mojo request being
                    // received by Chrome and the connection being lost.
                    CrashPoint::kAfterResponseReceived),
             Values(UwsFoundStatus::kNoUwsFound,
                    UwsFoundStatus::kUwsFoundRebootRequired,
                    UwsFoundStatus::kUwsFoundNoRebootRequired),
             Values(UserResponse::kAccepted,
                    UserResponse::kDenied,
                    UserResponse::kDismissed)));
 
 }  // namespace
 }  // namespace safe_browsing