Refactor Autofill for out of process iframes (OOPIF).

components/password_manager/core/browser/password_generation_manager_unittest.cc

 // Copyright 2014 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 <vector>
 
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/pref_service.h"
 #include "base/prefs/testing_pref_service.h"
 #include "base/strings/utf_string_conversions.h"
@@ skipping 15 matching lines @@
 using base::ASCIIToUTF16;
 
 namespace password_manager {
 
 namespace {
 
 class TestPasswordManagerDriver : public StubPasswordManagerDriver {
  public:
   TestPasswordManagerDriver(PasswordManagerClient* client)
       : password_manager_(client),
-        password_generation_manager_(client),
-        password_autofill_manager_(client, NULL),
-        is_off_the_record_(false) {}
+        password_generation_manager_(client, this),
+        password_autofill_manager_(client, this, NULL) {}
   ~TestPasswordManagerDriver() override {}
 
   // PasswordManagerDriver implementation.
-  bool IsOffTheRecord() override { return is_off_the_record_; }
   PasswordGenerationManager* GetPasswordGenerationManager() override {
     return &password_generation_manager_;
   }
   PasswordManager* GetPasswordManager() override { return &password_manager_; }
   PasswordAutofillManager* GetPasswordAutofillManager() override {
     return &password_autofill_manager_;
   }
   void AccountCreationFormsFound(
       const std::vector<autofill::FormData>& forms) override {
     found_account_creation_forms_.insert(
         found_account_creation_forms_.begin(), forms.begin(), forms.end());
   }
 
   const std::vector<autofill::FormData>& GetFoundAccountCreationForms() {
     return found_account_creation_forms_;
   }
-  void set_is_off_the_record(bool is_off_the_record) {
-    is_off_the_record_ = is_off_the_record;
-  }
 
  private:
   PasswordManager password_manager_;
   PasswordGenerationManager password_generation_manager_;
   PasswordAutofillManager password_autofill_manager_;
   std::vector<autofill::FormData> found_account_creation_forms_;
-  bool is_off_the_record_;
 };
 
 class TestPasswordManagerClient : public StubPasswordManagerClient {
  public:
   TestPasswordManagerClient(scoped_ptr<PrefService> prefs)
       : prefs_(prefs.Pass()),
         store_(new TestPasswordStore),
         driver_(this),
-        is_sync_enabled_(false) {}
+        is_sync_enabled_(false),
+        is_off_the_record_(false) {}
 
   ~TestPasswordManagerClient() override {
     store_->Shutdown();
   }
 
+  bool IsOffTheRecord() override { return is_off_the_record_; }
   PasswordStore* GetPasswordStore() override { return store_.get(); }
   PrefService* GetPrefs() override { return prefs_.get(); }
-  PasswordManagerDriver* GetDriver() override { return &driver_; }
   bool IsPasswordSyncEnabled(CustomPassphraseState state) override {
     return is_sync_enabled_;
   }
 
   void set_is_password_sync_enabled(bool enabled) {
     is_sync_enabled_ = enabled;
   }
 
+  TestPasswordManagerDriver* test_driver() { return &driver_; }
+  void set_is_off_the_record(bool is_otr) { is_off_the_record_ = is_otr; }
+
  private:
   scoped_ptr<PrefService> prefs_;
   scoped_refptr<TestPasswordStore> store_;
   TestPasswordManagerDriver driver_;
   bool is_sync_enabled_;
+  bool is_off_the_record_;
 };
 
 // Unlike the base AutofillMetrics, exposes copy and assignment constructors,
 // which are handy for briefer test code.  The AutofillMetrics class is
 // stateless, so this is safe.
 class TestAutofillMetrics : public autofill::AutofillMetrics {
  public:
   TestAutofillMetrics() {}
   ~TestAutofillMetrics() override {}
 };
 
 }  // anonymous namespace
 
 class PasswordGenerationManagerTest : public testing::Test {
  protected:
   void SetUp() override {
     // Construct a PrefService and register all necessary prefs before handing
     // it off to |client_|, as the initialization flow of |client_| will
     // indirectly cause those prefs to be immediately accessed.
     scoped_ptr<TestingPrefServiceSimple> prefs(new TestingPrefServiceSimple());
     prefs->registry()->RegisterBooleanPref(prefs::kPasswordManagerSavingEnabled,
                                            true);
     client_.reset(new TestPasswordManagerClient(prefs.Pass()));
   }
 
   void TearDown() override { client_.reset(); }
 
   PasswordGenerationManager* GetGenerationManager() {
-    return client_->GetDriver()->GetPasswordGenerationManager();
+    return client_->test_driver()->GetPasswordGenerationManager();
   }
 
-  TestPasswordManagerDriver* GetTestDriver() {
-    return static_cast<TestPasswordManagerDriver*>(client_->GetDriver());
-  }
+  TestPasswordManagerDriver* GetTestDriver() { return client_->test_driver(); }
 
   bool IsGenerationEnabled() {
     return GetGenerationManager()->IsGenerationEnabled();
   }
 
   void DetectAccountCreationForms(
       const std::vector<autofill::FormStructure*>& forms) {
     GetGenerationManager()->DetectAccountCreationForms(forms);
   }
 
@@ skipping 67 matching lines @@
   DetectAccountCreationForms(forms);
   EXPECT_EQ(1u, GetTestDriver()->GetFoundAccountCreationForms().size());
   EXPECT_EQ(GURL("http://accounts.yahoo.com/"),
             GetTestDriver()->GetFoundAccountCreationForms()[0].origin);
 }
 
 TEST_F(PasswordGenerationManagerTest, UpdatePasswordSyncStateIncognito) {
   // Disable password manager by going incognito. Even though password
   // syncing is enabled, generation should still
   // be disabled.
-  GetTestDriver()->set_is_off_the_record(true);
+  client_->set_is_off_the_record(true);
   PrefService* prefs = client_->GetPrefs();
   prefs->SetBoolean(prefs::kPasswordManagerSavingEnabled, true);
   client_->set_is_password_sync_enabled(true);
 
   EXPECT_FALSE(IsGenerationEnabled());
 }
 
 }  // namespace password_manager