Detecting form locale

chrome/browser/autofill/autofill_manager_unittest.cc

 // Copyright (c) 2010 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/ref_counted.h"
 #include "base/scoped_ptr.h"
 #include "base/scoped_vector.h"
 #include "base/string16.h"
 #include "base/tuple.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autofill/autofill_common_unittest.h"
 #include "chrome/browser/autofill/autofill_manager.h"
 #include "chrome/browser/autofill/autofill_profile.h"
 #include "chrome/browser/autofill/credit_card.h"
 #include "chrome/browser/autofill/personal_data_manager.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/renderer_host/test/test_render_view_host.h"
 #include "chrome/browser/tab_contents/test_tab_contents.h"
 #include "chrome/common/ipc_test_sink.h"
+#include "chrome/common/notification_source.h"
+#include "chrome/common/notification_service.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/render_messages.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/glue/form_data.h"
 #include "webkit/glue/form_field.h"
 
 using webkit_glue::FormData;
 
 typedef Tuple5<int,
@@ skipping 89 matching lines @@
   virtual bool IsAutoFillEnabled() const { return true; }
 
   AutoFillProfile* GetLabeledProfile(const char* label) {
     return test_personal_data_->GetLabeledProfile(label);
   }
 
   void AddProfile(AutoFillProfile* profile) {
     test_personal_data_->AddProfile(profile);
   }
 
+  size_t NumFormStructures() const { return form_structures()->size(); }
+  const FormStructure* GetFormStructureAtIndex(size_t index) const {
+    return (*form_structures())[index];
+  }
+
  private:
   TestPersonalDataManager* test_personal_data_;
 
   DISALLOW_COPY_AND_ASSIGN(TestAutoFillManager);
 };
 
 void CreateTestFormData(FormData* form) {
   form->name = ASCIIToUTF16("MyForm");
   form->method = ASCIIToUTF16("POST");
   form->origin = GURL("http://myform.com/form.html");
@@ skipping 1096 matching lines @@
   ASSERT_FALSE(profile()->GetPrefs()->GetBoolean(
       prefs::kAutoFillAuxiliaryProfilesEnabled));
   profile()->GetPrefs()->SetBoolean(
       prefs::kAutoFillAuxiliaryProfilesEnabled, true);
   profile()->GetPrefs()->ClearPref(prefs::kAutoFillAuxiliaryProfilesEnabled);
   ASSERT_FALSE(profile()->GetPrefs()->GetBoolean(
       prefs::kAutoFillAuxiliaryProfilesEnabled));
 #endif
 }
 
+// Make sure that we properly respond to language detected notifications.
+TEST_F(AutoFillManagerTest, LanguageDetectedNotification) {
+  FormData form1;
+  CreateTestFormData(&form1);
+  form1.name = ASCIIToUTF16("MyForm1");
+  form1.origin = GURL("http://myform.co.uk/form.html");
+  form1.action = GURL("http://myform.co.uk/submit.html");
+  form1.locale = "xx";
+
+  FormData form2;
+  CreateTestFormData(&form2);
+  form2.name = ASCIIToUTF16("MyForm2");
+  form2.origin = GURL("http://myform.fr/form.html");
+  form2.action = GURL("http://myform.fr/submit.html");
+  form2.locale = "xx";
+
+  // Set up our FormStructures.
+  std::vector<FormData> forms;
+  forms.push_back(form1);
+  autofill_manager_->FormsSeen(forms);
+
+  // We should not yet have tried to detect the locale.
+  ASSERT_EQ(1U, autofill_manager_->NumFormStructures());
+  EXPECT_EQ("xx", autofill_manager_->GetFormStructureAtIndex(0)->locale());
+
+  TabContents* tab_contents = autofill_manager_->tab_contents();
+  autofill_manager_->Observe(NotificationType::TAB_LANGUAGE_DETERMINED,
+                             Source<TabContents>(tab_contents),
+                             NotificationService::NoDetails());
+
+  // Now we should have detected the locale.
+  ASSERT_EQ(1U, autofill_manager_->NumFormStructures());
+  EXPECT_EQ("en-GB", autofill_manager_->GetFormStructureAtIndex(0)->locale());
+
+  forms.clear();
+  forms.push_back(form2);
+  autofill_manager_->FormsSeen(forms);
+
+  // We should detect the locale of this new form immediately.
+  ASSERT_EQ(2U, autofill_manager_->NumFormStructures());
+  EXPECT_EQ("en-GB", autofill_manager_->GetFormStructureAtIndex(0)->locale());
+  EXPECT_EQ("fr-FR", autofill_manager_->GetFormStructureAtIndex(1)->locale());
+
+  forms.clear();
+  forms.push_back(form1);
+  forms.push_back(form2);
+  autofill_manager_->Reset();
+  autofill_manager_->FormsSeen(forms);
+
+  // We should not yet have tried to detect the locale.
+  ASSERT_EQ(2U, autofill_manager_->NumFormStructures());
+  EXPECT_EQ("xx", autofill_manager_->GetFormStructureAtIndex(0)->locale());
+  EXPECT_EQ("xx", autofill_manager_->GetFormStructureAtIndex(1)->locale());
+
+  autofill_manager_->Observe(NotificationType::TAB_LANGUAGE_DETERMINED,
+                             Source<TabContents>(tab_contents),
+                             NotificationService::NoDetails());
+
+  // Now we should have detected both locales.
+  ASSERT_EQ(2U, autofill_manager_->NumFormStructures());
+  EXPECT_EQ("en-GB", autofill_manager_->GetFormStructureAtIndex(0)->locale());
+  EXPECT_EQ("fr-FR", autofill_manager_->GetFormStructureAtIndex(1)->locale());
+}