Ignore ajax on specified pages.

components/autofill/browser/autofill_manager_unittest.cc

 // Copyright (c) 2012 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 <algorithm>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 610 matching lines @@
         new AutocheckoutPageMetaData());
     start_of_flow->current_page_number = 0;
     start_of_flow->total_pages = 3;
     WebElementDescriptor* proceed_element =
         &start_of_flow->proceed_element_descriptor;
     proceed_element->descriptor = "#foo";
     proceed_element->retrieval_method = WebElementDescriptor::ID;
     autocheckout_manager()->OnLoadedPageMetaData(start_of_flow.Pass());
   }
 
+  // Set autocheckout manager's page meta data to first page on Autocheckout
+  // flow.
+  void MarkAsFirstPageInAutocheckoutFlowIgnoringAjax() {
+    scoped_ptr<AutocheckoutPageMetaData> start_of_flow(
+        new AutocheckoutPageMetaData());
+    start_of_flow->current_page_number = 0;
+    start_of_flow->total_pages = 3;
+    start_of_flow->ignore_ajax = true;
+    WebElementDescriptor* proceed_element =
+        &start_of_flow->proceed_element_descriptor;
+    proceed_element->descriptor = "#foo";
+    proceed_element->retrieval_method = WebElementDescriptor::ID;
+    autocheckout_manager()->OnLoadedPageMetaData(start_of_flow.Pass());
+  }
+
  private:
   // Weak reference.
   TestPersonalDataManager* personal_data_;
 
   bool autofill_enabled_;
   std::vector<std::pair<WebFormElement::AutocompleteResult, FormData> >
       request_autocomplete_results_;
 
   scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
 
@@ skipping 2535 matching lines @@
   autofill_manager_->OnRequestAutocomplete(FormData(),
                                            GURL());
 
   EXPECT_EQ(1U, autofill_manager_->request_autocomplete_results().size());
   EXPECT_EQ(WebFormElement::AutocompleteResultErrorDisabled,
             autofill_manager_->request_autocomplete_results()[0].first);
 }
 
 namespace {
 
-class MockAutofillManagerDelegate
-    : public autofill::TestAutofillManagerDelegate {
+class MockAutofillManagerDelegate : public TestAutofillManagerDelegate {
  public:
-  MockAutofillManagerDelegate() {}
+  MockAutofillManagerDelegate()
+      : autocheckout_bubble_shown_(false) {}
+
   virtual ~MockAutofillManagerDelegate() {}
 
-  MOCK_METHOD3(ShowAutocheckoutBubble,
-               void(const gfx::RectF& bounds,
-                    bool is_google_user,
-                    const base::Callback<void(bool)>& callback));
+  virtual void ShowAutocheckoutBubble(
+      const gfx::RectF& bounds,
+      bool is_google_user,
+      const base::Callback<void(bool)>& callback) OVERRIDE {
+    autocheckout_bubble_shown_ = true;
+    callback.Run(true);
+  }
+
+  virtual void ShowRequestAutocompleteDialog(
+      const FormData& form,
+      const GURL& source_url,
+      DialogType dialog_type,
+      const base::Callback<void(const FormStructure*,
+                                const std::string&)>& callback) OVERRIDE {
+    callback.Run(user_supplied_data_.get(), "google_transaction_id");
+  }
+
+  void SetUserSuppliedData(scoped_ptr<FormStructure> user_supplied_data) {
+    user_supplied_data_.reset(user_supplied_data.release());
+  }
+
+  bool autocheckout_bubble_shown() const {
+    return autocheckout_bubble_shown_;
+  }
+
  private:
-  DISALLOW_COPY_AND_ASSIGN(MockAutofillManagerDelegate);
+  bool autocheckout_bubble_shown_;
+  scoped_ptr<FormStructure> user_supplied_data_;
+
+ DISALLOW_COPY_AND_ASSIGN(MockAutofillManagerDelegate);
 };
 
 class MockAutofillExternalDelegate : public AutofillExternalDelegate {
  public:
   explicit MockAutofillExternalDelegate(content::WebContents* web_contents,
                                         AutofillManager* autofill_manager)
       : AutofillExternalDelegate(web_contents, autofill_manager) {}
   virtual ~MockAutofillExternalDelegate() {}
 
   MOCK_METHOD5(OnQuery, void(int query_id,
                              const FormData& form,
                              const FormFieldData& field,
                              const gfx::RectF& bounds,
                              bool display_warning));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockAutofillExternalDelegate);
 };
 
 }  // namespace
 
 // Test that Autocheckout bubble is offered when server specifies field types.
 TEST_F(AutofillManagerTest, TestBubbleShown) {
   MockAutofillManagerDelegate delegate;
   autofill_manager_.reset(new TestAutofillManager(
       autofill_driver_.get(), &delegate, &personal_data_));
   autofill_manager_->set_autofill_enabled(true);
   autofill_manager_->MarkAsFirstPageInAutocheckoutFlow();
 
-  EXPECT_CALL(delegate, ShowAutocheckoutBubble(_, _, _));
-
   FormData form;
   CreateTestAddressFormData(&form);
 
   TestFormStructure* form_structure = new TestFormStructure(form);
   AutofillMetrics metrics_logger;  // ignored
   form_structure->DetermineHeuristicTypes(metrics_logger);
 
   // Build and add form structure with server data.
   std::vector<AutofillFieldType> heuristic_types, server_types;
   for (size_t i = 0; i < form.fields.size(); ++i) {
     heuristic_types.push_back(UNKNOWN_TYPE);
     server_types.push_back(form_structure->field(i)->type());
   }
   form_structure->SetFieldTypes(heuristic_types, server_types);
   autofill_manager_->AddSeenForm(form_structure);
 
   autofill_manager_->OnMaybeShowAutocheckoutBubble(form, gfx::RectF());
+
+  ASSERT_TRUE(delegate.autocheckout_bubble_shown());

[Ilya Sherman, 2013/06/19 23:25:13]

nit: EXPECT_TRUE

[Dane Wallinga, 2013/06/20 19:46:41]

Done.

 }
 
 // Test that Autocheckout bubble is not offered when server doesn't have data
 // for the form.
 TEST_F(AutofillManagerTest, TestAutocheckoutBubbleNotShown) {
   MockAutofillManagerDelegate delegate;
   autofill_manager_.reset(new TestAutofillManager(
       autofill_driver_.get(), &delegate, &personal_data_));
   autofill_manager_->set_autofill_enabled(true);
   autofill_manager_->MarkAsFirstPageInAutocheckoutFlow();
 
   FormData form;
   CreateTestAddressFormData(&form);
 
   TestFormStructure* form_structure = new TestFormStructure(form);
   AutofillMetrics metrics_logger;  // ignored
   form_structure->DetermineHeuristicTypes(metrics_logger);
 
   // Build form structure without server data.
   std::vector<AutofillFieldType> heuristic_types, server_types;
   for (size_t i = 0; i < form.fields.size(); ++i) {
     heuristic_types.push_back(form_structure->field(i)->type());
     server_types.push_back(NO_SERVER_DATA);
   }
   form_structure->SetFieldTypes(heuristic_types, server_types);
   autofill_manager_->AddSeenForm(form_structure);
 
   autofill_manager_->OnMaybeShowAutocheckoutBubble(form, gfx::RectF());
+
+  ASSERT_FALSE(delegate.autocheckout_bubble_shown());

[Ilya Sherman, 2013/06/19 23:25:13]

nit: EXPECT_FALSE

[Dane Wallinga, 2013/06/20 19:46:41]

Done.

 }
 
 // Test our external delegate is called at the right time.
 TEST_F(AutofillManagerTest, TestExternalDelegate) {
   MockAutofillExternalDelegate external_delegate(web_contents(),
                                                  autofill_manager_.get());
   EXPECT_CALL(external_delegate, OnQuery(_, _, _, _, _));
   autofill_manager_->SetExternalDelegate(&external_delegate);
 
   FormData form;
   CreateTestAddressFormData(&form);
   std::vector<FormData> forms(1, form);
   FormsSeen(forms);
   const FormFieldData& field = form.fields[0];
   GetAutofillSuggestions(form, field);  // should call the delegate's OnQuery()
 
   autofill_manager_->SetExternalDelegate(NULL);
 }
 
+// Test that in the case of Autocheckout, forms seen are in order supplied.
+TEST_F(AutofillManagerTest, DynamicFormsSeenAndIgnored) {
+  MockAutofillManagerDelegate delegate;
+  autofill_manager_.reset(new TestAutofillManager(
+      autofill_driver_.get(), &delegate, &personal_data_));
+  FormData shipping_options;
+  CreateTestShippingOptionsFormData(&shipping_options);
+  FormData user_supplied;
+  CreateTestFormWithAutocompleteAttribute(&user_supplied);
+  FormData address;
+  CreateTestAddressFormData(&address);
+
+  autofill_manager_->set_autocheckout_url_prefix("test-prefix");
+  // Push address only
+  std::vector<FormData> forms;
+  forms.push_back(address);
+
+  // Build and add form structure with server data.
+  TestFormStructure* form_structure = new TestFormStructure(address);

[Ilya Sherman, 2013/06/19 23:25:13]

nit: Wrap this in a scoped_ptr.

[Dane Wallinga, 2013/06/20 19:46:41]

You sure? A raw pointer is used elsewhere in this file for form_structure, and
I'm getting some sort of error during the tear down if I try wrapping it.

[Ilya Sherman, 2013/06/21 05:10:14]

The preferred style is to wrap all new calls in scoped_ptrs, and then call
.Pass() or .release() to transfer ownership.

+  std::vector<AutofillFieldType> heuristic_types, server_types;
+  for (size_t i = 0; i < address.fields.size(); ++i) {
+    heuristic_types.push_back(UNKNOWN_TYPE);
+    server_types.push_back(form_structure->field(i)->type());
+  }
+  form_structure->SetFieldTypes(heuristic_types, server_types);
+  autofill_manager_->AddSeenForm(form_structure);
+
+  // Make sure normal form is handled correctly.
+  //  FormsSeen(forms);

[Ilya Sherman, 2013/06/19 23:25:13]

Why doesn't this line get to join the party?

[Dane Wallinga, 2013/06/20 19:46:41]

heh, oops.

+  autofill_manager_->MarkAsFirstPageInAutocheckoutFlowIgnoringAjax();
+  std::vector<FormStructure*> form_structures;
+  form_structures = autofill_manager_->GetFormStructures();
+  ASSERT_EQ(1U, form_structures.size());
+  EXPECT_EQ("/form.html", form_structures[0]->source_url().path());
+
+  scoped_ptr<FormStructure> filled_form(new TestFormStructure(address));
+  delegate.SetUserSuppliedData(filled_form.Pass());
+  autofill_manager_->OnMaybeShowAutocheckoutBubble(address, gfx::RectF());
+
+  // Push other forms
+  forms.push_back(shipping_options);
+  forms.push_back(user_supplied);
+
+  // FormStructure should contain the same forms as before.
+  DynamicFormsSeen(forms);
+  form_structures = autofill_manager_->GetFormStructures();
+  ASSERT_EQ(1U, form_structures.size());
+  EXPECT_EQ("/form.html", form_structures[0]->source_url().path());
+}
+
 }  // namespace autofill