components/autofill/core/browser/address_field_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/memory/scoped_ptr.h"
   6 #include "base/memory/scoped_vector.h"
   7 #include "base/strings/string16.h"
   8 #include "base/strings/utf_string_conversions.h"
   9 #include "components/autofill/core/browser/address_field.h"
  10 #include "components/autofill/core/browser/autofill_field.h"
  11 #include "components/autofill/core/browser/autofill_scanner.h"
  12 #include "components/autofill/core/common/form_field_data.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14
  15 using base::ASCIIToUTF16;
  16
  17 namespace autofill {
  18
  19 class AddressFieldTest : public testing::Test {
  20  public:
  21   AddressFieldTest() {}
  22
  23  protected:
  24   ScopedVector<AutofillField> list_;
  25   scoped_ptr<AddressField> field_;
  26   ServerFieldTypeMap field_type_map_;
  27
  28   // Downcast for tests.
  29   static scoped_ptr<AddressField> Parse(AutofillScanner* scanner) {
  30     scoped_ptr<FormField> field = AddressField::Parse(scanner);
  31     return make_scoped_ptr(static_cast<AddressField*>(field.release()));
  32   }
  33
  34  private:
  35   DISALLOW_COPY_AND_ASSIGN(AddressFieldTest);
  36 };
  37
  38 TEST_F(AddressFieldTest, Empty) {
  39   AutofillScanner scanner(list_.get());
  40   field_ = Parse(&scanner);
  41   ASSERT_EQ(nullptr, field_.get());
  42 }
  43
  44 TEST_F(AddressFieldTest, NonParse) {
  45   list_.push_back(new AutofillField);
  46   AutofillScanner scanner(list_.get());
  47   field_ = Parse(&scanner);
  48   ASSERT_EQ(nullptr, field_.get());
  49 }
  50
  51 TEST_F(AddressFieldTest, ParseOneLineAddress) {
  52   FormFieldData field;
  53   field.form_control_type = "text";
  54
  55   field.label = ASCIIToUTF16("Address");
  56   field.name = ASCIIToUTF16("address");
  57   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr1")));
  58
  59   AutofillScanner scanner(list_.get());
  60   field_ = Parse(&scanner);
  61   ASSERT_NE(nullptr, field_.get());
  62   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
  63   ASSERT_TRUE(
  64       field_type_map_.find(ASCIIToUTF16("addr1")) != field_type_map_.end());
  65   EXPECT_EQ(ADDRESS_HOME_LINE1, field_type_map_[ASCIIToUTF16("addr1")]);
  66 }
  67
  68 TEST_F(AddressFieldTest, ParseTwoLineAddress) {
  69   FormFieldData field;
  70   field.form_control_type = "text";
  71
  72   field.label = ASCIIToUTF16("Address");
  73   field.name = ASCIIToUTF16("address");
  74   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr1")));
  75
  76   field.label = base::string16();
  77   field.name = ASCIIToUTF16("address2");
  78   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr2")));
  79
  80   AutofillScanner scanner(list_.get());
  81   field_ = Parse(&scanner);
  82   ASSERT_NE(nullptr, field_.get());
  83   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
  84   ASSERT_TRUE(field_type_map_.find(ASCIIToUTF16("addr1")) !=
  85               field_type_map_.end());
  86   EXPECT_EQ(ADDRESS_HOME_LINE1, field_type_map_[ASCIIToUTF16("addr1")]);
  87   ASSERT_TRUE(field_type_map_.find(ASCIIToUTF16("addr2")) !=
  88               field_type_map_.end());
  89   EXPECT_EQ(ADDRESS_HOME_LINE2, field_type_map_[ASCIIToUTF16("addr2")]);
  90 }
  91
  92 TEST_F(AddressFieldTest, ParseThreeLineAddress) {
  93   FormFieldData field;
  94   field.form_control_type = "text";
  95
  96   field.label = ASCIIToUTF16("Address Line1");
  97   field.name = ASCIIToUTF16("Address1");
  98   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr1")));
  99
 100   field.label = ASCIIToUTF16("Address Line2");
 101   field.name = ASCIIToUTF16("Address2");
 102   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr2")));
 103
 104   field.label = ASCIIToUTF16("Address Line3");
 105   field.name = ASCIIToUTF16("Address3");
 106   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr3")));
 107
 108   AutofillScanner scanner(list_.get());
 109   field_ = Parse(&scanner);
 110   ASSERT_NE(nullptr, field_.get());
 111   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 112   ASSERT_TRUE(
 113       field_type_map_.find(ASCIIToUTF16("addr1")) != field_type_map_.end());
 114   EXPECT_EQ(ADDRESS_HOME_LINE1, field_type_map_[ASCIIToUTF16("addr1")]);
 115   ASSERT_TRUE(
 116       field_type_map_.find(ASCIIToUTF16("addr2")) != field_type_map_.end());
 117   EXPECT_EQ(ADDRESS_HOME_LINE2, field_type_map_[ASCIIToUTF16("addr2")]);
 118   ASSERT_TRUE(
 119       field_type_map_.find(ASCIIToUTF16("addr3")) != field_type_map_.end());
 120   EXPECT_EQ(ADDRESS_HOME_LINE3, field_type_map_[ASCIIToUTF16("addr3")]);
 121 }
 122
 123 TEST_F(AddressFieldTest, ParseStreetAddressFromTextArea) {
 124   FormFieldData field;
 125   field.form_control_type = "textarea";
 126
 127   field.label = ASCIIToUTF16("Address");
 128   field.name = ASCIIToUTF16("address");
 129   list_.push_back(new AutofillField(field, ASCIIToUTF16("addr")));
 130
 131   AutofillScanner scanner(list_.get());
 132   field_ = Parse(&scanner);
 133   ASSERT_NE(nullptr, field_.get());
 134   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 135   ASSERT_TRUE(
 136       field_type_map_.find(ASCIIToUTF16("addr")) != field_type_map_.end());
 137   EXPECT_EQ(ADDRESS_HOME_STREET_ADDRESS, field_type_map_[ASCIIToUTF16("addr")]);
 138 }
 139
 140 TEST_F(AddressFieldTest, ParseCity) {
 141   FormFieldData field;
 142   field.form_control_type = "text";
 143
 144   field.label = ASCIIToUTF16("City");
 145   field.name = ASCIIToUTF16("city");
 146   list_.push_back(new AutofillField(field, ASCIIToUTF16("city1")));
 147
 148   AutofillScanner scanner(list_.get());
 149   field_ = Parse(&scanner);
 150   ASSERT_NE(nullptr, field_.get());
 151   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 152   ASSERT_TRUE(
 153       field_type_map_.find(ASCIIToUTF16("city1")) != field_type_map_.end());
 154   EXPECT_EQ(ADDRESS_HOME_CITY, field_type_map_[ASCIIToUTF16("city1")]);
 155 }
 156
 157 TEST_F(AddressFieldTest, ParseState) {
 158   FormFieldData field;
 159   field.form_control_type = "text";
 160
 161   field.label = ASCIIToUTF16("State");
 162   field.name = ASCIIToUTF16("state");
 163   list_.push_back(new AutofillField(field, ASCIIToUTF16("state1")));
 164
 165   AutofillScanner scanner(list_.get());
 166   field_ = Parse(&scanner);
 167   ASSERT_NE(nullptr, field_.get());
 168   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 169   ASSERT_TRUE(
 170       field_type_map_.find(ASCIIToUTF16("state1")) != field_type_map_.end());
 171   EXPECT_EQ(ADDRESS_HOME_STATE, field_type_map_[ASCIIToUTF16("state1")]);
 172 }
 173
 174 TEST_F(AddressFieldTest, ParseZip) {
 175   FormFieldData field;
 176   field.form_control_type = "text";
 177
 178   field.label = ASCIIToUTF16("Zip");
 179   field.name = ASCIIToUTF16("zip");
 180   list_.push_back(new AutofillField(field, ASCIIToUTF16("zip1")));
 181
 182   AutofillScanner scanner(list_.get());
 183   field_ = Parse(&scanner);
 184   ASSERT_NE(nullptr, field_.get());
 185   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 186   ASSERT_TRUE(
 187       field_type_map_.find(ASCIIToUTF16("zip1")) != field_type_map_.end());
 188   EXPECT_EQ(ADDRESS_HOME_ZIP, field_type_map_[ASCIIToUTF16("zip1")]);
 189 }
 190
 191 TEST_F(AddressFieldTest, ParseStateAndZipOneLabel) {
 192   FormFieldData field;
 193   field.form_control_type = "text";
 194
 195   field.label = ASCIIToUTF16("State/Province, Zip/Postal Code");
 196   field.name = ASCIIToUTF16("state");
 197   list_.push_back(new AutofillField(field, ASCIIToUTF16("state")));
 198
 199   field.label = ASCIIToUTF16("State/Province, Zip/Postal Code");
 200   field.name = ASCIIToUTF16("zip");
 201   list_.push_back(new AutofillField(field, ASCIIToUTF16("zip")));
 202
 203   AutofillScanner scanner(list_.get());
 204   field_ = Parse(&scanner);
 205   ASSERT_NE(nullptr, field_.get());
 206   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 207   ASSERT_TRUE(
 208       field_type_map_.find(ASCIIToUTF16("state")) != field_type_map_.end());
 209   EXPECT_EQ(ADDRESS_HOME_STATE, field_type_map_[ASCIIToUTF16("state")]);
 210   ASSERT_TRUE(
 211       field_type_map_.find(ASCIIToUTF16("zip")) != field_type_map_.end());
 212   EXPECT_EQ(ADDRESS_HOME_ZIP, field_type_map_[ASCIIToUTF16("zip")]);
 213 }
 214
 215 TEST_F(AddressFieldTest, ParseCountry) {
 216   FormFieldData field;
 217   field.form_control_type = "text";
 218
 219   field.label = ASCIIToUTF16("Country");
 220   field.name = ASCIIToUTF16("country");
 221   list_.push_back(new AutofillField(field, ASCIIToUTF16("country1")));
 222
 223   AutofillScanner scanner(list_.get());
 224   field_ = Parse(&scanner);
 225   ASSERT_NE(nullptr, field_.get());
 226   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 227   ASSERT_TRUE(
 228       field_type_map_.find(ASCIIToUTF16("country1")) != field_type_map_.end());
 229   EXPECT_EQ(ADDRESS_HOME_COUNTRY, field_type_map_[ASCIIToUTF16("country1")]);
 230 }
 231
 232 TEST_F(AddressFieldTest, ParseCompany) {
 233   FormFieldData field;
 234   field.form_control_type = "text";
 235
 236   field.label = ASCIIToUTF16("Company");
 237   field.name = ASCIIToUTF16("company");
 238   list_.push_back(new AutofillField(field, ASCIIToUTF16("company1")));
 239
 240   AutofillScanner scanner(list_.get());
 241   field_ = Parse(&scanner);
 242   ASSERT_NE(nullptr, field_.get());
 243   ASSERT_TRUE(field_->ClassifyField(&field_type_map_));
 244   ASSERT_TRUE(
 245       field_type_map_.find(ASCIIToUTF16("company1")) != field_type_map_.end());
 246   EXPECT_EQ(COMPANY_NAME, field_type_map_[ASCIIToUTF16("company1")]);
 247 }
 248
 249 }  // namespace autofill