1// Copyright (c) 2011 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 <string>
6
7#include "base/string16.h"
8#include "base/utf_string_conversions.h"
9#include "chrome/browser/autofill/address.h"
10#include "chrome/browser/autofill/autofill_type.h"
11#include "testing/gtest/include/gtest/gtest.h"
12
13// Test that the getters and setters for country code are working.
14TEST(AddressTest, CountryCode) {
15  Address address;
16  EXPECT_EQ(std::string(), address.country_code());
17
18  address.set_country_code("US");
19  EXPECT_EQ("US", address.country_code());
20
21  address.set_country_code("CA");
22  EXPECT_EQ("CA", address.country_code());
23}
24
25// Test that country codes are properly decoded as country names.
26TEST(AddressTest, GetCountry) {
27  Address address;
28  EXPECT_EQ(std::string(), address.country_code());
29
30  // Make sure that nothing breaks when the country code is missing.
31  string16 country = address.GetInfo(ADDRESS_HOME_COUNTRY);
32  EXPECT_EQ(string16(), country);
33
34  address.set_country_code("US");
35  country = address.GetInfo(ADDRESS_HOME_COUNTRY);
36  EXPECT_EQ(ASCIIToUTF16("United States"), country);
37
38  address.set_country_code("CA");
39  country = address.GetInfo(ADDRESS_HOME_COUNTRY);
40  EXPECT_EQ(ASCIIToUTF16("Canada"), country);
41}
42
43// Test that we properly detect country codes appropriate for each country.
44TEST(AddressTest, SetCountry) {
45  Address address;
46  EXPECT_EQ(std::string(), address.country_code());
47
48  // Test basic conversion.
49  address.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("United States"));
50  string16 country = address.GetInfo(ADDRESS_HOME_COUNTRY);
51  EXPECT_EQ("US", address.country_code());
52  EXPECT_EQ(ASCIIToUTF16("United States"), country);
53
54  // Test basic synonym detection.
55  address.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("USA"));
56  country = address.GetInfo(ADDRESS_HOME_COUNTRY);
57  EXPECT_EQ("US", address.country_code());
58  EXPECT_EQ(ASCIIToUTF16("United States"), country);
59
60  // Test case-insensitivity.
61  address.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("canADA"));
62  country = address.GetInfo(ADDRESS_HOME_COUNTRY);
63  EXPECT_EQ("CA", address.country_code());
64  EXPECT_EQ(ASCIIToUTF16("Canada"), country);
65
66  // Test country code detection.
67  address.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("JP"));
68  country = address.GetInfo(ADDRESS_HOME_COUNTRY);
69  EXPECT_EQ("JP", address.country_code());
70  EXPECT_EQ(ASCIIToUTF16("Japan"), country);
71
72  // Test that we ignore unknown countries.
73  address.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("Unknown"));
74  country = address.GetInfo(ADDRESS_HOME_COUNTRY);
75  EXPECT_EQ(std::string(), address.country_code());
76  EXPECT_EQ(string16(), country);
77}
78
79// Test that we properly match typed values to stored country data.
80TEST(AddressTest, IsCountry) {
81  Address address;
82  address.set_country_code("US");
83
84  const char* const kValidMatches[] = {
85    "United States",
86    "USA",
87    "US",
88    "United states",
89    "us"
90  };
91  for (size_t i = 0; i < arraysize(kValidMatches); ++i) {
92    SCOPED_TRACE(kValidMatches[i]);
93    FieldTypeSet possible_field_types;
94    address.GetPossibleFieldTypes(ASCIIToUTF16(kValidMatches[i]),
95                                  &possible_field_types);
96    ASSERT_EQ(1U, possible_field_types.size());
97    EXPECT_EQ(ADDRESS_HOME_COUNTRY, *possible_field_types.begin());
98  }
99
100  const char* const kInvalidMatches[] = {
101    "United",
102    "Garbage"
103  };
104  for (size_t i = 0; i < arraysize(kInvalidMatches); ++i) {
105    FieldTypeSet possible_field_types;
106    address.GetPossibleFieldTypes(ASCIIToUTF16(kInvalidMatches[i]),
107                                  &possible_field_types);
108    EXPECT_EQ(0U, possible_field_types.size());
109  }
110
111  // Make sure that garbage values don't match when the country code is empty.
112  address.set_country_code("");
113  EXPECT_EQ(std::string(), address.country_code());
114  FieldTypeSet possible_field_types;
115  address.GetPossibleFieldTypes(ASCIIToUTF16("Garbage"),
116                                &possible_field_types);
117  EXPECT_EQ(0U, possible_field_types.size());
118}
119