autofill_table_unittest.cc revision eb525c5499e34cc9c4b825d6d9e75bb07cc06ace
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 <vector>
6
7#include "base/file_util.h"
8#include "base/files/scoped_temp_dir.h"
9#include "base/guid.h"
10#include "base/path_service.h"
11#include "base/strings/string_number_conversions.h"
12#include "base/strings/string_util.h"
13#include "base/strings/utf_string_conversions.h"
14#include "base/time/time.h"
15#include "components/autofill/core/browser/autofill_profile.h"
16#include "components/autofill/core/browser/autofill_type.h"
17#include "components/autofill/core/browser/credit_card.h"
18#include "components/autofill/core/browser/webdata/autofill_change.h"
19#include "components/autofill/core/browser/webdata/autofill_entry.h"
20#include "components/autofill/core/browser/webdata/autofill_table.h"
21#include "components/autofill/core/common/form_field_data.h"
22#include "components/webdata/common/web_database.h"
23#include "components/webdata/encryptor/encryptor.h"
24#include "sql/statement.h"
25#include "testing/gtest/include/gtest/gtest.h"
26
27using base::Time;
28using base::TimeDelta;
29
30namespace autofill {
31
32// So we can compare AutofillKeys with EXPECT_EQ().
33std::ostream& operator<<(std::ostream& os, const AutofillKey& key) {
34  return os << UTF16ToASCII(key.name()) << ", " << UTF16ToASCII(key.value());
35}
36
37// So we can compare AutofillChanges with EXPECT_EQ().
38std::ostream& operator<<(std::ostream& os, const AutofillChange& change) {
39  switch (change.type()) {
40    case AutofillChange::ADD: {
41      os << "ADD";
42      break;
43    }
44    case AutofillChange::UPDATE: {
45      os << "UPDATE";
46      break;
47    }
48    case AutofillChange::REMOVE: {
49      os << "REMOVE";
50      break;
51    }
52  }
53  return os << " " << change.key();
54}
55
56namespace {
57
58bool CompareAutofillEntries(const AutofillEntry& a, const AutofillEntry& b) {
59  std::set<Time> timestamps1(a.timestamps().begin(), a.timestamps().end());
60  std::set<Time> timestamps2(b.timestamps().begin(), b.timestamps().end());
61
62  int compVal = a.key().name().compare(b.key().name());
63  if (compVal != 0) {
64    return compVal < 0;
65  }
66
67  compVal = a.key().value().compare(b.key().value());
68  if (compVal != 0) {
69    return compVal < 0;
70  }
71
72  if (timestamps1.size() != timestamps2.size()) {
73    return timestamps1.size() < timestamps2.size();
74  }
75
76  std::set<Time>::iterator it;
77  for (it = timestamps1.begin(); it != timestamps1.end(); it++) {
78    timestamps2.erase(*it);
79  }
80
81  return !timestamps2.empty();
82}
83
84}  // anonymous namespace
85
86class AutofillTableTest : public testing::Test {
87 public:
88  AutofillTableTest() {}
89  virtual ~AutofillTableTest() {}
90
91 protected:
92  typedef std::set<AutofillEntry,
93    bool (*)(const AutofillEntry&, const AutofillEntry&)> AutofillEntrySet;
94  typedef std::set<AutofillEntry, bool (*)(const AutofillEntry&,
95    const AutofillEntry&)>::iterator AutofillEntrySetIterator;
96
97  virtual void SetUp() {
98#if defined(OS_MACOSX)
99    Encryptor::UseMockKeychain(true);
100#endif
101    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
102    file_ = temp_dir_.path().AppendASCII("TestWebDatabase");
103
104    table_.reset(new AutofillTable("en-US"));
105    db_.reset(new WebDatabase);
106    db_->AddTable(table_.get());
107    ASSERT_EQ(sql::INIT_OK, db_->Init(file_));
108  }
109
110  static AutofillEntry MakeAutofillEntry(const char* name,
111                                         const char* value,
112                                         time_t timestamp0,
113                                         time_t timestamp1) {
114    std::vector<Time> timestamps;
115    if (timestamp0 >= 0)
116      timestamps.push_back(Time::FromTimeT(timestamp0));
117    if (timestamp1 >= 0)
118      timestamps.push_back(Time::FromTimeT(timestamp1));
119    return AutofillEntry(
120        AutofillKey(ASCIIToUTF16(name), ASCIIToUTF16(value)), timestamps);
121  }
122
123  base::FilePath file_;
124  base::ScopedTempDir temp_dir_;
125  scoped_ptr<AutofillTable> table_;
126  scoped_ptr<WebDatabase> db_;
127
128 private:
129  DISALLOW_COPY_AND_ASSIGN(AutofillTableTest);
130};
131
132TEST_F(AutofillTableTest, Autofill) {
133  Time t1 = Time::Now();
134
135  // Simulate the submission of a handful of entries in a field called "Name",
136  // some more often than others.
137  AutofillChangeList changes;
138  FormFieldData field;
139  field.name = ASCIIToUTF16("Name");
140  field.value = ASCIIToUTF16("Superman");
141  base::Time now = base::Time::Now();
142  base::TimeDelta two_seconds = base::TimeDelta::FromSeconds(2);
143  EXPECT_FALSE(table_->HasFormElements());
144  EXPECT_TRUE(table_->AddFormFieldValue(field, &changes));
145  EXPECT_TRUE(table_->HasFormElements());
146  std::vector<base::string16> v;
147  for (int i = 0; i < 5; i++) {
148    field.value = ASCIIToUTF16("Clark Kent");
149    EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
150                                              now + i * two_seconds));
151  }
152  for (int i = 0; i < 3; i++) {
153    field.value = ASCIIToUTF16("Clark Sutter");
154    EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
155                                              now + i * two_seconds));
156  }
157  for (int i = 0; i < 2; i++) {
158    field.name = ASCIIToUTF16("Favorite Color");
159    field.value = ASCIIToUTF16("Green");
160    EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
161                                              now + i * two_seconds));
162  }
163
164  int count = 0;
165  int64 pair_id = 0;
166
167  // We have added the name Clark Kent 5 times, so count should be 5 and pair_id
168  // should be somthing non-zero.
169  field.name = ASCIIToUTF16("Name");
170  field.value = ASCIIToUTF16("Clark Kent");
171  EXPECT_TRUE(table_->GetIDAndCountOfFormElement(field, &pair_id, &count));
172  EXPECT_EQ(5, count);
173  EXPECT_NE(0, pair_id);
174
175  // Storing in the data base should be case sensitive, so there should be no
176  // database entry for clark kent lowercase.
177  field.value = ASCIIToUTF16("clark kent");
178  EXPECT_TRUE(table_->GetIDAndCountOfFormElement(field, &pair_id, &count));
179  EXPECT_EQ(0, count);
180
181  field.name = ASCIIToUTF16("Favorite Color");
182  field.value = ASCIIToUTF16("Green");
183  EXPECT_TRUE(table_->GetIDAndCountOfFormElement(field, &pair_id, &count));
184  EXPECT_EQ(2, count);
185
186  // This is meant to get a list of suggestions for Name.  The empty prefix
187  // in the second argument means it should return all suggestions for a name
188  // no matter what they start with.  The order that the names occur in the list
189  // should be decreasing order by count.
190  EXPECT_TRUE(table_->GetFormValuesForElementName(
191      ASCIIToUTF16("Name"), base::string16(), &v, 6));
192  EXPECT_EQ(3U, v.size());
193  if (v.size() == 3) {
194    EXPECT_EQ(ASCIIToUTF16("Clark Kent"), v[0]);
195    EXPECT_EQ(ASCIIToUTF16("Clark Sutter"), v[1]);
196    EXPECT_EQ(ASCIIToUTF16("Superman"), v[2]);
197  }
198
199  // If we query again limiting the list size to 1, we should only get the most
200  // frequent entry.
201  EXPECT_TRUE(table_->GetFormValuesForElementName(
202      ASCIIToUTF16("Name"), base::string16(), &v, 1));
203  EXPECT_EQ(1U, v.size());
204  if (v.size() == 1) {
205    EXPECT_EQ(ASCIIToUTF16("Clark Kent"), v[0]);
206  }
207
208  // Querying for suggestions given a prefix is case-insensitive, so the prefix
209  // "cLa" shoud get suggestions for both Clarks.
210  EXPECT_TRUE(table_->GetFormValuesForElementName(
211      ASCIIToUTF16("Name"), ASCIIToUTF16("cLa"), &v, 6));
212  EXPECT_EQ(2U, v.size());
213  if (v.size() == 2) {
214    EXPECT_EQ(ASCIIToUTF16("Clark Kent"), v[0]);
215    EXPECT_EQ(ASCIIToUTF16("Clark Sutter"), v[1]);
216  }
217
218  // Removing all elements since the beginning of this function should remove
219  // everything from the database.
220  changes.clear();
221  EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(t1, Time(), &changes));
222
223  const AutofillChange expected_changes[] = {
224    AutofillChange(AutofillChange::REMOVE,
225                   AutofillKey(ASCIIToUTF16("Name"),
226                               ASCIIToUTF16("Superman"))),
227    AutofillChange(AutofillChange::REMOVE,
228                   AutofillKey(ASCIIToUTF16("Name"),
229                               ASCIIToUTF16("Clark Kent"))),
230    AutofillChange(AutofillChange::REMOVE,
231                   AutofillKey(ASCIIToUTF16("Name"),
232                               ASCIIToUTF16("Clark Sutter"))),
233    AutofillChange(AutofillChange::REMOVE,
234                   AutofillKey(ASCIIToUTF16("Favorite Color"),
235                               ASCIIToUTF16("Green"))),
236  };
237  EXPECT_EQ(arraysize(expected_changes), changes.size());
238  for (size_t i = 0; i < arraysize(expected_changes); i++) {
239    EXPECT_EQ(expected_changes[i], changes[i]);
240  }
241
242  field.name = ASCIIToUTF16("Name");
243  field.value = ASCIIToUTF16("Clark Kent");
244  EXPECT_TRUE(table_->GetIDAndCountOfFormElement(field, &pair_id, &count));
245  EXPECT_EQ(0, count);
246
247  EXPECT_TRUE(table_->GetFormValuesForElementName(
248      ASCIIToUTF16("Name"), base::string16(), &v, 6));
249  EXPECT_EQ(0U, v.size());
250
251  // Now add some values with empty strings.
252  const base::string16 kValue = ASCIIToUTF16("  toto   ");
253  field.name = ASCIIToUTF16("blank");
254  field.value = base::string16();
255  EXPECT_TRUE(table_->AddFormFieldValue(field, &changes));
256  field.name = ASCIIToUTF16("blank");
257  field.value = ASCIIToUTF16(" ");
258  EXPECT_TRUE(table_->AddFormFieldValue(field, &changes));
259  field.name = ASCIIToUTF16("blank");
260  field.value = ASCIIToUTF16("      ");
261  EXPECT_TRUE(table_->AddFormFieldValue(field, &changes));
262  field.name = ASCIIToUTF16("blank");
263  field.value = kValue;
264  EXPECT_TRUE(table_->AddFormFieldValue(field, &changes));
265
266  // They should be stored normally as the DB layer does not check for empty
267  // values.
268  v.clear();
269  EXPECT_TRUE(table_->GetFormValuesForElementName(
270      ASCIIToUTF16("blank"), base::string16(), &v, 10));
271  EXPECT_EQ(4U, v.size());
272
273  // Now we'll check that ClearAutofillEmptyValueElements() works as expected.
274  table_->ClearAutofillEmptyValueElements();
275
276  v.clear();
277  EXPECT_TRUE(table_->GetFormValuesForElementName(
278      ASCIIToUTF16("blank"), base::string16(), &v, 10));
279  ASSERT_EQ(1U, v.size());
280
281  EXPECT_EQ(kValue, v[0]);
282}
283
284TEST_F(AutofillTableTest, Autofill_RemoveBetweenChanges) {
285  TimeDelta one_day(TimeDelta::FromDays(1));
286  Time t1 = Time::Now();
287  Time t2 = t1 + one_day;
288
289  AutofillChangeList changes;
290  FormFieldData field;
291  field.name = ASCIIToUTF16("Name");
292  field.value = ASCIIToUTF16("Superman");
293  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t1));
294  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t2));
295
296  changes.clear();
297  EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(t1, t2, &changes));
298  ASSERT_EQ(1U, changes.size());
299  EXPECT_EQ(AutofillChange(AutofillChange::UPDATE,
300                           AutofillKey(ASCIIToUTF16("Name"),
301                                       ASCIIToUTF16("Superman"))),
302            changes[0]);
303  changes.clear();
304
305  EXPECT_TRUE(
306      table_->RemoveFormElementsAddedBetween(t2, t2 + one_day, &changes));
307  ASSERT_EQ(1U, changes.size());
308  EXPECT_EQ(AutofillChange(AutofillChange::REMOVE,
309                           AutofillKey(ASCIIToUTF16("Name"),
310                                       ASCIIToUTF16("Superman"))),
311            changes[0]);
312}
313
314TEST_F(AutofillTableTest, Autofill_AddChanges) {
315  TimeDelta one_day(TimeDelta::FromDays(1));
316  Time t1 = Time::Now();
317  Time t2 = t1 + one_day;
318
319  AutofillChangeList changes;
320  FormFieldData field;
321  field.name = ASCIIToUTF16("Name");
322  field.value = ASCIIToUTF16("Superman");
323  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t1));
324  ASSERT_EQ(1U, changes.size());
325  EXPECT_EQ(AutofillChange(AutofillChange::ADD,
326                           AutofillKey(ASCIIToUTF16("Name"),
327                                       ASCIIToUTF16("Superman"))),
328            changes[0]);
329
330  changes.clear();
331  EXPECT_TRUE(
332      table_->AddFormFieldValueTime(field, &changes, t2));
333  ASSERT_EQ(1U, changes.size());
334  EXPECT_EQ(AutofillChange(AutofillChange::UPDATE,
335                           AutofillKey(ASCIIToUTF16("Name"),
336                                       ASCIIToUTF16("Superman"))),
337            changes[0]);
338}
339
340TEST_F(AutofillTableTest, Autofill_UpdateOneWithOneTimestamp) {
341  AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, -1));
342  std::vector<AutofillEntry> entries;
343  entries.push_back(entry);
344  ASSERT_TRUE(table_->UpdateAutofillEntries(entries));
345
346  FormFieldData field;
347  field.name = ASCIIToUTF16("foo");
348  field.value = ASCIIToUTF16("bar");
349  int64 pair_id;
350  int count;
351  ASSERT_TRUE(table_->GetIDAndCountOfFormElement(field, &pair_id, &count));
352  EXPECT_LE(0, pair_id);
353  EXPECT_EQ(1, count);
354
355  std::vector<AutofillEntry> all_entries;
356  ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries));
357  ASSERT_EQ(1U, all_entries.size());
358  EXPECT_TRUE(entry == all_entries[0]);
359}
360
361TEST_F(AutofillTableTest, Autofill_UpdateOneWithTwoTimestamps) {
362  AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, 2));
363  std::vector<AutofillEntry> entries;
364  entries.push_back(entry);
365  ASSERT_TRUE(table_->UpdateAutofillEntries(entries));
366
367  FormFieldData field;
368  field.name = ASCIIToUTF16("foo");
369  field.value = ASCIIToUTF16("bar");
370  int64 pair_id;
371  int count;
372  ASSERT_TRUE(table_->GetIDAndCountOfFormElement(field, &pair_id, &count));
373  EXPECT_LE(0, pair_id);
374  EXPECT_EQ(2, count);
375
376  std::vector<AutofillEntry> all_entries;
377  ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries));
378  ASSERT_EQ(1U, all_entries.size());
379  EXPECT_TRUE(entry == all_entries[0]);
380}
381
382TEST_F(AutofillTableTest, Autofill_GetAutofillTimestamps) {
383  AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, 2));
384  std::vector<AutofillEntry> entries;
385  entries.push_back(entry);
386  ASSERT_TRUE(table_->UpdateAutofillEntries(entries));
387
388  std::vector<Time> timestamps;
389  ASSERT_TRUE(table_->GetAutofillTimestamps(ASCIIToUTF16("foo"),
390                                            ASCIIToUTF16("bar"),
391                                            &timestamps));
392  ASSERT_EQ(2U, timestamps.size());
393  EXPECT_TRUE(Time::FromTimeT(1) == timestamps[0]);
394  EXPECT_TRUE(Time::FromTimeT(2) == timestamps[1]);
395}
396
397TEST_F(AutofillTableTest, Autofill_UpdateTwo) {
398  AutofillEntry entry0(MakeAutofillEntry("foo", "bar0", 1, -1));
399  AutofillEntry entry1(MakeAutofillEntry("foo", "bar1", 2, 3));
400  std::vector<AutofillEntry> entries;
401  entries.push_back(entry0);
402  entries.push_back(entry1);
403  ASSERT_TRUE(table_->UpdateAutofillEntries(entries));
404
405  FormFieldData field0;
406  field0.name = ASCIIToUTF16("foo");
407  field0.value = ASCIIToUTF16("bar0");
408  int64 pair_id;
409  int count;
410  ASSERT_TRUE(table_->GetIDAndCountOfFormElement(field0, &pair_id, &count));
411  EXPECT_LE(0, pair_id);
412  EXPECT_EQ(1, count);
413
414  FormFieldData field1;
415  field1.name = ASCIIToUTF16("foo");
416  field1.value = ASCIIToUTF16("bar1");
417  ASSERT_TRUE(table_->GetIDAndCountOfFormElement(field1, &pair_id, &count));
418  EXPECT_LE(0, pair_id);
419  EXPECT_EQ(2, count);
420}
421
422TEST_F(AutofillTableTest, Autofill_UpdateReplace) {
423  AutofillChangeList changes;
424  // Add a form field.  This will be replaced.
425  FormFieldData field;
426  field.name = ASCIIToUTF16("Name");
427  field.value = ASCIIToUTF16("Superman");
428  EXPECT_TRUE(table_->AddFormFieldValue(field, &changes));
429
430  AutofillEntry entry(MakeAutofillEntry("Name", "Superman", 1, 2));
431  std::vector<AutofillEntry> entries;
432  entries.push_back(entry);
433  ASSERT_TRUE(table_->UpdateAutofillEntries(entries));
434
435  std::vector<AutofillEntry> all_entries;
436  ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries));
437  ASSERT_EQ(1U, all_entries.size());
438  EXPECT_TRUE(entry == all_entries[0]);
439}
440
441TEST_F(AutofillTableTest, Autofill_UpdateDontReplace) {
442  Time t = Time::Now();
443  AutofillEntry existing(
444      MakeAutofillEntry("Name", "Superman", t.ToTimeT(), -1));
445
446  AutofillChangeList changes;
447  // Add a form field.  This will NOT be replaced.
448  FormFieldData field;
449  field.name = existing.key().name();
450  field.value = existing.key().value();
451  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t));
452  AutofillEntry entry(MakeAutofillEntry("Name", "Clark Kent", 1, 2));
453  std::vector<AutofillEntry> entries;
454  entries.push_back(entry);
455  ASSERT_TRUE(table_->UpdateAutofillEntries(entries));
456
457  std::vector<AutofillEntry> all_entries;
458  ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries));
459  ASSERT_EQ(2U, all_entries.size());
460  AutofillEntrySet expected_entries(all_entries.begin(),
461                                    all_entries.end(),
462                                    CompareAutofillEntries);
463  EXPECT_EQ(1U, expected_entries.count(existing));
464  EXPECT_EQ(1U, expected_entries.count(entry));
465}
466
467TEST_F(AutofillTableTest, Autofill_AddFormFieldValues) {
468  Time t = Time::Now();
469
470  // Add multiple values for "firstname" and "lastname" names.  Test that only
471  // first value of each gets added. Related to security issue:
472  // http://crbug.com/51727.
473  std::vector<FormFieldData> elements;
474  FormFieldData field;
475  field.name = ASCIIToUTF16("firstname");
476  field.value = ASCIIToUTF16("Joe");
477  elements.push_back(field);
478
479  field.name = ASCIIToUTF16("firstname");
480  field.value = ASCIIToUTF16("Jane");
481  elements.push_back(field);
482
483  field.name = ASCIIToUTF16("lastname");
484  field.value = ASCIIToUTF16("Smith");
485  elements.push_back(field);
486
487  field.name = ASCIIToUTF16("lastname");
488  field.value = ASCIIToUTF16("Jones");
489  elements.push_back(field);
490
491  std::vector<AutofillChange> changes;
492  table_->AddFormFieldValuesTime(elements, &changes, t);
493
494  ASSERT_EQ(2U, changes.size());
495  EXPECT_EQ(changes[0], AutofillChange(AutofillChange::ADD,
496                                       AutofillKey(ASCIIToUTF16("firstname"),
497                                       ASCIIToUTF16("Joe"))));
498  EXPECT_EQ(changes[1], AutofillChange(AutofillChange::ADD,
499                                       AutofillKey(ASCIIToUTF16("lastname"),
500                                       ASCIIToUTF16("Smith"))));
501
502  std::vector<AutofillEntry> all_entries;
503  ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries));
504  ASSERT_EQ(2U, all_entries.size());
505}
506
507TEST_F(AutofillTableTest, AutofillProfile) {
508  // Add a 'Home' profile.
509  AutofillProfile home_profile;
510  home_profile.set_origin(std::string());
511  home_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John"));
512  home_profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
513  home_profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith"));
514  home_profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz"));
515  home_profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Google"));
516  home_profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way"));
517  home_profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5"));
518  home_profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
519  home_profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
520  home_profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
521  home_profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
522  home_profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567"));
523
524  Time pre_creation_time = Time::Now();
525  EXPECT_TRUE(table_->AddAutofillProfile(home_profile));
526  Time post_creation_time = Time::Now();
527
528  // Get the 'Home' profile.
529  AutofillProfile* db_profile;
530  ASSERT_TRUE(table_->GetAutofillProfile(home_profile.guid(), &db_profile));
531  EXPECT_EQ(home_profile, *db_profile);
532  sql::Statement s_home(db_->GetSQLConnection()->GetUniqueStatement(
533      "SELECT date_modified "
534      "FROM autofill_profiles WHERE guid=?"));
535  s_home.BindString(0, home_profile.guid());
536  ASSERT_TRUE(s_home.is_valid());
537  ASSERT_TRUE(s_home.Step());
538  EXPECT_GE(s_home.ColumnInt64(0), pre_creation_time.ToTimeT());
539  EXPECT_LE(s_home.ColumnInt64(0), post_creation_time.ToTimeT());
540  EXPECT_FALSE(s_home.Step());
541  delete db_profile;
542
543  // Add a 'Billing' profile.
544  AutofillProfile billing_profile = home_profile;
545  billing_profile.set_guid(base::GenerateGUID());
546  billing_profile.set_origin("https://www.example.com/");
547  billing_profile.SetRawInfo(ADDRESS_HOME_LINE1,
548                             ASCIIToUTF16("5678 Bottom Street"));
549  billing_profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("suite 3"));
550
551  pre_creation_time = Time::Now();
552  EXPECT_TRUE(table_->AddAutofillProfile(billing_profile));
553  post_creation_time = Time::Now();
554
555  // Get the 'Billing' profile.
556  ASSERT_TRUE(table_->GetAutofillProfile(billing_profile.guid(), &db_profile));
557  EXPECT_EQ(billing_profile, *db_profile);
558  sql::Statement s_billing(db_->GetSQLConnection()->GetUniqueStatement(
559      "SELECT date_modified FROM autofill_profiles WHERE guid=?"));
560  s_billing.BindString(0, billing_profile.guid());
561  ASSERT_TRUE(s_billing.is_valid());
562  ASSERT_TRUE(s_billing.Step());
563  EXPECT_GE(s_billing.ColumnInt64(0), pre_creation_time.ToTimeT());
564  EXPECT_LE(s_billing.ColumnInt64(0), post_creation_time.ToTimeT());
565  EXPECT_FALSE(s_billing.Step());
566  delete db_profile;
567
568  // Update the 'Billing' profile, name only.
569  billing_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Jane"));
570  Time pre_modification_time = Time::Now();
571  EXPECT_TRUE(table_->UpdateAutofillProfile(billing_profile));
572  Time post_modification_time = Time::Now();
573  ASSERT_TRUE(table_->GetAutofillProfile(billing_profile.guid(), &db_profile));
574  EXPECT_EQ(billing_profile, *db_profile);
575  sql::Statement s_billing_updated(db_->GetSQLConnection()->GetUniqueStatement(
576      "SELECT date_modified FROM autofill_profiles WHERE guid=?"));
577  s_billing_updated.BindString(0, billing_profile.guid());
578  ASSERT_TRUE(s_billing_updated.is_valid());
579  ASSERT_TRUE(s_billing_updated.Step());
580  EXPECT_GE(s_billing_updated.ColumnInt64(0),
581            pre_modification_time.ToTimeT());
582  EXPECT_LE(s_billing_updated.ColumnInt64(0),
583            post_modification_time.ToTimeT());
584  EXPECT_FALSE(s_billing_updated.Step());
585  delete db_profile;
586
587  // Update the 'Billing' profile.
588  billing_profile.set_origin("Chrome settings");
589  billing_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Janice"));
590  billing_profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("C."));
591  billing_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Joplin"));
592  billing_profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("jane@singer.com"));
593  billing_profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Indy"));
594  billing_profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("Open Road"));
595  billing_profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("Route 66"));
596  billing_profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("NFA"));
597  billing_profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("NY"));
598  billing_profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("10011"));
599  billing_profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
600  billing_profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER,
601                             ASCIIToUTF16("18181230000"));
602  Time pre_modification_time_2 = Time::Now();
603  EXPECT_TRUE(table_->UpdateAutofillProfile(billing_profile));
604  Time post_modification_time_2 = Time::Now();
605  ASSERT_TRUE(table_->GetAutofillProfile(billing_profile.guid(), &db_profile));
606  EXPECT_EQ(billing_profile, *db_profile);
607  sql::Statement s_billing_updated_2(
608      db_->GetSQLConnection()->GetUniqueStatement(
609          "SELECT date_modified FROM autofill_profiles WHERE guid=?"));
610  s_billing_updated_2.BindString(0, billing_profile.guid());
611  ASSERT_TRUE(s_billing_updated_2.is_valid());
612  ASSERT_TRUE(s_billing_updated_2.Step());
613  EXPECT_GE(s_billing_updated_2.ColumnInt64(0),
614            pre_modification_time_2.ToTimeT());
615  EXPECT_LE(s_billing_updated_2.ColumnInt64(0),
616            post_modification_time_2.ToTimeT());
617  EXPECT_FALSE(s_billing_updated_2.Step());
618  delete db_profile;
619
620  // Remove the 'Billing' profile.
621  EXPECT_TRUE(table_->RemoveAutofillProfile(billing_profile.guid()));
622  EXPECT_FALSE(table_->GetAutofillProfile(billing_profile.guid(), &db_profile));
623}
624
625TEST_F(AutofillTableTest, AutofillProfileMultiValueNames) {
626  AutofillProfile p;
627  const base::string16 kJohnDoe(ASCIIToUTF16("John Doe"));
628  const base::string16 kJohnPDoe(ASCIIToUTF16("John P. Doe"));
629  std::vector<base::string16> set_values;
630  set_values.push_back(kJohnDoe);
631  set_values.push_back(kJohnPDoe);
632  p.SetRawMultiInfo(NAME_FULL, set_values);
633
634  EXPECT_TRUE(table_->AddAutofillProfile(p));
635
636  AutofillProfile* db_profile;
637  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
638  EXPECT_EQ(p, *db_profile);
639  EXPECT_EQ(0, p.Compare(*db_profile));
640  delete db_profile;
641
642  // Update the values.
643  const base::string16 kNoOne(ASCIIToUTF16("No One"));
644  set_values[1] = kNoOne;
645  p.SetRawMultiInfo(NAME_FULL, set_values);
646  EXPECT_TRUE(table_->UpdateAutofillProfile(p));
647  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
648  EXPECT_EQ(p, *db_profile);
649  EXPECT_EQ(0, p.Compare(*db_profile));
650  delete db_profile;
651
652  // Delete values.
653  set_values.clear();
654  p.SetRawMultiInfo(NAME_FULL, set_values);
655  EXPECT_TRUE(table_->UpdateAutofillProfile(p));
656  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
657  EXPECT_EQ(p, *db_profile);
658  EXPECT_EQ(0, p.Compare(*db_profile));
659  EXPECT_EQ(base::string16(), db_profile->GetRawInfo(NAME_FULL));
660  delete db_profile;
661}
662
663TEST_F(AutofillTableTest, AutofillProfileMultiValueEmails) {
664  AutofillProfile p;
665  const base::string16 kJohnDoe(ASCIIToUTF16("john@doe.com"));
666  const base::string16 kJohnPDoe(ASCIIToUTF16("john_p@doe.com"));
667  std::vector<base::string16> set_values;
668  set_values.push_back(kJohnDoe);
669  set_values.push_back(kJohnPDoe);
670  p.SetRawMultiInfo(EMAIL_ADDRESS, set_values);
671
672  EXPECT_TRUE(table_->AddAutofillProfile(p));
673
674  AutofillProfile* db_profile;
675  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
676  EXPECT_EQ(p, *db_profile);
677  EXPECT_EQ(0, p.Compare(*db_profile));
678  delete db_profile;
679
680  // Update the values.
681  const base::string16 kNoOne(ASCIIToUTF16("no@one.com"));
682  set_values[1] = kNoOne;
683  p.SetRawMultiInfo(EMAIL_ADDRESS, set_values);
684  EXPECT_TRUE(table_->UpdateAutofillProfile(p));
685  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
686  EXPECT_EQ(p, *db_profile);
687  EXPECT_EQ(0, p.Compare(*db_profile));
688  delete db_profile;
689
690  // Delete values.
691  set_values.clear();
692  p.SetRawMultiInfo(EMAIL_ADDRESS, set_values);
693  EXPECT_TRUE(table_->UpdateAutofillProfile(p));
694  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
695  EXPECT_EQ(p, *db_profile);
696  EXPECT_EQ(0, p.Compare(*db_profile));
697  EXPECT_EQ(base::string16(), db_profile->GetRawInfo(EMAIL_ADDRESS));
698  delete db_profile;
699}
700
701TEST_F(AutofillTableTest, AutofillProfileMultiValuePhone) {
702  AutofillProfile p;
703  const base::string16 kJohnDoe(ASCIIToUTF16("4151112222"));
704  const base::string16 kJohnPDoe(ASCIIToUTF16("4151113333"));
705  std::vector<base::string16> set_values;
706  set_values.push_back(kJohnDoe);
707  set_values.push_back(kJohnPDoe);
708  p.SetRawMultiInfo(PHONE_HOME_WHOLE_NUMBER, set_values);
709
710  EXPECT_TRUE(table_->AddAutofillProfile(p));
711
712  AutofillProfile* db_profile;
713  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
714  EXPECT_EQ(p, *db_profile);
715  EXPECT_EQ(0, p.Compare(*db_profile));
716  delete db_profile;
717
718  // Update the values.
719  const base::string16 kNoOne(ASCIIToUTF16("4151110000"));
720  set_values[1] = kNoOne;
721  p.SetRawMultiInfo(PHONE_HOME_WHOLE_NUMBER, set_values);
722  EXPECT_TRUE(table_->UpdateAutofillProfile(p));
723  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
724  EXPECT_EQ(p, *db_profile);
725  EXPECT_EQ(0, p.Compare(*db_profile));
726  delete db_profile;
727
728  // Delete values.
729  set_values.clear();
730  p.SetRawMultiInfo(PHONE_HOME_WHOLE_NUMBER, set_values);
731  EXPECT_TRUE(table_->UpdateAutofillProfile(p));
732  ASSERT_TRUE(table_->GetAutofillProfile(p.guid(), &db_profile));
733  EXPECT_EQ(p, *db_profile);
734  EXPECT_EQ(0, p.Compare(*db_profile));
735  EXPECT_EQ(base::string16(), db_profile->GetRawInfo(EMAIL_ADDRESS));
736  delete db_profile;
737}
738
739TEST_F(AutofillTableTest, AutofillProfileTrash) {
740  std::vector<std::string> guids;
741  table_->GetAutofillProfilesInTrash(&guids);
742  EXPECT_TRUE(guids.empty());
743
744  ASSERT_TRUE(table_->AddAutofillGUIDToTrash(
745      "00000000-0000-0000-0000-000000000000"));
746  ASSERT_TRUE(table_->AddAutofillGUIDToTrash(
747      "00000000-0000-0000-0000-000000000001"));
748  ASSERT_TRUE(table_->GetAutofillProfilesInTrash(&guids));
749  EXPECT_EQ(2UL, guids.size());
750  EXPECT_EQ("00000000-0000-0000-0000-000000000000", guids[0]);
751  EXPECT_EQ("00000000-0000-0000-0000-000000000001", guids[1]);
752
753  ASSERT_TRUE(table_->EmptyAutofillProfilesTrash());
754  ASSERT_TRUE(table_->GetAutofillProfilesInTrash(&guids));
755  EXPECT_TRUE(guids.empty());
756}
757
758TEST_F(AutofillTableTest, AutofillProfileTrashInteraction) {
759  std::vector<std::string> guids;
760  table_->GetAutofillProfilesInTrash(&guids);
761  EXPECT_TRUE(guids.empty());
762
763  AutofillProfile profile;
764  profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John"));
765  profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
766  profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith"));
767  profile.SetRawInfo(EMAIL_ADDRESS,ASCIIToUTF16("js@smith.xyz"));
768  profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1 Main St"));
769  profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
770  profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
771  profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
772  profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
773
774  // Mark this profile as in the trash.  This stops |AddAutofillProfile| from
775  // adding it.
776  EXPECT_TRUE(table_->AddAutofillGUIDToTrash(profile.guid()));
777  EXPECT_TRUE(table_->AddAutofillProfile(profile));
778  AutofillProfile* added_profile = NULL;
779  EXPECT_FALSE(table_->GetAutofillProfile(profile.guid(), &added_profile));
780  EXPECT_EQ(static_cast<AutofillProfile*>(NULL), added_profile);
781
782  // Add the profile for real this time.
783  EXPECT_TRUE(table_->EmptyAutofillProfilesTrash());
784  EXPECT_TRUE(table_->GetAutofillProfilesInTrash(&guids));
785  EXPECT_TRUE(guids.empty());
786  EXPECT_TRUE(table_->AddAutofillProfile(profile));
787  EXPECT_TRUE(table_->GetAutofillProfile(profile.guid(),
788                                                        &added_profile));
789  ASSERT_NE(static_cast<AutofillProfile*>(NULL), added_profile);
790  delete added_profile;
791
792  // Mark this profile as in the trash.  This stops |UpdateAutofillProfileMulti|
793  // from updating it.  In normal operation a profile should not be both in the
794  // trash and in the profiles table simultaneously.
795  EXPECT_TRUE(table_->AddAutofillGUIDToTrash(profile.guid()));
796  profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Jane"));
797  EXPECT_TRUE(table_->UpdateAutofillProfile(profile));
798  AutofillProfile* updated_profile = NULL;
799  EXPECT_TRUE(table_->GetAutofillProfile(profile.guid(), &updated_profile));
800  ASSERT_NE(static_cast<AutofillProfile*>(NULL), added_profile);
801  EXPECT_EQ(ASCIIToUTF16("John"), updated_profile->GetRawInfo(NAME_FIRST));
802  delete updated_profile;
803
804  // Try to delete the trashed profile.  This stops |RemoveAutofillProfile| from
805  // deleting it.  In normal operation deletion is done by migration step, and
806  // removal from trash is done by |WebDataService|.  |RemoveAutofillProfile|
807  // does remove the item from the trash if it is found however, so that if
808  // other clients remove it (via Sync say) then it is gone and doesn't need to
809  // be processed further by |WebDataService|.
810  EXPECT_TRUE(table_->RemoveAutofillProfile(profile.guid()));
811  AutofillProfile* removed_profile = NULL;
812  EXPECT_TRUE(table_->GetAutofillProfile(profile.guid(), &removed_profile));
813  EXPECT_FALSE(table_->IsAutofillGUIDInTrash(profile.guid()));
814  ASSERT_NE(static_cast<AutofillProfile*>(NULL), removed_profile);
815  delete removed_profile;
816
817  // Check that emptying the trash now allows removal to occur.
818  EXPECT_TRUE(table_->EmptyAutofillProfilesTrash());
819  EXPECT_TRUE(table_->RemoveAutofillProfile(profile.guid()));
820  removed_profile = NULL;
821  EXPECT_FALSE(table_->GetAutofillProfile(profile.guid(), &removed_profile));
822  EXPECT_EQ(static_cast<AutofillProfile*>(NULL), removed_profile);
823}
824
825TEST_F(AutofillTableTest, CreditCard) {
826  // Add a 'Work' credit card.
827  CreditCard work_creditcard;
828  work_creditcard.set_origin("https://www.example.com/");
829  work_creditcard.SetRawInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
830  work_creditcard.SetRawInfo(CREDIT_CARD_NUMBER,
831                             ASCIIToUTF16("1234567890123456"));
832  work_creditcard.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04"));
833  work_creditcard.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR,
834                             ASCIIToUTF16("2013"));
835
836  Time pre_creation_time = Time::Now();
837  EXPECT_TRUE(table_->AddCreditCard(work_creditcard));
838  Time post_creation_time = Time::Now();
839
840  // Get the 'Work' credit card.
841  CreditCard* db_creditcard;
842  ASSERT_TRUE(table_->GetCreditCard(work_creditcard.guid(), &db_creditcard));
843  EXPECT_EQ(work_creditcard, *db_creditcard);
844  sql::Statement s_work(db_->GetSQLConnection()->GetUniqueStatement(
845      "SELECT guid, name_on_card, expiration_month, expiration_year, "
846      "card_number_encrypted, date_modified "
847      "FROM credit_cards WHERE guid=?"));
848  s_work.BindString(0, work_creditcard.guid());
849  ASSERT_TRUE(s_work.is_valid());
850  ASSERT_TRUE(s_work.Step());
851  EXPECT_GE(s_work.ColumnInt64(5), pre_creation_time.ToTimeT());
852  EXPECT_LE(s_work.ColumnInt64(5), post_creation_time.ToTimeT());
853  EXPECT_FALSE(s_work.Step());
854  delete db_creditcard;
855
856  // Add a 'Target' credit card.
857  CreditCard target_creditcard;
858  target_creditcard.set_origin(std::string());
859  target_creditcard.SetRawInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
860  target_creditcard.SetRawInfo(CREDIT_CARD_NUMBER,
861                               ASCIIToUTF16("1111222233334444"));
862  target_creditcard.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("06"));
863  target_creditcard.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR,
864                               ASCIIToUTF16("2012"));
865
866  pre_creation_time = Time::Now();
867  EXPECT_TRUE(table_->AddCreditCard(target_creditcard));
868  post_creation_time = Time::Now();
869  ASSERT_TRUE(table_->GetCreditCard(target_creditcard.guid(), &db_creditcard));
870  EXPECT_EQ(target_creditcard, *db_creditcard);
871  sql::Statement s_target(db_->GetSQLConnection()->GetUniqueStatement(
872      "SELECT guid, name_on_card, expiration_month, expiration_year, "
873      "card_number_encrypted, date_modified "
874      "FROM credit_cards WHERE guid=?"));
875  s_target.BindString(0, target_creditcard.guid());
876  ASSERT_TRUE(s_target.is_valid());
877  ASSERT_TRUE(s_target.Step());
878  EXPECT_GE(s_target.ColumnInt64(5), pre_creation_time.ToTimeT());
879  EXPECT_LE(s_target.ColumnInt64(5), post_creation_time.ToTimeT());
880  EXPECT_FALSE(s_target.Step());
881  delete db_creditcard;
882
883  // Update the 'Target' credit card.
884  target_creditcard.set_origin("Interactive Autofill dialog");
885  target_creditcard.SetRawInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Charles Grady"));
886  Time pre_modification_time = Time::Now();
887  EXPECT_TRUE(table_->UpdateCreditCard(target_creditcard));
888  Time post_modification_time = Time::Now();
889  ASSERT_TRUE(table_->GetCreditCard(target_creditcard.guid(), &db_creditcard));
890  EXPECT_EQ(target_creditcard, *db_creditcard);
891  sql::Statement s_target_updated(db_->GetSQLConnection()->GetUniqueStatement(
892      "SELECT guid, name_on_card, expiration_month, expiration_year, "
893      "card_number_encrypted, date_modified "
894      "FROM credit_cards WHERE guid=?"));
895  s_target_updated.BindString(0, target_creditcard.guid());
896  ASSERT_TRUE(s_target_updated.is_valid());
897  ASSERT_TRUE(s_target_updated.Step());
898  EXPECT_GE(s_target_updated.ColumnInt64(5), pre_modification_time.ToTimeT());
899  EXPECT_LE(s_target_updated.ColumnInt64(5), post_modification_time.ToTimeT());
900  EXPECT_FALSE(s_target_updated.Step());
901  delete db_creditcard;
902
903  // Remove the 'Target' credit card.
904  EXPECT_TRUE(table_->RemoveCreditCard(target_creditcard.guid()));
905  EXPECT_FALSE(table_->GetCreditCard(target_creditcard.guid(), &db_creditcard));
906}
907
908TEST_F(AutofillTableTest, UpdateAutofillProfile) {
909  // Add a profile to the db.
910  AutofillProfile profile;
911  profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John"));
912  profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
913  profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith"));
914  profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@example.com"));
915  profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Google"));
916  profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way"));
917  profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5"));
918  profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
919  profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
920  profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
921  profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
922  profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567"));
923  table_->AddAutofillProfile(profile);
924
925  // Set a mocked value for the profile's creation time.
926  const time_t mock_creation_date = Time::Now().ToTimeT() - 13;
927  sql::Statement s_mock_creation_date(
928      db_->GetSQLConnection()->GetUniqueStatement(
929          "UPDATE autofill_profiles SET date_modified = ?"));
930  ASSERT_TRUE(s_mock_creation_date.is_valid());
931  s_mock_creation_date.BindInt64(0, mock_creation_date);
932  ASSERT_TRUE(s_mock_creation_date.Run());
933
934  // Get the profile.
935  AutofillProfile* tmp_profile;
936  ASSERT_TRUE(table_->GetAutofillProfile(profile.guid(), &tmp_profile));
937  scoped_ptr<AutofillProfile> db_profile(tmp_profile);
938  EXPECT_EQ(profile, *db_profile);
939  sql::Statement s_original(db_->GetSQLConnection()->GetUniqueStatement(
940      "SELECT date_modified FROM autofill_profiles"));
941  ASSERT_TRUE(s_original.is_valid());
942  ASSERT_TRUE(s_original.Step());
943  EXPECT_EQ(mock_creation_date, s_original.ColumnInt64(0));
944  EXPECT_FALSE(s_original.Step());
945
946  // Now, update the profile and save the update to the database.
947  // The modification date should change to reflect the update.
948  profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz"));
949  table_->UpdateAutofillProfile(profile);
950
951  // Get the profile.
952  ASSERT_TRUE(table_->GetAutofillProfile(profile.guid(), &tmp_profile));
953  db_profile.reset(tmp_profile);
954  EXPECT_EQ(profile, *db_profile);
955  sql::Statement s_updated(db_->GetSQLConnection()->GetUniqueStatement(
956      "SELECT date_modified FROM autofill_profiles"));
957  ASSERT_TRUE(s_updated.is_valid());
958  ASSERT_TRUE(s_updated.Step());
959  EXPECT_LT(mock_creation_date, s_updated.ColumnInt64(0));
960  EXPECT_FALSE(s_updated.Step());
961
962  // Set a mocked value for the profile's modification time.
963  const time_t mock_modification_date = Time::Now().ToTimeT() - 7;
964  sql::Statement s_mock_modification_date(
965      db_->GetSQLConnection()->GetUniqueStatement(
966          "UPDATE autofill_profiles SET date_modified = ?"));
967  ASSERT_TRUE(s_mock_modification_date.is_valid());
968  s_mock_modification_date.BindInt64(0, mock_modification_date);
969  ASSERT_TRUE(s_mock_modification_date.Run());
970
971  // Finally, call into |UpdateAutofillProfile()| without changing the
972  // profile.  The modification date should not change.
973  table_->UpdateAutofillProfile(profile);
974
975  // Get the profile.
976  ASSERT_TRUE(table_->GetAutofillProfile(profile.guid(), &tmp_profile));
977  db_profile.reset(tmp_profile);
978  EXPECT_EQ(profile, *db_profile);
979  sql::Statement s_unchanged(db_->GetSQLConnection()->GetUniqueStatement(
980      "SELECT date_modified FROM autofill_profiles"));
981  ASSERT_TRUE(s_unchanged.is_valid());
982  ASSERT_TRUE(s_unchanged.Step());
983  EXPECT_EQ(mock_modification_date, s_unchanged.ColumnInt64(0));
984  EXPECT_FALSE(s_unchanged.Step());
985}
986
987TEST_F(AutofillTableTest, UpdateCreditCard) {
988  // Add a credit card to the db.
989  CreditCard credit_card;
990  credit_card.SetRawInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
991  credit_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1234567890123456"));
992  credit_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04"));
993  credit_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2013"));
994  table_->AddCreditCard(credit_card);
995
996  // Set a mocked value for the credit card's creation time.
997  const time_t mock_creation_date = Time::Now().ToTimeT() - 13;
998  sql::Statement s_mock_creation_date(
999      db_->GetSQLConnection()->GetUniqueStatement(
1000          "UPDATE credit_cards SET date_modified = ?"));
1001  ASSERT_TRUE(s_mock_creation_date.is_valid());
1002  s_mock_creation_date.BindInt64(0, mock_creation_date);
1003  ASSERT_TRUE(s_mock_creation_date.Run());
1004
1005  // Get the credit card.
1006  CreditCard* tmp_credit_card;
1007  ASSERT_TRUE(table_->GetCreditCard(credit_card.guid(), &tmp_credit_card));
1008  scoped_ptr<CreditCard> db_credit_card(tmp_credit_card);
1009  EXPECT_EQ(credit_card, *db_credit_card);
1010  sql::Statement s_original(db_->GetSQLConnection()->GetUniqueStatement(
1011      "SELECT date_modified FROM credit_cards"));
1012  ASSERT_TRUE(s_original.is_valid());
1013  ASSERT_TRUE(s_original.Step());
1014  EXPECT_EQ(mock_creation_date, s_original.ColumnInt64(0));
1015  EXPECT_FALSE(s_original.Step());
1016
1017  // Now, update the credit card and save the update to the database.
1018  // The modification date should change to reflect the update.
1019  credit_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("01"));
1020  table_->UpdateCreditCard(credit_card);
1021
1022  // Get the credit card.
1023  ASSERT_TRUE(table_->GetCreditCard(credit_card.guid(), &tmp_credit_card));
1024  db_credit_card.reset(tmp_credit_card);
1025  EXPECT_EQ(credit_card, *db_credit_card);
1026  sql::Statement s_updated(db_->GetSQLConnection()->GetUniqueStatement(
1027      "SELECT date_modified FROM credit_cards"));
1028  ASSERT_TRUE(s_updated.is_valid());
1029  ASSERT_TRUE(s_updated.Step());
1030  EXPECT_LT(mock_creation_date, s_updated.ColumnInt64(0));
1031  EXPECT_FALSE(s_updated.Step());
1032
1033  // Set a mocked value for the credit card's modification time.
1034  const time_t mock_modification_date = Time::Now().ToTimeT() - 7;
1035  sql::Statement s_mock_modification_date(
1036      db_->GetSQLConnection()->GetUniqueStatement(
1037          "UPDATE credit_cards SET date_modified = ?"));
1038  ASSERT_TRUE(s_mock_modification_date.is_valid());
1039  s_mock_modification_date.BindInt64(0, mock_modification_date);
1040  ASSERT_TRUE(s_mock_modification_date.Run());
1041
1042  // Finally, call into |UpdateCreditCard()| without changing the credit card.
1043  // The modification date should not change.
1044  table_->UpdateCreditCard(credit_card);
1045
1046  // Get the credit card.
1047  ASSERT_TRUE(table_->GetCreditCard(credit_card.guid(), &tmp_credit_card));
1048  db_credit_card.reset(tmp_credit_card);
1049  EXPECT_EQ(credit_card, *db_credit_card);
1050  sql::Statement s_unchanged(db_->GetSQLConnection()->GetUniqueStatement(
1051      "SELECT date_modified FROM credit_cards"));
1052  ASSERT_TRUE(s_unchanged.is_valid());
1053  ASSERT_TRUE(s_unchanged.Step());
1054  EXPECT_EQ(mock_modification_date, s_unchanged.ColumnInt64(0));
1055  EXPECT_FALSE(s_unchanged.Step());
1056}
1057
1058TEST_F(AutofillTableTest, UpdateProfileOriginOnly) {
1059  // Add a profile to the db.
1060  AutofillProfile profile;
1061  profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John"));
1062  profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
1063  profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith"));
1064  profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@example.com"));
1065  profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Google"));
1066  profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way"));
1067  profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5"));
1068  profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
1069  profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
1070  profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
1071  profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
1072  profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567"));
1073  table_->AddAutofillProfile(profile);
1074
1075  // Set a mocked value for the profile's creation time.
1076  const time_t mock_creation_date = Time::Now().ToTimeT() - 13;
1077  sql::Statement s_mock_creation_date(
1078      db_->GetSQLConnection()->GetUniqueStatement(
1079          "UPDATE autofill_profiles SET date_modified = ?"));
1080  ASSERT_TRUE(s_mock_creation_date.is_valid());
1081  s_mock_creation_date.BindInt64(0, mock_creation_date);
1082  ASSERT_TRUE(s_mock_creation_date.Run());
1083
1084  // Get the profile.
1085  AutofillProfile* tmp_profile;
1086  ASSERT_TRUE(table_->GetAutofillProfile(profile.guid(), &tmp_profile));
1087  scoped_ptr<AutofillProfile> db_profile(tmp_profile);
1088  EXPECT_EQ(profile, *db_profile);
1089  sql::Statement s_original(db_->GetSQLConnection()->GetUniqueStatement(
1090      "SELECT date_modified FROM autofill_profiles"));
1091  ASSERT_TRUE(s_original.is_valid());
1092  ASSERT_TRUE(s_original.Step());
1093  EXPECT_EQ(mock_creation_date, s_original.ColumnInt64(0));
1094  EXPECT_FALSE(s_original.Step());
1095
1096  // Now, update just the profile's origin and save the update to the database.
1097  // The modification date should change to reflect the update.
1098  profile.set_origin("https://www.example.com/");
1099  table_->UpdateAutofillProfile(profile);
1100
1101  // Get the profile.
1102  ASSERT_TRUE(table_->GetAutofillProfile(profile.guid(), &tmp_profile));
1103  db_profile.reset(tmp_profile);
1104  EXPECT_EQ(profile, *db_profile);
1105  sql::Statement s_updated(db_->GetSQLConnection()->GetUniqueStatement(
1106      "SELECT date_modified FROM autofill_profiles"));
1107  ASSERT_TRUE(s_updated.is_valid());
1108  ASSERT_TRUE(s_updated.Step());
1109  EXPECT_LT(mock_creation_date, s_updated.ColumnInt64(0));
1110  EXPECT_FALSE(s_updated.Step());
1111}
1112
1113TEST_F(AutofillTableTest, UpdateCreditCardOriginOnly) {
1114  // Add a credit card to the db.
1115  CreditCard credit_card;
1116  credit_card.SetRawInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
1117  credit_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1234567890123456"));
1118  credit_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04"));
1119  credit_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2013"));
1120  table_->AddCreditCard(credit_card);
1121
1122  // Set a mocked value for the credit card's creation time.
1123  const time_t mock_creation_date = Time::Now().ToTimeT() - 13;
1124  sql::Statement s_mock_creation_date(
1125      db_->GetSQLConnection()->GetUniqueStatement(
1126          "UPDATE credit_cards SET date_modified = ?"));
1127  ASSERT_TRUE(s_mock_creation_date.is_valid());
1128  s_mock_creation_date.BindInt64(0, mock_creation_date);
1129  ASSERT_TRUE(s_mock_creation_date.Run());
1130
1131  // Get the credit card.
1132  CreditCard* tmp_credit_card;
1133  ASSERT_TRUE(table_->GetCreditCard(credit_card.guid(), &tmp_credit_card));
1134  scoped_ptr<CreditCard> db_credit_card(tmp_credit_card);
1135  EXPECT_EQ(credit_card, *db_credit_card);
1136  sql::Statement s_original(db_->GetSQLConnection()->GetUniqueStatement(
1137      "SELECT date_modified FROM credit_cards"));
1138  ASSERT_TRUE(s_original.is_valid());
1139  ASSERT_TRUE(s_original.Step());
1140  EXPECT_EQ(mock_creation_date, s_original.ColumnInt64(0));
1141  EXPECT_FALSE(s_original.Step());
1142
1143  // Now, update just the credit card's origin and save the update to the
1144  // database.  The modification date should change to reflect the update.
1145  credit_card.set_origin("https://www.example.com/");
1146  table_->UpdateCreditCard(credit_card);
1147
1148  // Get the credit card.
1149  ASSERT_TRUE(table_->GetCreditCard(credit_card.guid(), &tmp_credit_card));
1150  db_credit_card.reset(tmp_credit_card);
1151  EXPECT_EQ(credit_card, *db_credit_card);
1152  sql::Statement s_updated(db_->GetSQLConnection()->GetUniqueStatement(
1153      "SELECT date_modified FROM credit_cards"));
1154  ASSERT_TRUE(s_updated.is_valid());
1155  ASSERT_TRUE(s_updated.Step());
1156  EXPECT_LT(mock_creation_date, s_updated.ColumnInt64(0));
1157  EXPECT_FALSE(s_updated.Step());
1158}
1159
1160TEST_F(AutofillTableTest, RemoveAutofillDataModifiedBetween) {
1161  // Populate the autofill_profiles and credit_cards tables.
1162  ASSERT_TRUE(db_->GetSQLConnection()->Execute(
1163      "INSERT INTO autofill_profiles (guid, date_modified) "
1164      "VALUES('00000000-0000-0000-0000-000000000000', 11);"
1165      "INSERT INTO autofill_profiles (guid, date_modified) "
1166      "VALUES('00000000-0000-0000-0000-000000000001', 21);"
1167      "INSERT INTO autofill_profiles (guid, date_modified) "
1168      "VALUES('00000000-0000-0000-0000-000000000002', 31);"
1169      "INSERT INTO autofill_profiles (guid, date_modified) "
1170      "VALUES('00000000-0000-0000-0000-000000000003', 41);"
1171      "INSERT INTO autofill_profiles (guid, date_modified) "
1172      "VALUES('00000000-0000-0000-0000-000000000004', 51);"
1173      "INSERT INTO autofill_profiles (guid, date_modified) "
1174      "VALUES('00000000-0000-0000-0000-000000000005', 61);"
1175      "INSERT INTO credit_cards (guid, date_modified) "
1176      "VALUES('00000000-0000-0000-0000-000000000006', 17);"
1177      "INSERT INTO credit_cards (guid, date_modified) "
1178      "VALUES('00000000-0000-0000-0000-000000000007', 27);"
1179      "INSERT INTO credit_cards (guid, date_modified) "
1180      "VALUES('00000000-0000-0000-0000-000000000008', 37);"
1181      "INSERT INTO credit_cards (guid, date_modified) "
1182      "VALUES('00000000-0000-0000-0000-000000000009', 47);"
1183      "INSERT INTO credit_cards (guid, date_modified) "
1184      "VALUES('00000000-0000-0000-0000-000000000010', 57);"
1185      "INSERT INTO credit_cards (guid, date_modified) "
1186      "VALUES('00000000-0000-0000-0000-000000000011', 67);"));
1187
1188  // Remove all entries modified in the bounded time range [17,41).
1189  std::vector<std::string> profile_guids;
1190  std::vector<std::string> credit_card_guids;
1191  table_->RemoveAutofillDataModifiedBetween(
1192      Time::FromTimeT(17), Time::FromTimeT(41),
1193      &profile_guids, &credit_card_guids);
1194  ASSERT_EQ(2UL, profile_guids.size());
1195  EXPECT_EQ("00000000-0000-0000-0000-000000000001", profile_guids[0]);
1196  EXPECT_EQ("00000000-0000-0000-0000-000000000002", profile_guids[1]);
1197  sql::Statement s_autofill_profiles_bounded(
1198      db_->GetSQLConnection()->GetUniqueStatement(
1199          "SELECT date_modified FROM autofill_profiles"));
1200  ASSERT_TRUE(s_autofill_profiles_bounded.is_valid());
1201  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1202  EXPECT_EQ(11, s_autofill_profiles_bounded.ColumnInt64(0));
1203  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1204  EXPECT_EQ(41, s_autofill_profiles_bounded.ColumnInt64(0));
1205  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1206  EXPECT_EQ(51, s_autofill_profiles_bounded.ColumnInt64(0));
1207  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1208  EXPECT_EQ(61, s_autofill_profiles_bounded.ColumnInt64(0));
1209  EXPECT_FALSE(s_autofill_profiles_bounded.Step());
1210  ASSERT_EQ(3UL, credit_card_guids.size());
1211  EXPECT_EQ("00000000-0000-0000-0000-000000000006", credit_card_guids[0]);
1212  EXPECT_EQ("00000000-0000-0000-0000-000000000007", credit_card_guids[1]);
1213  EXPECT_EQ("00000000-0000-0000-0000-000000000008", credit_card_guids[2]);
1214  sql::Statement s_credit_cards_bounded(
1215      db_->GetSQLConnection()->GetUniqueStatement(
1216          "SELECT date_modified FROM credit_cards"));
1217  ASSERT_TRUE(s_credit_cards_bounded.is_valid());
1218  ASSERT_TRUE(s_credit_cards_bounded.Step());
1219  EXPECT_EQ(47, s_credit_cards_bounded.ColumnInt64(0));
1220  ASSERT_TRUE(s_credit_cards_bounded.Step());
1221  EXPECT_EQ(57, s_credit_cards_bounded.ColumnInt64(0));
1222  ASSERT_TRUE(s_credit_cards_bounded.Step());
1223  EXPECT_EQ(67, s_credit_cards_bounded.ColumnInt64(0));
1224  EXPECT_FALSE(s_credit_cards_bounded.Step());
1225
1226  // Remove all entries modified on or after time 51 (unbounded range).
1227  table_->RemoveAutofillDataModifiedBetween(
1228      Time::FromTimeT(51), Time(),
1229      &profile_guids, &credit_card_guids);
1230  ASSERT_EQ(2UL, profile_guids.size());
1231  EXPECT_EQ("00000000-0000-0000-0000-000000000004", profile_guids[0]);
1232  EXPECT_EQ("00000000-0000-0000-0000-000000000005", profile_guids[1]);
1233  sql::Statement s_autofill_profiles_unbounded(
1234      db_->GetSQLConnection()->GetUniqueStatement(
1235          "SELECT date_modified FROM autofill_profiles"));
1236  ASSERT_TRUE(s_autofill_profiles_unbounded.is_valid());
1237  ASSERT_TRUE(s_autofill_profiles_unbounded.Step());
1238  EXPECT_EQ(11, s_autofill_profiles_unbounded.ColumnInt64(0));
1239  ASSERT_TRUE(s_autofill_profiles_unbounded.Step());
1240  EXPECT_EQ(41, s_autofill_profiles_unbounded.ColumnInt64(0));
1241  EXPECT_FALSE(s_autofill_profiles_unbounded.Step());
1242  ASSERT_EQ(2UL, credit_card_guids.size());
1243  EXPECT_EQ("00000000-0000-0000-0000-000000000010", credit_card_guids[0]);
1244  EXPECT_EQ("00000000-0000-0000-0000-000000000011", credit_card_guids[1]);
1245  sql::Statement s_credit_cards_unbounded(
1246      db_->GetSQLConnection()->GetUniqueStatement(
1247          "SELECT date_modified FROM credit_cards"));
1248  ASSERT_TRUE(s_credit_cards_unbounded.is_valid());
1249  ASSERT_TRUE(s_credit_cards_unbounded.Step());
1250  EXPECT_EQ(47, s_credit_cards_unbounded.ColumnInt64(0));
1251  EXPECT_FALSE(s_credit_cards_unbounded.Step());
1252
1253  // Remove all remaining entries.
1254  table_->RemoveAutofillDataModifiedBetween(
1255      Time(), Time(),
1256      &profile_guids, &credit_card_guids);
1257  ASSERT_EQ(2UL, profile_guids.size());
1258  EXPECT_EQ("00000000-0000-0000-0000-000000000000", profile_guids[0]);
1259  EXPECT_EQ("00000000-0000-0000-0000-000000000003", profile_guids[1]);
1260  sql::Statement s_autofill_profiles_empty(
1261      db_->GetSQLConnection()->GetUniqueStatement(
1262          "SELECT date_modified FROM autofill_profiles"));
1263  ASSERT_TRUE(s_autofill_profiles_empty.is_valid());
1264  EXPECT_FALSE(s_autofill_profiles_empty.Step());
1265  ASSERT_EQ(1UL, credit_card_guids.size());
1266  EXPECT_EQ("00000000-0000-0000-0000-000000000009", credit_card_guids[0]);
1267  sql::Statement s_credit_cards_empty(
1268      db_->GetSQLConnection()->GetUniqueStatement(
1269          "SELECT date_modified FROM credit_cards"));
1270  ASSERT_TRUE(s_credit_cards_empty.is_valid());
1271  EXPECT_FALSE(s_credit_cards_empty.Step());
1272}
1273
1274TEST_F(AutofillTableTest, RemoveOriginURLsModifiedBetween) {
1275  // Populate the autofill_profiles and credit_cards tables.
1276  ASSERT_TRUE(db_->GetSQLConnection()->Execute(
1277      "INSERT INTO autofill_profiles (guid, origin, date_modified) "
1278      "VALUES('00000000-0000-0000-0000-000000000000', '', 11);"
1279      "INSERT INTO autofill_profiles (guid, origin, date_modified) "
1280      "VALUES('00000000-0000-0000-0000-000000000001', "
1281      "       'https://www.example.com/', 21);"
1282      "INSERT INTO autofill_profiles (guid, origin, date_modified) "
1283      "VALUES('00000000-0000-0000-0000-000000000002', 'Chrome settings', 31);"
1284      "INSERT INTO credit_cards (guid, origin, date_modified) "
1285      "VALUES('00000000-0000-0000-0000-000000000003', '', 17);"
1286      "INSERT INTO credit_cards (guid, origin, date_modified) "
1287      "VALUES('00000000-0000-0000-0000-000000000004', "
1288      "       'https://www.example.com/', 27);"
1289      "INSERT INTO credit_cards (guid, origin, date_modified) "
1290      "VALUES('00000000-0000-0000-0000-000000000005', 'Chrome settings', "
1291      "       37);"));
1292
1293  // Remove all origin URLs set in the bounded time range [21,27).
1294  ScopedVector<AutofillProfile> profiles;
1295  table_->RemoveOriginURLsModifiedBetween(
1296      Time::FromTimeT(21), Time::FromTimeT(27), &profiles);
1297  ASSERT_EQ(1UL, profiles.size());
1298  EXPECT_EQ("00000000-0000-0000-0000-000000000001", profiles[0]->guid());
1299  sql::Statement s_autofill_profiles_bounded(
1300      db_->GetSQLConnection()->GetUniqueStatement(
1301          "SELECT date_modified, origin FROM autofill_profiles"));
1302  ASSERT_TRUE(s_autofill_profiles_bounded.is_valid());
1303  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1304  EXPECT_EQ(11, s_autofill_profiles_bounded.ColumnInt64(0));
1305  EXPECT_EQ(std::string(), s_autofill_profiles_bounded.ColumnString(1));
1306  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1307  EXPECT_EQ(21, s_autofill_profiles_bounded.ColumnInt64(0));
1308  EXPECT_EQ(std::string(), s_autofill_profiles_bounded.ColumnString(1));
1309  ASSERT_TRUE(s_autofill_profiles_bounded.Step());
1310  EXPECT_EQ(31, s_autofill_profiles_bounded.ColumnInt64(0));
1311  EXPECT_EQ("Chrome settings", s_autofill_profiles_bounded.ColumnString(1));
1312  sql::Statement s_credit_cards_bounded(
1313      db_->GetSQLConnection()->GetUniqueStatement(
1314          "SELECT date_modified, origin FROM credit_cards"));
1315  ASSERT_TRUE(s_credit_cards_bounded.is_valid());
1316  ASSERT_TRUE(s_credit_cards_bounded.Step());
1317  EXPECT_EQ(17, s_credit_cards_bounded.ColumnInt64(0));
1318  EXPECT_EQ(std::string(), s_credit_cards_bounded.ColumnString(1));
1319  ASSERT_TRUE(s_credit_cards_bounded.Step());
1320  EXPECT_EQ(27, s_credit_cards_bounded.ColumnInt64(0));
1321  EXPECT_EQ("https://www.example.com/",
1322            s_credit_cards_bounded.ColumnString(1));
1323  ASSERT_TRUE(s_credit_cards_bounded.Step());
1324  EXPECT_EQ(37, s_credit_cards_bounded.ColumnInt64(0));
1325  EXPECT_EQ("Chrome settings", s_credit_cards_bounded.ColumnString(1));
1326
1327  // Remove all origin URLS.
1328  profiles.clear();
1329  table_->RemoveOriginURLsModifiedBetween(Time(), Time(), &profiles);
1330  EXPECT_EQ(0UL, profiles.size());
1331  sql::Statement s_autofill_profiles_all(
1332      db_->GetSQLConnection()->GetUniqueStatement(
1333          "SELECT date_modified, origin FROM autofill_profiles"));
1334  ASSERT_TRUE(s_autofill_profiles_all.is_valid());
1335  ASSERT_TRUE(s_autofill_profiles_all.Step());
1336  EXPECT_EQ(11, s_autofill_profiles_all.ColumnInt64(0));
1337  EXPECT_EQ(std::string(), s_autofill_profiles_all.ColumnString(1));
1338  ASSERT_TRUE(s_autofill_profiles_all.Step());
1339  EXPECT_EQ(21, s_autofill_profiles_all.ColumnInt64(0));
1340  EXPECT_EQ(std::string(), s_autofill_profiles_all.ColumnString(1));
1341  ASSERT_TRUE(s_autofill_profiles_all.Step());
1342  EXPECT_EQ(31, s_autofill_profiles_all.ColumnInt64(0));
1343  EXPECT_EQ("Chrome settings", s_autofill_profiles_all.ColumnString(1));
1344  sql::Statement s_credit_cards_all(
1345      db_->GetSQLConnection()->GetUniqueStatement(
1346          "SELECT date_modified, origin FROM credit_cards"));
1347  ASSERT_TRUE(s_credit_cards_all.is_valid());
1348  ASSERT_TRUE(s_credit_cards_all.Step());
1349  EXPECT_EQ(17, s_credit_cards_all.ColumnInt64(0));
1350  EXPECT_EQ(std::string(), s_credit_cards_all.ColumnString(1));
1351  ASSERT_TRUE(s_credit_cards_all.Step());
1352  EXPECT_EQ(27, s_credit_cards_all.ColumnInt64(0));
1353  EXPECT_EQ(std::string(), s_credit_cards_all.ColumnString(1));
1354  ASSERT_TRUE(s_credit_cards_all.Step());
1355  EXPECT_EQ(37, s_credit_cards_all.ColumnInt64(0));
1356  EXPECT_EQ("Chrome settings", s_credit_cards_all.ColumnString(1));
1357}
1358
1359TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_NoResults) {
1360  std::vector<AutofillEntry> entries;
1361  ASSERT_TRUE(table_->GetAllAutofillEntries(&entries));
1362
1363  EXPECT_EQ(0U, entries.size());
1364}
1365
1366TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_OneResult) {
1367  AutofillChangeList changes;
1368  std::map<std::string, std::vector<Time> > name_value_times_map;
1369
1370  time_t start = 0;
1371  std::vector<Time> timestamps1;
1372  FormFieldData field;
1373  field.name = ASCIIToUTF16("Name");
1374  field.value = ASCIIToUTF16("Superman");
1375  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
1376                                            Time::FromTimeT(start)));
1377  timestamps1.push_back(Time::FromTimeT(start));
1378  std::string key1("NameSuperman");
1379  name_value_times_map.insert(std::pair<std::string,
1380    std::vector<Time> > (key1, timestamps1));
1381
1382  AutofillEntrySet expected_entries(CompareAutofillEntries);
1383  AutofillKey ak1(ASCIIToUTF16("Name"), ASCIIToUTF16("Superman"));
1384  AutofillEntry ae1(ak1, timestamps1);
1385
1386  expected_entries.insert(ae1);
1387
1388  std::vector<AutofillEntry> entries;
1389  ASSERT_TRUE(table_->GetAllAutofillEntries(&entries));
1390  AutofillEntrySet entry_set(entries.begin(), entries.end(),
1391    CompareAutofillEntries);
1392
1393  // make sure the lists of entries match
1394  ASSERT_EQ(expected_entries.size(), entry_set.size());
1395  AutofillEntrySetIterator it;
1396  for (it = entry_set.begin(); it != entry_set.end(); it++) {
1397    expected_entries.erase(*it);
1398  }
1399
1400  EXPECT_EQ(0U, expected_entries.size());
1401}
1402
1403TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_TwoDistinct) {
1404  AutofillChangeList changes;
1405  std::map<std::string, std::vector<Time> > name_value_times_map;
1406  time_t start = 0;
1407
1408  std::vector<Time> timestamps1;
1409  FormFieldData field;
1410  field.name = ASCIIToUTF16("Name");
1411  field.value = ASCIIToUTF16("Superman");
1412  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
1413                                            Time::FromTimeT(start)));
1414  timestamps1.push_back(Time::FromTimeT(start));
1415  std::string key1("NameSuperman");
1416  name_value_times_map.insert(std::pair<std::string,
1417    std::vector<Time> > (key1, timestamps1));
1418
1419  start++;
1420  std::vector<Time> timestamps2;
1421  field.name = ASCIIToUTF16("Name");
1422  field.value = ASCIIToUTF16("Clark Kent");
1423  EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
1424                                            Time::FromTimeT(start)));
1425  timestamps2.push_back(Time::FromTimeT(start));
1426  std::string key2("NameClark Kent");
1427  name_value_times_map.insert(std::pair<std::string,
1428    std::vector<Time> > (key2, timestamps2));
1429
1430  AutofillEntrySet expected_entries(CompareAutofillEntries);
1431  AutofillKey ak1(ASCIIToUTF16("Name"), ASCIIToUTF16("Superman"));
1432  AutofillKey ak2(ASCIIToUTF16("Name"), ASCIIToUTF16("Clark Kent"));
1433  AutofillEntry ae1(ak1, timestamps1);
1434  AutofillEntry ae2(ak2, timestamps2);
1435
1436  expected_entries.insert(ae1);
1437  expected_entries.insert(ae2);
1438
1439  std::vector<AutofillEntry> entries;
1440  ASSERT_TRUE(table_->GetAllAutofillEntries(&entries));
1441  AutofillEntrySet entry_set(entries.begin(), entries.end(),
1442    CompareAutofillEntries);
1443
1444  // make sure the lists of entries match
1445  ASSERT_EQ(expected_entries.size(), entry_set.size());
1446  AutofillEntrySetIterator it;
1447  for (it = entry_set.begin(); it != entry_set.end(); it++) {
1448    expected_entries.erase(*it);
1449  }
1450
1451  EXPECT_EQ(0U, expected_entries.size());
1452}
1453
1454TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_TwoSame) {
1455  AutofillChangeList changes;
1456  std::map<std::string, std::vector<Time> > name_value_times_map;
1457
1458  time_t start = 0;
1459  std::vector<Time> timestamps;
1460  for (int i = 0; i < 2; i++) {
1461    FormFieldData field;
1462    field.name = ASCIIToUTF16("Name");
1463    field.value = ASCIIToUTF16("Superman");
1464    EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes,
1465                                              Time::FromTimeT(start)));
1466    timestamps.push_back(Time::FromTimeT(start));
1467    start++;
1468  }
1469
1470  std::string key("NameSuperman");
1471  name_value_times_map.insert(std::pair<std::string,
1472      std::vector<Time> > (key, timestamps));
1473
1474  AutofillEntrySet expected_entries(CompareAutofillEntries);
1475  AutofillKey ak1(ASCIIToUTF16("Name"), ASCIIToUTF16("Superman"));
1476  AutofillEntry ae1(ak1, timestamps);
1477
1478  expected_entries.insert(ae1);
1479
1480  std::vector<AutofillEntry> entries;
1481  ASSERT_TRUE(table_->GetAllAutofillEntries(&entries));
1482  AutofillEntrySet entry_set(entries.begin(), entries.end(),
1483    CompareAutofillEntries);
1484
1485  // make sure the lists of entries match
1486  ASSERT_EQ(expected_entries.size(), entry_set.size());
1487  AutofillEntrySetIterator it;
1488  for (it = entry_set.begin(); it != entry_set.end(); it++) {
1489    expected_entries.erase(*it);
1490  }
1491
1492  EXPECT_EQ(0U, expected_entries.size());
1493}
1494
1495}  // namespace autofill
1496