// Copyright (c) 2011 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 "app/sql/statement.h"
#include "base/file_util.h"
#include "base/path_service.h"
#include "base/string_number_conversions.h"
#include "base/time.h"
#include "base/utf_string_conversions.h"
#include "chrome/common/chrome_paths.h"
#include "chrome/browser/autofill/autofill_profile.h"
#include "chrome/browser/autofill/autofill_type.h"
#include "chrome/browser/autofill/credit_card.h"
#include "chrome/browser/password_manager/encryptor.h"
#include "chrome/browser/webdata/autofill_change.h"
#include "chrome/browser/webdata/autofill_entry.h"
#include "chrome/browser/webdata/web_database.h"
#include "chrome/common/guid.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/glue/form_field.h"
using base::Time;
using base::TimeDelta;
using webkit_glue::FormField;
// So we can compare AutofillKeys with EXPECT_EQ().
std::ostream& operator<<(std::ostream& os, const AutofillKey& key) {
return os << UTF16ToASCII(key.name()) << ", " << UTF16ToASCII(key.value());
}
// So we can compare AutofillChanges with EXPECT_EQ().
std::ostream& operator<<(std::ostream& os, const AutofillChange& change) {
switch (change.type()) {
case AutofillChange::ADD: {
os << "ADD";
break;
}
case AutofillChange::UPDATE: {
os << "UPDATE";
break;
}
case AutofillChange::REMOVE: {
os << "REMOVE";
break;
}
}
return os << " " << change.key();
}
namespace {
bool CompareAutofillEntries(const AutofillEntry& a, const AutofillEntry& b) {
std::set<Time> timestamps1(a.timestamps().begin(), a.timestamps().end());
std::set<Time> timestamps2(b.timestamps().begin(), b.timestamps().end());
int compVal = a.key().name().compare(b.key().name());
if (compVal != 0) {
return compVal < 0;
}
compVal = a.key().value().compare(b.key().value());
if (compVal != 0) {
return compVal < 0;
}
if (timestamps1.size() != timestamps2.size()) {
return timestamps1.size() < timestamps2.size();
}
std::set<Time>::iterator it;
for (it = timestamps1.begin(); it != timestamps1.end(); it++) {
timestamps2.erase(*it);
}
return !timestamps2.empty();
}
} // anonymous namespace
class AutofillTableTest : public testing::Test {
public:
AutofillTableTest() {}
virtual ~AutofillTableTest() {}
protected:
typedef std::vector<AutofillChange> AutofillChangeList;
typedef std::set<AutofillEntry,
bool (*)(const AutofillEntry&, const AutofillEntry&)> AutofillEntrySet;
typedef std::set<AutofillEntry, bool (*)(const AutofillEntry&,
const AutofillEntry&)>::iterator AutofillEntrySetIterator;
virtual void SetUp() {
#if defined(OS_MACOSX)
Encryptor::UseMockKeychain(true);
#endif
PathService::Get(chrome::DIR_TEST_DATA, &file_);
const std::string test_db = "TestWebDatabase" +
base::Int64ToString(Time::Now().ToTimeT()) +
".db";
file_ = file_.AppendASCII(test_db);
file_util::Delete(file_, false);
}
virtual void TearDown() {
file_util::Delete(file_, false);
}
static AutofillEntry MakeAutofillEntry(const char* name,
const char* value,
time_t timestamp0,
time_t timestamp1) {
std::vector<Time> timestamps;
if (timestamp0 >= 0)
timestamps.push_back(Time::FromTimeT(timestamp0));
if (timestamp1 >= 0)
timestamps.push_back(Time::FromTimeT(timestamp1));
return AutofillEntry(
AutofillKey(ASCIIToUTF16(name), ASCIIToUTF16(value)), timestamps);
}
FilePath file_;
private:
DISALLOW_COPY_AND_ASSIGN(AutofillTableTest);
};
TEST_F(AutofillTableTest, Autofill) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
Time t1 = Time::Now();
// Simulate the submission of a handful of entries in a field called "Name",
// some more often than others.
AutofillChangeList changes;
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes));
std::vector<string16> v;
for (int i = 0; i < 5; i++) {
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Kent"),
string16(),
0,
false),
&changes));
}
for (int i = 0; i < 3; i++) {
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Sutter"),
string16(),
0,
false),
&changes));
}
for (int i = 0; i < 2; i++) {
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("Favorite Color"),
ASCIIToUTF16("Green"),
string16(),
0,
false),
&changes));
}
int count = 0;
int64 pair_id = 0;
// We have added the name Clark Kent 5 times, so count should be 5 and pair_id
// should be somthing non-zero.
EXPECT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Kent"),
string16(),
0,
false),
&pair_id, &count));
EXPECT_EQ(5, count);
EXPECT_NE(0, pair_id);
// Storing in the data base should be case sensitive, so there should be no
// database entry for clark kent lowercase.
EXPECT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("clark kent"),
string16(),
0,
false),
&pair_id, &count));
EXPECT_EQ(0, count);
EXPECT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
FormField(string16(),
ASCIIToUTF16("Favorite Color"),
ASCIIToUTF16("Green"),
string16(),
0,
false),
&pair_id, &count));
EXPECT_EQ(2, count);
// This is meant to get a list of suggestions for Name. The empty prefix
// in the second argument means it should return all suggestions for a name
// no matter what they start with. The order that the names occur in the list
// should be decreasing order by count.
EXPECT_TRUE(db.GetAutofillTable()->GetFormValuesForElementName(
ASCIIToUTF16("Name"), string16(), &v, 6));
EXPECT_EQ(3U, v.size());
if (v.size() == 3) {
EXPECT_EQ(ASCIIToUTF16("Clark Kent"), v[0]);
EXPECT_EQ(ASCIIToUTF16("Clark Sutter"), v[1]);
EXPECT_EQ(ASCIIToUTF16("Superman"), v[2]);
}
// If we query again limiting the list size to 1, we should only get the most
// frequent entry.
EXPECT_TRUE(db.GetAutofillTable()->GetFormValuesForElementName(
ASCIIToUTF16("Name"), string16(), &v, 1));
EXPECT_EQ(1U, v.size());
if (v.size() == 1) {
EXPECT_EQ(ASCIIToUTF16("Clark Kent"), v[0]);
}
// Querying for suggestions given a prefix is case-insensitive, so the prefix
// "cLa" shoud get suggestions for both Clarks.
EXPECT_TRUE(db.GetAutofillTable()->GetFormValuesForElementName(
ASCIIToUTF16("Name"), ASCIIToUTF16("cLa"), &v, 6));
EXPECT_EQ(2U, v.size());
if (v.size() == 2) {
EXPECT_EQ(ASCIIToUTF16("Clark Kent"), v[0]);
EXPECT_EQ(ASCIIToUTF16("Clark Sutter"), v[1]);
}
// Removing all elements since the beginning of this function should remove
// everything from the database.
changes.clear();
EXPECT_TRUE(db.GetAutofillTable()->RemoveFormElementsAddedBetween(
t1, Time(), &changes));
const AutofillChange expected_changes[] = {
AutofillChange(AutofillChange::REMOVE,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"))),
AutofillChange(AutofillChange::REMOVE,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Kent"))),
AutofillChange(AutofillChange::REMOVE,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Sutter"))),
AutofillChange(AutofillChange::REMOVE,
AutofillKey(ASCIIToUTF16("Favorite Color"),
ASCIIToUTF16("Green"))),
};
EXPECT_EQ(arraysize(expected_changes), changes.size());
for (size_t i = 0; i < arraysize(expected_changes); i++) {
EXPECT_EQ(expected_changes[i], changes[i]);
}
EXPECT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Kent"),
string16(),
0,
false),
&pair_id, &count));
EXPECT_EQ(0, count);
EXPECT_TRUE(db.GetAutofillTable()->GetFormValuesForElementName(
ASCIIToUTF16("Name"), string16(), &v, 6));
EXPECT_EQ(0U, v.size());
// Now add some values with empty strings.
const string16 kValue = ASCIIToUTF16(" toto ");
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("blank"),
string16(),
string16(),
0,
false),
&changes));
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("blank"),
ASCIIToUTF16(" "),
string16(),
0,
false),
&changes));
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("blank"),
ASCIIToUTF16(" "),
string16(),
0,
false),
&changes));
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("blank"),
kValue,
string16(),
0,
false),
&changes));
// They should be stored normally as the DB layer does not check for empty
// values.
v.clear();
EXPECT_TRUE(db.GetAutofillTable()->GetFormValuesForElementName(
ASCIIToUTF16("blank"), string16(), &v, 10));
EXPECT_EQ(4U, v.size());
// Now we'll check that ClearAutofillEmptyValueElements() works as expected.
db.GetAutofillTable()->ClearAutofillEmptyValueElements();
v.clear();
EXPECT_TRUE(db.GetAutofillTable()->GetFormValuesForElementName(
ASCIIToUTF16("blank"), string16(), &v, 10));
ASSERT_EQ(1U, v.size());
EXPECT_EQ(kValue, v[0]);
}
TEST_F(AutofillTableTest, Autofill_RemoveBetweenChanges) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
TimeDelta one_day(TimeDelta::FromDays(1));
Time t1 = Time::Now();
Time t2 = t1 + one_day;
AutofillChangeList changes;
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
t1));
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
t2));
changes.clear();
EXPECT_TRUE(db.GetAutofillTable()->RemoveFormElementsAddedBetween(
t1, t2, &changes));
ASSERT_EQ(1U, changes.size());
EXPECT_EQ(AutofillChange(AutofillChange::UPDATE,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"))),
changes[0]);
changes.clear();
EXPECT_TRUE(db.GetAutofillTable()->RemoveFormElementsAddedBetween(
t2, t2 + one_day, &changes));
ASSERT_EQ(1U, changes.size());
EXPECT_EQ(AutofillChange(AutofillChange::REMOVE,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"))),
changes[0]);
}
TEST_F(AutofillTableTest, Autofill_AddChanges) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
TimeDelta one_day(TimeDelta::FromDays(1));
Time t1 = Time::Now();
Time t2 = t1 + one_day;
AutofillChangeList changes;
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
t1));
ASSERT_EQ(1U, changes.size());
EXPECT_EQ(AutofillChange(AutofillChange::ADD,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"))),
changes[0]);
changes.clear();
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
t2));
ASSERT_EQ(1U, changes.size());
EXPECT_EQ(AutofillChange(AutofillChange::UPDATE,
AutofillKey(ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"))),
changes[0]);
}
TEST_F(AutofillTableTest, Autofill_UpdateOneWithOneTimestamp) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, -1));
std::vector<AutofillEntry> entries;
entries.push_back(entry);
ASSERT_TRUE(db.GetAutofillTable()->UpdateAutofillEntries(entries));
FormField field(string16(),
ASCIIToUTF16("foo"),
ASCIIToUTF16("bar"),
string16(),
0,
false);
int64 pair_id;
int count;
ASSERT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
field, &pair_id, &count));
EXPECT_LE(0, pair_id);
EXPECT_EQ(1, count);
std::vector<AutofillEntry> all_entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&all_entries));
ASSERT_EQ(1U, all_entries.size());
EXPECT_TRUE(entry == all_entries[0]);
}
TEST_F(AutofillTableTest, Autofill_UpdateOneWithTwoTimestamps) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, 2));
std::vector<AutofillEntry> entries;
entries.push_back(entry);
ASSERT_TRUE(db.GetAutofillTable()->UpdateAutofillEntries(entries));
FormField field(string16(),
ASCIIToUTF16("foo"),
ASCIIToUTF16("bar"),
string16(),
0,
false);
int64 pair_id;
int count;
ASSERT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
field, &pair_id, &count));
EXPECT_LE(0, pair_id);
EXPECT_EQ(2, count);
std::vector<AutofillEntry> all_entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&all_entries));
ASSERT_EQ(1U, all_entries.size());
EXPECT_TRUE(entry == all_entries[0]);
}
TEST_F(AutofillTableTest, Autofill_GetAutofillTimestamps) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, 2));
std::vector<AutofillEntry> entries;
entries.push_back(entry);
ASSERT_TRUE(db.GetAutofillTable()->UpdateAutofillEntries(entries));
std::vector<Time> timestamps;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillTimestamps(ASCIIToUTF16("foo"),
ASCIIToUTF16("bar"),
×tamps));
ASSERT_EQ(2U, timestamps.size());
EXPECT_TRUE(Time::FromTimeT(1) == timestamps[0]);
EXPECT_TRUE(Time::FromTimeT(2) == timestamps[1]);
}
TEST_F(AutofillTableTest, Autofill_UpdateTwo) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillEntry entry0(MakeAutofillEntry("foo", "bar0", 1, -1));
AutofillEntry entry1(MakeAutofillEntry("foo", "bar1", 2, 3));
std::vector<AutofillEntry> entries;
entries.push_back(entry0);
entries.push_back(entry1);
ASSERT_TRUE(db.GetAutofillTable()->UpdateAutofillEntries(entries));
FormField field0(string16(),
ASCIIToUTF16("foo"),
ASCIIToUTF16("bar0"),
string16(),
0,
false);
int64 pair_id;
int count;
ASSERT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
field0, &pair_id, &count));
EXPECT_LE(0, pair_id);
EXPECT_EQ(1, count);
FormField field1(string16(),
ASCIIToUTF16("foo"),
ASCIIToUTF16("bar1"),
string16(),
0,
false);
ASSERT_TRUE(db.GetAutofillTable()->GetIDAndCountOfFormElement(
field1, &pair_id, &count));
EXPECT_LE(0, pair_id);
EXPECT_EQ(2, count);
}
TEST_F(AutofillTableTest, Autofill_UpdateReplace) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillChangeList changes;
// Add a form field. This will be replaced.
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValue(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes));
AutofillEntry entry(MakeAutofillEntry("Name", "Superman", 1, 2));
std::vector<AutofillEntry> entries;
entries.push_back(entry);
ASSERT_TRUE(db.GetAutofillTable()->UpdateAutofillEntries(entries));
std::vector<AutofillEntry> all_entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&all_entries));
ASSERT_EQ(1U, all_entries.size());
EXPECT_TRUE(entry == all_entries[0]);
}
TEST_F(AutofillTableTest, Autofill_UpdateDontReplace) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
Time t = Time::Now();
AutofillEntry existing(
MakeAutofillEntry("Name", "Superman", t.ToTimeT(), -1));
AutofillChangeList changes;
// Add a form field. This will NOT be replaced.
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
existing.key().name(),
existing.key().value(),
string16(),
0,
false),
&changes,
t));
AutofillEntry entry(MakeAutofillEntry("Name", "Clark Kent", 1, 2));
std::vector<AutofillEntry> entries;
entries.push_back(entry);
ASSERT_TRUE(db.GetAutofillTable()->UpdateAutofillEntries(entries));
std::vector<AutofillEntry> all_entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&all_entries));
ASSERT_EQ(2U, all_entries.size());
AutofillEntrySet expected_entries(all_entries.begin(),
all_entries.end(),
CompareAutofillEntries);
EXPECT_EQ(1U, expected_entries.count(existing));
EXPECT_EQ(1U, expected_entries.count(entry));
}
TEST_F(AutofillTableTest, Autofill_AddFormFieldValues) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
Time t = Time::Now();
// Add multiple values for "firstname" and "lastname" names. Test that only
// first value of each gets added. Related to security issue:
// http://crbug.com/51727.
std::vector<FormField> elements;
elements.push_back(FormField(string16(),
ASCIIToUTF16("firstname"),
ASCIIToUTF16("Joe"),
string16(),
0,
false));
elements.push_back(FormField(string16(),
ASCIIToUTF16("firstname"),
ASCIIToUTF16("Jane"),
string16(),
0,
false));
elements.push_back(FormField(string16(),
ASCIIToUTF16("lastname"),
ASCIIToUTF16("Smith"),
string16(),
0,
false));
elements.push_back(FormField(string16(),
ASCIIToUTF16("lastname"),
ASCIIToUTF16("Jones"),
string16(),
0,
false));
std::vector<AutofillChange> changes;
db.GetAutofillTable()->AddFormFieldValuesTime(elements, &changes, t);
ASSERT_EQ(2U, changes.size());
EXPECT_EQ(changes[0], AutofillChange(AutofillChange::ADD,
AutofillKey(ASCIIToUTF16("firstname"),
ASCIIToUTF16("Joe"))));
EXPECT_EQ(changes[1], AutofillChange(AutofillChange::ADD,
AutofillKey(ASCIIToUTF16("lastname"),
ASCIIToUTF16("Smith"))));
std::vector<AutofillEntry> all_entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&all_entries));
ASSERT_EQ(2U, all_entries.size());
}
TEST_F(AutofillTableTest, AutofillProfile) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
// Add a 'Home' profile.
AutofillProfile home_profile;
home_profile.SetInfo(NAME_FIRST, ASCIIToUTF16("John"));
home_profile.SetInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
home_profile.SetInfo(NAME_LAST, ASCIIToUTF16("Smith"));
home_profile.SetInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz"));
home_profile.SetInfo(COMPANY_NAME, ASCIIToUTF16("Google"));
home_profile.SetInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way"));
home_profile.SetInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5"));
home_profile.SetInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
home_profile.SetInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
home_profile.SetInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
home_profile.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
home_profile.SetInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567"));
home_profile.SetInfo(PHONE_FAX_WHOLE_NUMBER, ASCIIToUTF16("1915243678"));
Time pre_creation_time = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(home_profile));
Time post_creation_time = Time::Now();
// Get the 'Home' profile.
AutofillProfile* db_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(
home_profile.guid(), &db_profile));
EXPECT_EQ(home_profile, *db_profile);
sql::Statement s_home(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified "
"FROM autofill_profiles WHERE guid=?"));
s_home.BindString(0, home_profile.guid());
ASSERT_TRUE(s_home);
ASSERT_TRUE(s_home.Step());
EXPECT_GE(s_home.ColumnInt64(0), pre_creation_time.ToTimeT());
EXPECT_LE(s_home.ColumnInt64(0), post_creation_time.ToTimeT());
EXPECT_FALSE(s_home.Step());
delete db_profile;
// Add a 'Billing' profile.
AutofillProfile billing_profile = home_profile;
billing_profile.set_guid(guid::GenerateGUID());
billing_profile.SetInfo(ADDRESS_HOME_LINE1,
ASCIIToUTF16("5678 Bottom Street"));
billing_profile.SetInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("suite 3"));
pre_creation_time = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(billing_profile));
post_creation_time = Time::Now();
// Get the 'Billing' profile.
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(
billing_profile.guid(), &db_profile));
EXPECT_EQ(billing_profile, *db_profile);
sql::Statement s_billing(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles WHERE guid=?"));
s_billing.BindString(0, billing_profile.guid());
ASSERT_TRUE(s_billing);
ASSERT_TRUE(s_billing.Step());
EXPECT_GE(s_billing.ColumnInt64(0), pre_creation_time.ToTimeT());
EXPECT_LE(s_billing.ColumnInt64(0), post_creation_time.ToTimeT());
EXPECT_FALSE(s_billing.Step());
delete db_profile;
// Update the 'Billing' profile, name only.
billing_profile.SetInfo(NAME_FIRST, ASCIIToUTF16("Jane"));
Time pre_modification_time = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(
billing_profile));
Time post_modification_time = Time::Now();
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(
billing_profile.guid(), &db_profile));
EXPECT_EQ(billing_profile, *db_profile);
sql::Statement s_billing_updated(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles WHERE guid=?"));
s_billing_updated.BindString(0, billing_profile.guid());
ASSERT_TRUE(s_billing_updated);
ASSERT_TRUE(s_billing_updated.Step());
EXPECT_GE(s_billing_updated.ColumnInt64(0),
pre_modification_time.ToTimeT());
EXPECT_LE(s_billing_updated.ColumnInt64(0),
post_modification_time.ToTimeT());
EXPECT_FALSE(s_billing_updated.Step());
delete db_profile;
// Update the 'Billing' profile.
billing_profile.SetInfo(NAME_FIRST, ASCIIToUTF16("Janice"));
billing_profile.SetInfo(NAME_MIDDLE, ASCIIToUTF16("C."));
billing_profile.SetInfo(NAME_FIRST, ASCIIToUTF16("Joplin"));
billing_profile.SetInfo(EMAIL_ADDRESS, ASCIIToUTF16("jane@singer.com"));
billing_profile.SetInfo(COMPANY_NAME, ASCIIToUTF16("Indy"));
billing_profile.SetInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("Open Road"));
billing_profile.SetInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("Route 66"));
billing_profile.SetInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("NFA"));
billing_profile.SetInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("NY"));
billing_profile.SetInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("10011"));
billing_profile.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("United States"));
billing_profile.SetInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181230000"));
billing_profile.SetInfo(PHONE_FAX_WHOLE_NUMBER, ASCIIToUTF16("1915240000"));
Time pre_modification_time_2 = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(
billing_profile));
Time post_modification_time_2 = Time::Now();
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(
billing_profile.guid(), &db_profile));
EXPECT_EQ(billing_profile, *db_profile);
sql::Statement s_billing_updated_2(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles WHERE guid=?"));
s_billing_updated_2.BindString(0, billing_profile.guid());
ASSERT_TRUE(s_billing_updated_2);
ASSERT_TRUE(s_billing_updated_2.Step());
EXPECT_GE(s_billing_updated_2.ColumnInt64(0),
pre_modification_time_2.ToTimeT());
EXPECT_LE(s_billing_updated_2.ColumnInt64(0),
post_modification_time_2.ToTimeT());
EXPECT_FALSE(s_billing_updated_2.Step());
delete db_profile;
// Remove the 'Billing' profile.
EXPECT_TRUE(db.GetAutofillTable()->RemoveAutofillProfile(
billing_profile.guid()));
EXPECT_FALSE(db.GetAutofillTable()->GetAutofillProfile(
billing_profile.guid(), &db_profile));
}
TEST_F(AutofillTableTest, AutofillProfileMultiValueNames) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillProfile p;
const string16 kJohnDoe(ASCIIToUTF16("John Doe"));
const string16 kJohnPDoe(ASCIIToUTF16("John P. Doe"));
std::vector<string16> set_values;
set_values.push_back(kJohnDoe);
set_values.push_back(kJohnPDoe);
p.SetMultiInfo(NAME_FULL, set_values);
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(p));
AutofillProfile* db_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Update the values.
const string16 kNoOne(ASCIIToUTF16("No One"));
set_values[1] = kNoOne;
p.SetMultiInfo(NAME_FULL, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Delete values.
set_values.clear();
p.SetMultiInfo(NAME_FULL, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
EXPECT_EQ(string16(), db_profile->GetInfo(NAME_FULL));
delete db_profile;
}
TEST_F(AutofillTableTest, AutofillProfileSingleValue) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillProfile p;
const string16 kJohnDoe(ASCIIToUTF16("John Doe"));
const string16 kJohnPDoe(ASCIIToUTF16("John P. Doe"));
std::vector<string16> set_values;
set_values.push_back(kJohnDoe);
set_values.push_back(kJohnPDoe);
p.SetMultiInfo(NAME_FULL, set_values);
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(p));
AutofillProfile* db_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Update the values. This update is the "single value" update, it should
// not perturb the multi-values following the zeroth entry. This simulates
// the Sync use-case until Sync can be changed to be multi-value aware.
const string16 kNoOne(ASCIIToUTF16("No One"));
set_values.resize(1);
set_values[0] = kNoOne;
p.SetMultiInfo(NAME_FULL, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfile(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_NE(0, p.CompareMulti(*db_profile));
db_profile->GetMultiInfo(NAME_FULL, &set_values);
ASSERT_EQ(2UL, set_values.size());
EXPECT_EQ(kNoOne, set_values[0]);
EXPECT_EQ(kJohnPDoe, set_values[1]);
delete db_profile;
}
TEST_F(AutofillTableTest, AutofillProfileMultiValueEmails) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillProfile p;
const string16 kJohnDoe(ASCIIToUTF16("john@doe.com"));
const string16 kJohnPDoe(ASCIIToUTF16("john_p@doe.com"));
std::vector<string16> set_values;
set_values.push_back(kJohnDoe);
set_values.push_back(kJohnPDoe);
p.SetMultiInfo(EMAIL_ADDRESS, set_values);
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(p));
AutofillProfile* db_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Update the values.
const string16 kNoOne(ASCIIToUTF16("no@one.com"));
set_values[1] = kNoOne;
p.SetMultiInfo(EMAIL_ADDRESS, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Delete values.
set_values.clear();
p.SetMultiInfo(EMAIL_ADDRESS, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
EXPECT_EQ(string16(), db_profile->GetInfo(EMAIL_ADDRESS));
delete db_profile;
}
TEST_F(AutofillTableTest, AutofillProfileMultiValuePhone) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillProfile p;
const string16 kJohnDoe(ASCIIToUTF16("4151112222"));
const string16 kJohnPDoe(ASCIIToUTF16("4151113333"));
std::vector<string16> set_values;
set_values.push_back(kJohnDoe);
set_values.push_back(kJohnPDoe);
p.SetMultiInfo(PHONE_HOME_WHOLE_NUMBER, set_values);
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(p));
AutofillProfile* db_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Update the values.
const string16 kNoOne(ASCIIToUTF16("4151110000"));
set_values[1] = kNoOne;
p.SetMultiInfo(PHONE_HOME_WHOLE_NUMBER, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Delete values.
set_values.clear();
p.SetMultiInfo(PHONE_HOME_WHOLE_NUMBER, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
EXPECT_EQ(string16(), db_profile->GetInfo(EMAIL_ADDRESS));
delete db_profile;
}
TEST_F(AutofillTableTest, AutofillProfileMultiValueFax) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillProfile p;
const string16 kJohnDoe(ASCIIToUTF16("4151112222"));
const string16 kJohnPDoe(ASCIIToUTF16("4151113333"));
std::vector<string16> set_values;
set_values.push_back(kJohnDoe);
set_values.push_back(kJohnPDoe);
p.SetMultiInfo(PHONE_FAX_WHOLE_NUMBER, set_values);
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(p));
AutofillProfile* db_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Update the values.
const string16 kNoOne(ASCIIToUTF16("4151110000"));
set_values[1] = kNoOne;
p.SetMultiInfo(PHONE_FAX_WHOLE_NUMBER, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
delete db_profile;
// Delete values.
set_values.clear();
p.SetMultiInfo(PHONE_FAX_WHOLE_NUMBER, set_values);
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(p));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(p.guid(), &db_profile));
EXPECT_EQ(p, *db_profile);
EXPECT_EQ(0, p.CompareMulti(*db_profile));
EXPECT_EQ(string16(), db_profile->GetInfo(EMAIL_ADDRESS));
delete db_profile;
}
TEST_F(AutofillTableTest, AutofillProfileTrash) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
std::vector<std::string> guids;
db.GetAutofillTable()->GetAutofillProfilesInTrash(&guids);
EXPECT_TRUE(guids.empty());
ASSERT_TRUE(db.GetAutofillTable()->AddAutofillGUIDToTrash(
"00000000-0000-0000-0000-000000000000"));
ASSERT_TRUE(db.GetAutofillTable()->AddAutofillGUIDToTrash(
"00000000-0000-0000-0000-000000000001"));
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfilesInTrash(&guids));
EXPECT_EQ(2UL, guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000000", guids[0]);
EXPECT_EQ("00000000-0000-0000-0000-000000000001", guids[1]);
ASSERT_TRUE(db.GetAutofillTable()->EmptyAutofillProfilesTrash());
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfilesInTrash(&guids));
EXPECT_TRUE(guids.empty());
}
TEST_F(AutofillTableTest, AutofillProfileTrashInteraction) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
std::vector<std::string> guids;
db.GetAutofillTable()->GetAutofillProfilesInTrash(&guids);
EXPECT_TRUE(guids.empty());
AutofillProfile profile;
profile.SetInfo(NAME_FIRST, ASCIIToUTF16("John"));
profile.SetInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
profile.SetInfo(NAME_LAST, ASCIIToUTF16("Smith"));
profile.SetInfo(EMAIL_ADDRESS,ASCIIToUTF16("js@smith.xyz"));
profile.SetInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1 Main St"));
profile.SetInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
profile.SetInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
profile.SetInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
profile.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
// Mark this profile as in the trash. This stops |AddAutofillProfile| from
// adding it.
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillGUIDToTrash(profile.guid()));
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(profile));
AutofillProfile* added_profile = NULL;
EXPECT_FALSE(db.GetAutofillTable()->GetAutofillProfile(
profile.guid(), &added_profile));
EXPECT_EQ(static_cast<AutofillProfile*>(NULL), added_profile);
// Add the profile for real this time.
EXPECT_TRUE(db.GetAutofillTable()->EmptyAutofillProfilesTrash());
EXPECT_TRUE(db.GetAutofillTable()->GetAutofillProfilesInTrash(&guids));
EXPECT_TRUE(guids.empty());
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillProfile(profile));
EXPECT_TRUE(db.GetAutofillTable()->GetAutofillProfile(profile.guid(),
&added_profile));
ASSERT_NE(static_cast<AutofillProfile*>(NULL), added_profile);
delete added_profile;
// Mark this profile as in the trash. This stops |UpdateAutofillProfileMulti|
// from updating it. In normal operation a profile should not be both in the
// trash and in the profiles table simultaneously.
EXPECT_TRUE(db.GetAutofillTable()->AddAutofillGUIDToTrash(profile.guid()));
profile.SetInfo(NAME_FIRST, ASCIIToUTF16("Jane"));
EXPECT_TRUE(db.GetAutofillTable()->UpdateAutofillProfileMulti(profile));
AutofillProfile* updated_profile = NULL;
EXPECT_TRUE(db.GetAutofillTable()->GetAutofillProfile(
profile.guid(), &updated_profile));
ASSERT_NE(static_cast<AutofillProfile*>(NULL), added_profile);
EXPECT_EQ(ASCIIToUTF16("John"), updated_profile->GetInfo(NAME_FIRST));
delete updated_profile;
// Try to delete the trashed profile. This stops |RemoveAutofillProfile| from
// deleting it. In normal operation deletion is done by migration step, and
// removal from trash is done by |WebDataService|. |RemoveAutofillProfile|
// does remove the item from the trash if it is found however, so that if
// other clients remove it (via Sync say) then it is gone and doesn't need to
// be processed further by |WebDataService|.
EXPECT_TRUE(db.GetAutofillTable()->RemoveAutofillProfile(profile.guid()));
AutofillProfile* removed_profile = NULL;
EXPECT_TRUE(db.GetAutofillTable()->GetAutofillProfile(profile.guid(),
&removed_profile));
EXPECT_FALSE(db.GetAutofillTable()->IsAutofillGUIDInTrash(profile.guid()));
ASSERT_NE(static_cast<AutofillProfile*>(NULL), removed_profile);
delete removed_profile;
// Check that emptying the trash now allows removal to occur.
EXPECT_TRUE(db.GetAutofillTable()->EmptyAutofillProfilesTrash());
EXPECT_TRUE(db.GetAutofillTable()->RemoveAutofillProfile(profile.guid()));
removed_profile = NULL;
EXPECT_FALSE(db.GetAutofillTable()->GetAutofillProfile(profile.guid(),
&removed_profile));
EXPECT_EQ(static_cast<AutofillProfile*>(NULL), removed_profile);
}
TEST_F(AutofillTableTest, CreditCard) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
// Add a 'Work' credit card.
CreditCard work_creditcard;
work_creditcard.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
work_creditcard.SetInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1234567890123456"));
work_creditcard.SetInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04"));
work_creditcard.SetInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2013"));
Time pre_creation_time = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->AddCreditCard(work_creditcard));
Time post_creation_time = Time::Now();
// Get the 'Work' credit card.
CreditCard* db_creditcard;
ASSERT_TRUE(db.GetAutofillTable()->GetCreditCard(work_creditcard.guid(),
&db_creditcard));
EXPECT_EQ(work_creditcard, *db_creditcard);
sql::Statement s_work(db.GetSQLConnection()->GetUniqueStatement(
"SELECT guid, name_on_card, expiration_month, expiration_year, "
"card_number_encrypted, date_modified "
"FROM credit_cards WHERE guid=?"));
s_work.BindString(0, work_creditcard.guid());
ASSERT_TRUE(s_work);
ASSERT_TRUE(s_work.Step());
EXPECT_GE(s_work.ColumnInt64(5), pre_creation_time.ToTimeT());
EXPECT_LE(s_work.ColumnInt64(5), post_creation_time.ToTimeT());
EXPECT_FALSE(s_work.Step());
delete db_creditcard;
// Add a 'Target' credit card.
CreditCard target_creditcard;
target_creditcard.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
target_creditcard.SetInfo(CREDIT_CARD_NUMBER,
ASCIIToUTF16("1111222233334444"));
target_creditcard.SetInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("06"));
target_creditcard.SetInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2012"));
pre_creation_time = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->AddCreditCard(target_creditcard));
post_creation_time = Time::Now();
ASSERT_TRUE(db.GetAutofillTable()->GetCreditCard(target_creditcard.guid(),
&db_creditcard));
EXPECT_EQ(target_creditcard, *db_creditcard);
sql::Statement s_target(db.GetSQLConnection()->GetUniqueStatement(
"SELECT guid, name_on_card, expiration_month, expiration_year, "
"card_number_encrypted, date_modified "
"FROM credit_cards WHERE guid=?"));
s_target.BindString(0, target_creditcard.guid());
ASSERT_TRUE(s_target);
ASSERT_TRUE(s_target.Step());
EXPECT_GE(s_target.ColumnInt64(5), pre_creation_time.ToTimeT());
EXPECT_LE(s_target.ColumnInt64(5), post_creation_time.ToTimeT());
EXPECT_FALSE(s_target.Step());
delete db_creditcard;
// Update the 'Target' credit card.
target_creditcard.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Charles Grady"));
Time pre_modification_time = Time::Now();
EXPECT_TRUE(db.GetAutofillTable()->UpdateCreditCard(target_creditcard));
Time post_modification_time = Time::Now();
ASSERT_TRUE(db.GetAutofillTable()->GetCreditCard(target_creditcard.guid(),
&db_creditcard));
EXPECT_EQ(target_creditcard, *db_creditcard);
sql::Statement s_target_updated(db.GetSQLConnection()->GetUniqueStatement(
"SELECT guid, name_on_card, expiration_month, expiration_year, "
"card_number_encrypted, date_modified "
"FROM credit_cards WHERE guid=?"));
s_target_updated.BindString(0, target_creditcard.guid());
ASSERT_TRUE(s_target_updated);
ASSERT_TRUE(s_target_updated.Step());
EXPECT_GE(s_target_updated.ColumnInt64(5), pre_modification_time.ToTimeT());
EXPECT_LE(s_target_updated.ColumnInt64(5), post_modification_time.ToTimeT());
EXPECT_FALSE(s_target_updated.Step());
delete db_creditcard;
// Remove the 'Target' credit card.
EXPECT_TRUE(db.GetAutofillTable()->RemoveCreditCard(
target_creditcard.guid()));
EXPECT_FALSE(db.GetAutofillTable()->GetCreditCard(target_creditcard.guid(),
&db_creditcard));
}
TEST_F(AutofillTableTest, UpdateAutofillProfile) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
// Add a profile to the db.
AutofillProfile profile;
profile.SetInfo(NAME_FIRST, ASCIIToUTF16("John"));
profile.SetInfo(NAME_MIDDLE, ASCIIToUTF16("Q."));
profile.SetInfo(NAME_LAST, ASCIIToUTF16("Smith"));
profile.SetInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@example.com"));
profile.SetInfo(COMPANY_NAME, ASCIIToUTF16("Google"));
profile.SetInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way"));
profile.SetInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5"));
profile.SetInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles"));
profile.SetInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA"));
profile.SetInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025"));
profile.SetInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US"));
profile.SetInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567"));
profile.SetInfo(PHONE_FAX_WHOLE_NUMBER, ASCIIToUTF16("1915243678"));
db.GetAutofillTable()->AddAutofillProfile(profile);
// Set a mocked value for the profile's creation time.
const time_t mock_creation_date = Time::Now().ToTimeT() - 13;
sql::Statement s_mock_creation_date(db.GetSQLConnection()->GetUniqueStatement(
"UPDATE autofill_profiles SET date_modified = ?"));
ASSERT_TRUE(s_mock_creation_date);
s_mock_creation_date.BindInt64(0, mock_creation_date);
ASSERT_TRUE(s_mock_creation_date.Run());
// Get the profile.
AutofillProfile* tmp_profile;
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(profile.guid(),
&tmp_profile));
scoped_ptr<AutofillProfile> db_profile(tmp_profile);
EXPECT_EQ(profile, *db_profile);
sql::Statement s_original(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles"));
ASSERT_TRUE(s_original);
ASSERT_TRUE(s_original.Step());
EXPECT_EQ(mock_creation_date, s_original.ColumnInt64(0));
EXPECT_FALSE(s_original.Step());
// Now, update the profile and save the update to the database.
// The modification date should change to reflect the update.
profile.SetInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz"));
db.GetAutofillTable()->UpdateAutofillProfileMulti(profile);
// Get the profile.
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(profile.guid(),
&tmp_profile));
db_profile.reset(tmp_profile);
EXPECT_EQ(profile, *db_profile);
sql::Statement s_updated(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles"));
ASSERT_TRUE(s_updated);
ASSERT_TRUE(s_updated.Step());
EXPECT_LT(mock_creation_date, s_updated.ColumnInt64(0));
EXPECT_FALSE(s_updated.Step());
// Set a mocked value for the profile's modification time.
const time_t mock_modification_date = Time::Now().ToTimeT() - 7;
sql::Statement s_mock_modification_date(
db.GetSQLConnection()->GetUniqueStatement(
"UPDATE autofill_profiles SET date_modified = ?"));
ASSERT_TRUE(s_mock_modification_date);
s_mock_modification_date.BindInt64(0, mock_modification_date);
ASSERT_TRUE(s_mock_modification_date.Run());
// Finally, call into |UpdateAutofillProfileMulti()| without changing the
// profile. The modification date should not change.
db.GetAutofillTable()->UpdateAutofillProfileMulti(profile);
// Get the profile.
ASSERT_TRUE(db.GetAutofillTable()->GetAutofillProfile(profile.guid(),
&tmp_profile));
db_profile.reset(tmp_profile);
EXPECT_EQ(profile, *db_profile);
sql::Statement s_unchanged(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles"));
ASSERT_TRUE(s_unchanged);
ASSERT_TRUE(s_unchanged.Step());
EXPECT_EQ(mock_modification_date, s_unchanged.ColumnInt64(0));
EXPECT_FALSE(s_unchanged.Step());
}
TEST_F(AutofillTableTest, UpdateCreditCard) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
// Add a credit card to the db.
CreditCard credit_card;
credit_card.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Jack Torrance"));
credit_card.SetInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1234567890123456"));
credit_card.SetInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04"));
credit_card.SetInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2013"));
db.GetAutofillTable()->AddCreditCard(credit_card);
// Set a mocked value for the credit card's creation time.
const time_t mock_creation_date = Time::Now().ToTimeT() - 13;
sql::Statement s_mock_creation_date(db.GetSQLConnection()->GetUniqueStatement(
"UPDATE credit_cards SET date_modified = ?"));
ASSERT_TRUE(s_mock_creation_date);
s_mock_creation_date.BindInt64(0, mock_creation_date);
ASSERT_TRUE(s_mock_creation_date.Run());
// Get the credit card.
CreditCard* tmp_credit_card;
ASSERT_TRUE(db.GetAutofillTable()->GetCreditCard(credit_card.guid(),
&tmp_credit_card));
scoped_ptr<CreditCard> db_credit_card(tmp_credit_card);
EXPECT_EQ(credit_card, *db_credit_card);
sql::Statement s_original(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM credit_cards"));
ASSERT_TRUE(s_original);
ASSERT_TRUE(s_original.Step());
EXPECT_EQ(mock_creation_date, s_original.ColumnInt64(0));
EXPECT_FALSE(s_original.Step());
// Now, update the credit card and save the update to the database.
// The modification date should change to reflect the update.
credit_card.SetInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("01"));
db.GetAutofillTable()->UpdateCreditCard(credit_card);
// Get the credit card.
ASSERT_TRUE(db.GetAutofillTable()->GetCreditCard(credit_card.guid(),
&tmp_credit_card));
db_credit_card.reset(tmp_credit_card);
EXPECT_EQ(credit_card, *db_credit_card);
sql::Statement s_updated(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM credit_cards"));
ASSERT_TRUE(s_updated);
ASSERT_TRUE(s_updated.Step());
EXPECT_LT(mock_creation_date, s_updated.ColumnInt64(0));
EXPECT_FALSE(s_updated.Step());
// Set a mocked value for the credit card's modification time.
const time_t mock_modification_date = Time::Now().ToTimeT() - 7;
sql::Statement s_mock_modification_date(
db.GetSQLConnection()->GetUniqueStatement(
"UPDATE credit_cards SET date_modified = ?"));
ASSERT_TRUE(s_mock_modification_date);
s_mock_modification_date.BindInt64(0, mock_modification_date);
ASSERT_TRUE(s_mock_modification_date.Run());
// Finally, call into |UpdateCreditCard()| without changing the credit card.
// The modification date should not change.
db.GetAutofillTable()->UpdateCreditCard(credit_card);
// Get the profile.
ASSERT_TRUE(db.GetAutofillTable()->GetCreditCard(credit_card.guid(),
&tmp_credit_card));
db_credit_card.reset(tmp_credit_card);
EXPECT_EQ(credit_card, *db_credit_card);
sql::Statement s_unchanged(db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM credit_cards"));
ASSERT_TRUE(s_unchanged);
ASSERT_TRUE(s_unchanged.Step());
EXPECT_EQ(mock_modification_date, s_unchanged.ColumnInt64(0));
EXPECT_FALSE(s_unchanged.Step());
}
TEST_F(AutofillTableTest, RemoveAutofillProfilesAndCreditCardsModifiedBetween) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
// Populate the autofill_profiles and credit_cards tables.
ASSERT_TRUE(db.GetSQLConnection()->Execute(
"INSERT INTO autofill_profiles (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000000', 11);"
"INSERT INTO autofill_profiles (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000001', 21);"
"INSERT INTO autofill_profiles (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000002', 31);"
"INSERT INTO autofill_profiles (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000003', 41);"
"INSERT INTO autofill_profiles (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000004', 51);"
"INSERT INTO autofill_profiles (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000005', 61);"
"INSERT INTO credit_cards (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000006', 17);"
"INSERT INTO credit_cards (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000007', 27);"
"INSERT INTO credit_cards (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000008', 37);"
"INSERT INTO credit_cards (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000009', 47);"
"INSERT INTO credit_cards (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000010', 57);"
"INSERT INTO credit_cards (guid, date_modified) "
"VALUES('00000000-0000-0000-0000-000000000011', 67);"));
// Remove all entries modified in the bounded time range [17,41).
std::vector<std::string> profile_guids;
std::vector<std::string> credit_card_guids;
db.GetAutofillTable()->RemoveAutofillProfilesAndCreditCardsModifiedBetween(
Time::FromTimeT(17), Time::FromTimeT(41),
&profile_guids, &credit_card_guids);
ASSERT_EQ(2UL, profile_guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000001", profile_guids[0]);
EXPECT_EQ("00000000-0000-0000-0000-000000000002", profile_guids[1]);
sql::Statement s_autofill_profiles_bounded(
db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles"));
ASSERT_TRUE(s_autofill_profiles_bounded);
ASSERT_TRUE(s_autofill_profiles_bounded.Step());
EXPECT_EQ(11, s_autofill_profiles_bounded.ColumnInt64(0));
ASSERT_TRUE(s_autofill_profiles_bounded.Step());
EXPECT_EQ(41, s_autofill_profiles_bounded.ColumnInt64(0));
ASSERT_TRUE(s_autofill_profiles_bounded.Step());
EXPECT_EQ(51, s_autofill_profiles_bounded.ColumnInt64(0));
ASSERT_TRUE(s_autofill_profiles_bounded.Step());
EXPECT_EQ(61, s_autofill_profiles_bounded.ColumnInt64(0));
EXPECT_FALSE(s_autofill_profiles_bounded.Step());
ASSERT_EQ(3UL, credit_card_guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000006", credit_card_guids[0]);
EXPECT_EQ("00000000-0000-0000-0000-000000000007", credit_card_guids[1]);
EXPECT_EQ("00000000-0000-0000-0000-000000000008", credit_card_guids[2]);
sql::Statement s_credit_cards_bounded(
db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM credit_cards"));
ASSERT_TRUE(s_credit_cards_bounded);
ASSERT_TRUE(s_credit_cards_bounded.Step());
EXPECT_EQ(47, s_credit_cards_bounded.ColumnInt64(0));
ASSERT_TRUE(s_credit_cards_bounded.Step());
EXPECT_EQ(57, s_credit_cards_bounded.ColumnInt64(0));
ASSERT_TRUE(s_credit_cards_bounded.Step());
EXPECT_EQ(67, s_credit_cards_bounded.ColumnInt64(0));
EXPECT_FALSE(s_credit_cards_bounded.Step());
// Remove all entries modified on or after time 51 (unbounded range).
db.GetAutofillTable()->RemoveAutofillProfilesAndCreditCardsModifiedBetween(
Time::FromTimeT(51), Time(),
&profile_guids, &credit_card_guids);
ASSERT_EQ(2UL, profile_guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000004", profile_guids[0]);
EXPECT_EQ("00000000-0000-0000-0000-000000000005", profile_guids[1]);
sql::Statement s_autofill_profiles_unbounded(
db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles"));
ASSERT_TRUE(s_autofill_profiles_unbounded);
ASSERT_TRUE(s_autofill_profiles_unbounded.Step());
EXPECT_EQ(11, s_autofill_profiles_unbounded.ColumnInt64(0));
ASSERT_TRUE(s_autofill_profiles_unbounded.Step());
EXPECT_EQ(41, s_autofill_profiles_unbounded.ColumnInt64(0));
EXPECT_FALSE(s_autofill_profiles_unbounded.Step());
ASSERT_EQ(2UL, credit_card_guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000010", credit_card_guids[0]);
EXPECT_EQ("00000000-0000-0000-0000-000000000011", credit_card_guids[1]);
sql::Statement s_credit_cards_unbounded(
db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM credit_cards"));
ASSERT_TRUE(s_credit_cards_unbounded);
ASSERT_TRUE(s_credit_cards_unbounded.Step());
EXPECT_EQ(47, s_credit_cards_unbounded.ColumnInt64(0));
EXPECT_FALSE(s_credit_cards_unbounded.Step());
// Remove all remaining entries.
db.GetAutofillTable()->RemoveAutofillProfilesAndCreditCardsModifiedBetween(
Time(), Time(),
&profile_guids, &credit_card_guids);
ASSERT_EQ(2UL, profile_guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000000", profile_guids[0]);
EXPECT_EQ("00000000-0000-0000-0000-000000000003", profile_guids[1]);
sql::Statement s_autofill_profiles_empty(
db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM autofill_profiles"));
ASSERT_TRUE(s_autofill_profiles_empty);
EXPECT_FALSE(s_autofill_profiles_empty.Step());
ASSERT_EQ(1UL, credit_card_guids.size());
EXPECT_EQ("00000000-0000-0000-0000-000000000009", credit_card_guids[0]);
sql::Statement s_credit_cards_empty(
db.GetSQLConnection()->GetUniqueStatement(
"SELECT date_modified FROM credit_cards"));
ASSERT_TRUE(s_credit_cards_empty);
EXPECT_FALSE(s_credit_cards_empty.Step());
}
TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_NoResults) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
std::vector<AutofillEntry> entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&entries));
EXPECT_EQ(0U, entries.size());
}
TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_OneResult) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillChangeList changes;
std::map<std::string, std::vector<Time> > name_value_times_map;
time_t start = 0;
std::vector<Time> timestamps1;
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
Time::FromTimeT(start)));
timestamps1.push_back(Time::FromTimeT(start));
std::string key1("NameSuperman");
name_value_times_map.insert(std::pair<std::string,
std::vector<Time> > (key1, timestamps1));
AutofillEntrySet expected_entries(CompareAutofillEntries);
AutofillKey ak1(ASCIIToUTF16("Name"), ASCIIToUTF16("Superman"));
AutofillEntry ae1(ak1, timestamps1);
expected_entries.insert(ae1);
std::vector<AutofillEntry> entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&entries));
AutofillEntrySet entry_set(entries.begin(), entries.end(),
CompareAutofillEntries);
// make sure the lists of entries match
ASSERT_EQ(expected_entries.size(), entry_set.size());
AutofillEntrySetIterator it;
for (it = entry_set.begin(); it != entry_set.end(); it++) {
expected_entries.erase(*it);
}
EXPECT_EQ(0U, expected_entries.size());
}
TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_TwoDistinct) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillChangeList changes;
std::map<std::string, std::vector<Time> > name_value_times_map;
time_t start = 0;
std::vector<Time> timestamps1;
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
Time::FromTimeT(start)));
timestamps1.push_back(Time::FromTimeT(start));
std::string key1("NameSuperman");
name_value_times_map.insert(std::pair<std::string,
std::vector<Time> > (key1, timestamps1));
start++;
std::vector<Time> timestamps2;
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Clark Kent"),
string16(),
0,
false),
&changes,
Time::FromTimeT(start)));
timestamps2.push_back(Time::FromTimeT(start));
std::string key2("NameClark Kent");
name_value_times_map.insert(std::pair<std::string,
std::vector<Time> > (key2, timestamps2));
AutofillEntrySet expected_entries(CompareAutofillEntries);
AutofillKey ak1(ASCIIToUTF16("Name"), ASCIIToUTF16("Superman"));
AutofillKey ak2(ASCIIToUTF16("Name"), ASCIIToUTF16("Clark Kent"));
AutofillEntry ae1(ak1, timestamps1);
AutofillEntry ae2(ak2, timestamps2);
expected_entries.insert(ae1);
expected_entries.insert(ae2);
std::vector<AutofillEntry> entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&entries));
AutofillEntrySet entry_set(entries.begin(), entries.end(),
CompareAutofillEntries);
// make sure the lists of entries match
ASSERT_EQ(expected_entries.size(), entry_set.size());
AutofillEntrySetIterator it;
for (it = entry_set.begin(); it != entry_set.end(); it++) {
expected_entries.erase(*it);
}
EXPECT_EQ(0U, expected_entries.size());
}
TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_TwoSame) {
WebDatabase db;
ASSERT_EQ(sql::INIT_OK, db.Init(file_));
AutofillChangeList changes;
std::map<std::string, std::vector<Time> > name_value_times_map;
time_t start = 0;
std::vector<Time> timestamps;
for (int i = 0; i < 2; i++) {
EXPECT_TRUE(db.GetAutofillTable()->AddFormFieldValueTime(
FormField(string16(),
ASCIIToUTF16("Name"),
ASCIIToUTF16("Superman"),
string16(),
0,
false),
&changes,
Time::FromTimeT(start)));
timestamps.push_back(Time::FromTimeT(start));
start++;
}
std::string key("NameSuperman");
name_value_times_map.insert(std::pair<std::string,
std::vector<Time> > (key, timestamps));
AutofillEntrySet expected_entries(CompareAutofillEntries);
AutofillKey ak1(ASCIIToUTF16("Name"), ASCIIToUTF16("Superman"));
AutofillEntry ae1(ak1, timestamps);
expected_entries.insert(ae1);
std::vector<AutofillEntry> entries;
ASSERT_TRUE(db.GetAutofillTable()->GetAllAutofillEntries(&entries));
AutofillEntrySet entry_set(entries.begin(), entries.end(),
CompareAutofillEntries);
// make sure the lists of entries match
ASSERT_EQ(expected_entries.size(), entry_set.size());
AutofillEntrySetIterator it;
for (it = entry_set.begin(); it != entry_set.end(); it++) {
expected_entries.erase(*it);
}
EXPECT_EQ(0U, expected_entries.size());
}