| // Copyright 2013 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 <stddef.h> |
| |
| #include <map> |
| #include <set> |
| #include <string> |
| #include <tuple> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/command_line.h" |
| #include "base/files/file_util.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/guid.h" |
| #include "base/macros.h" |
| #include "base/memory/scoped_vector.h" |
| #include "base/stl_util.h" |
| #include "base/strings/string_number_conversions.h" |
| #include "base/strings/string_util.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "base/time/time.h" |
| #include "build/build_config.h" |
| #include "components/autofill/core/browser/autofill_profile.h" |
| #include "components/autofill/core/browser/autofill_test_utils.h" |
| #include "components/autofill/core/browser/autofill_type.h" |
| #include "components/autofill/core/browser/credit_card.h" |
| #include "components/autofill/core/browser/webdata/autofill_change.h" |
| #include "components/autofill/core/browser/webdata/autofill_entry.h" |
| #include "components/autofill/core/browser/webdata/autofill_table.h" |
| #include "components/autofill/core/common/autofill_constants.h" |
| #include "components/autofill/core/common/autofill_switches.h" |
| #include "components/autofill/core/common/autofill_util.h" |
| #include "components/autofill/core/common/form_field_data.h" |
| #include "components/os_crypt/os_crypt_mocker.h" |
| #include "components/webdata/common/web_database.h" |
| #include "sql/statement.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using base::ASCIIToUTF16; |
| using base::Time; |
| using base::TimeDelta; |
| |
| namespace autofill { |
| |
| // So we can compare AutofillKeys with EXPECT_EQ(). |
| std::ostream& operator<<(std::ostream& os, const AutofillKey& key) { |
| return os << base::UTF16ToASCII(key.name()) << ", " |
| << base::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 { |
| |
| typedef std::set<AutofillEntry, |
| bool (*)(const AutofillEntry&, const AutofillEntry&)> AutofillEntrySet; |
| typedef AutofillEntrySet::iterator AutofillEntrySetIterator; |
| |
| bool CompareAutofillEntries(const AutofillEntry& a, const AutofillEntry& b) { |
| return std::tie(a.key().name(), a.key().value(), |
| a.date_created(), a.date_last_used()) < |
| std::tie(b.key().name(), b.key().value(), |
| b.date_created(), b.date_last_used()); |
| } |
| |
| AutofillEntry MakeAutofillEntry(const char* name, |
| const char* value, |
| time_t date_created, |
| time_t date_last_used) { |
| if (date_last_used < 0) |
| date_last_used = date_created; |
| return AutofillEntry(AutofillKey(ASCIIToUTF16(name), ASCIIToUTF16(value)), |
| Time::FromTimeT(date_created), |
| Time::FromTimeT(date_last_used)); |
| } |
| |
| // Checks |actual| and |expected| contain the same elements. |
| void CompareAutofillEntrySets(const AutofillEntrySet& actual, |
| const AutofillEntrySet& expected) { |
| ASSERT_EQ(expected.size(), actual.size()); |
| size_t count = 0; |
| for (AutofillEntrySet::const_iterator it = actual.begin(); |
| it != actual.end(); ++it) { |
| count += expected.count(*it); |
| } |
| EXPECT_EQ(actual.size(), count); |
| } |
| |
| int GetAutofillEntryCount(const base::string16& name, |
| const base::string16& value, |
| WebDatabase* db) { |
| sql::Statement s(db->GetSQLConnection()->GetUniqueStatement( |
| "SELECT count FROM autofill WHERE name = ? AND value = ?")); |
| s.BindString16(0, name); |
| s.BindString16(1, value); |
| s.Step(); |
| return s.ColumnInt(0); |
| } |
| |
| } // namespace |
| |
| class AutofillTableTest : public testing::Test { |
| public: |
| AutofillTableTest() {} |
| ~AutofillTableTest() override {} |
| |
| protected: |
| void SetUp() override { |
| OSCryptMocker::SetUpWithSingleton(); |
| ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| file_ = temp_dir_.path().AppendASCII("TestWebDatabase"); |
| |
| table_.reset(new AutofillTable); |
| db_.reset(new WebDatabase); |
| db_->AddTable(table_.get()); |
| ASSERT_EQ(sql::INIT_OK, db_->Init(file_)); |
| } |
| |
| void TearDown() override { OSCryptMocker::TearDown(); } |
| |
| base::FilePath file_; |
| base::ScopedTempDir temp_dir_; |
| std::unique_ptr<AutofillTable> table_; |
| std::unique_ptr<WebDatabase> db_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(AutofillTableTest); |
| }; |
| |
| TEST_F(AutofillTableTest, Autofill) { |
| Time t1 = Time::Now(); |
| |
| // Simulate the submission of a handful of entries in a field called "Name", |
| // some more often than others. |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| base::Time now = base::Time::Now(); |
| base::TimeDelta two_seconds = base::TimeDelta::FromSeconds(2); |
| EXPECT_TRUE(table_->AddFormFieldValue(field, &changes)); |
| std::vector<base::string16> v; |
| for (int i = 0; i < 5; ++i) { |
| field.value = ASCIIToUTF16("Clark Kent"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| now + i * two_seconds)); |
| } |
| for (int i = 0; i < 3; ++i) { |
| field.value = ASCIIToUTF16("Clark Sutter"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| now + i * two_seconds)); |
| } |
| for (int i = 0; i < 2; ++i) { |
| field.name = ASCIIToUTF16("Favorite Color"); |
| field.value = ASCIIToUTF16("Green"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| now + i * two_seconds)); |
| } |
| |
| // We have added the name Clark Kent 5 times, so count should be 5. |
| EXPECT_EQ(5, GetAutofillEntryCount(ASCIIToUTF16("Name"), |
| ASCIIToUTF16("Clark Kent"), db_.get())); |
| |
| // Storing in the data base should be case sensitive, so there should be no |
| // database entry for clark kent lowercase. |
| EXPECT_EQ(0, GetAutofillEntryCount(ASCIIToUTF16("Name"), |
| ASCIIToUTF16("clark kent"), db_.get())); |
| |
| EXPECT_EQ(2, GetAutofillEntryCount(ASCIIToUTF16("Favorite Color"), |
| ASCIIToUTF16("Green"), db_.get())); |
| |
| // 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(table_->GetFormValuesForElementName( |
| ASCIIToUTF16("Name"), base::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(table_->GetFormValuesForElementName( |
| ASCIIToUTF16("Name"), base::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(table_->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(table_->RemoveFormElementsAddedBetween(t1, Time(), &changes)); |
| |
| const AutofillChange kExpectedChanges[] = { |
| 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(kExpectedChanges), changes.size()); |
| for (size_t i = 0; i < arraysize(kExpectedChanges); ++i) { |
| EXPECT_EQ(kExpectedChanges[i], changes[i]); |
| } |
| |
| EXPECT_EQ(0, GetAutofillEntryCount(ASCIIToUTF16("Name"), |
| ASCIIToUTF16("Clark Kent"), db_.get())); |
| |
| EXPECT_TRUE(table_->GetFormValuesForElementName( |
| ASCIIToUTF16("Name"), base::string16(), &v, 6)); |
| EXPECT_EQ(0U, v.size()); |
| |
| // Now add some values with empty strings. |
| const base::string16 kValue = ASCIIToUTF16(" toto "); |
| field.name = ASCIIToUTF16("blank"); |
| field.value = base::string16(); |
| EXPECT_TRUE(table_->AddFormFieldValue(field, &changes)); |
| field.name = ASCIIToUTF16("blank"); |
| field.value = ASCIIToUTF16(" "); |
| EXPECT_TRUE(table_->AddFormFieldValue(field, &changes)); |
| field.name = ASCIIToUTF16("blank"); |
| field.value = ASCIIToUTF16(" "); |
| EXPECT_TRUE(table_->AddFormFieldValue(field, &changes)); |
| field.name = ASCIIToUTF16("blank"); |
| field.value = kValue; |
| EXPECT_TRUE(table_->AddFormFieldValue(field, &changes)); |
| |
| // They should be stored normally as the DB layer does not check for empty |
| // values. |
| v.clear(); |
| EXPECT_TRUE(table_->GetFormValuesForElementName( |
| ASCIIToUTF16("blank"), base::string16(), &v, 10)); |
| EXPECT_EQ(4U, v.size()); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_GetCountOfValuesContainedBetween) { |
| AutofillChangeList changes; |
| // This test makes time comparisons that are precise to a microsecond, but the |
| // database uses the time_t format which is only precise to a second. |
| // Make sure we use timestamps rounded to a second. |
| Time begin = Time::FromTimeT(Time::Now().ToTimeT()); |
| Time now = begin; |
| TimeDelta second = TimeDelta::FromSeconds(1); |
| |
| struct Entry { |
| const char* name; |
| const char* value; |
| } entries[] = { |
| { "Alter ego", "Superman" }, |
| { "Name", "Superman" }, |
| { "Name", "Clark Kent" }, |
| { "Name", "Superman" }, |
| { "Name", "Clark Sutter" }, |
| { "Nomen", "Clark Kent" } |
| }; |
| |
| for (Entry entry : entries) { |
| FormFieldData field; |
| field.name = ASCIIToUTF16(entry.name); |
| field.value = ASCIIToUTF16(entry.value); |
| ASSERT_TRUE(table_->AddFormFieldValueTime(field, &changes, now)); |
| now += second; |
| } |
| |
| // While the entry "Alter ego" : "Superman" is entirely contained within |
| // the first second, the value "Superman" itself appears in another entry, |
| // so it is not contained. |
| EXPECT_EQ(0, table_->GetCountOfValuesContainedBetween( |
| begin, begin + second)); |
| |
| // No values are entirely contained within the first three seconds either |
| // (note that the second time constraint is exclusive). |
| EXPECT_EQ(0, table_->GetCountOfValuesContainedBetween( |
| begin, begin + 3 * second)); |
| |
| // Only "Superman" is entirely contained within the first four seconds. |
| EXPECT_EQ(1, table_->GetCountOfValuesContainedBetween( |
| begin, begin + 4 * second)); |
| |
| // "Clark Kent" and "Clark Sutter" are contained between the first |
| // and seventh second. |
| EXPECT_EQ(2, table_->GetCountOfValuesContainedBetween( |
| begin + second, begin + 7 * second)); |
| |
| // Beginning from the third second, "Clark Kent" is not contained. |
| EXPECT_EQ(1, table_->GetCountOfValuesContainedBetween( |
| begin + 3 * second, begin + 7 * second)); |
| |
| // We have three distinct values total. |
| EXPECT_EQ(3, table_->GetCountOfValuesContainedBetween( |
| begin, begin + 7 * second)); |
| |
| // And we should get the same result for unlimited time interval. |
| EXPECT_EQ(3, table_->GetCountOfValuesContainedBetween(Time(), Time::Max())); |
| |
| // The null time interval is also interpreted as unlimited. |
| EXPECT_EQ(3, table_->GetCountOfValuesContainedBetween(Time(), Time())); |
| |
| // An interval that does not fully contain any entries returns zero. |
| EXPECT_EQ(0, table_->GetCountOfValuesContainedBetween( |
| begin + second, begin + 2 * second)); |
| |
| // So does an interval which has no intersection with any entry. |
| EXPECT_EQ(0, table_->GetCountOfValuesContainedBetween(Time(), begin)); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_RemoveBetweenChanges) { |
| TimeDelta one_day(TimeDelta::FromDays(1)); |
| Time t1 = Time::Now(); |
| Time t2 = t1 + one_day; |
| |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t1)); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t2)); |
| |
| changes.clear(); |
| EXPECT_TRUE(table_->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( |
| table_->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) { |
| TimeDelta one_day(TimeDelta::FromDays(1)); |
| Time t1 = Time::Now(); |
| Time t2 = t1 + one_day; |
| |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t1)); |
| ASSERT_EQ(1U, changes.size()); |
| EXPECT_EQ(AutofillChange(AutofillChange::ADD, |
| AutofillKey(ASCIIToUTF16("Name"), |
| ASCIIToUTF16("Superman"))), |
| changes[0]); |
| |
| changes.clear(); |
| EXPECT_TRUE( |
| table_->AddFormFieldValueTime(field, &changes, t2)); |
| ASSERT_EQ(1U, changes.size()); |
| EXPECT_EQ(AutofillChange(AutofillChange::UPDATE, |
| AutofillKey(ASCIIToUTF16("Name"), |
| ASCIIToUTF16("Superman"))), |
| changes[0]); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_UpdateOneWithOneTimestamp) { |
| AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, -1)); |
| std::vector<AutofillEntry> entries; |
| entries.push_back(entry); |
| ASSERT_TRUE(table_->UpdateAutofillEntries(entries)); |
| |
| EXPECT_EQ(1, GetAutofillEntryCount(ASCIIToUTF16("foo"), ASCIIToUTF16("bar"), |
| db_.get())); |
| |
| std::vector<AutofillEntry> all_entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries)); |
| ASSERT_EQ(1U, all_entries.size()); |
| EXPECT_EQ(entry, all_entries[0]); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_UpdateOneWithTwoTimestamps) { |
| AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, 2)); |
| std::vector<AutofillEntry> entries; |
| entries.push_back(entry); |
| ASSERT_TRUE(table_->UpdateAutofillEntries(entries)); |
| |
| EXPECT_EQ(2, GetAutofillEntryCount(ASCIIToUTF16("foo"), ASCIIToUTF16("bar"), |
| db_.get())); |
| |
| std::vector<AutofillEntry> all_entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries)); |
| ASSERT_EQ(1U, all_entries.size()); |
| EXPECT_EQ(entry, all_entries[0]); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_GetAutofillTimestamps) { |
| AutofillEntry entry(MakeAutofillEntry("foo", "bar", 1, 2)); |
| std::vector<AutofillEntry> entries; |
| entries.push_back(entry); |
| ASSERT_TRUE(table_->UpdateAutofillEntries(entries)); |
| |
| Time date_created, date_last_used; |
| ASSERT_TRUE(table_->GetAutofillTimestamps(ASCIIToUTF16("foo"), |
| ASCIIToUTF16("bar"), |
| &date_created, |
| &date_last_used)); |
| EXPECT_EQ(Time::FromTimeT(1), date_created); |
| EXPECT_EQ(Time::FromTimeT(2), date_last_used); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_UpdateTwo) { |
| 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(table_->UpdateAutofillEntries(entries)); |
| |
| EXPECT_EQ(1, GetAutofillEntryCount(ASCIIToUTF16("foo"), ASCIIToUTF16("bar0"), |
| db_.get())); |
| EXPECT_EQ(2, GetAutofillEntryCount(ASCIIToUTF16("foo"), ASCIIToUTF16("bar1"), |
| db_.get())); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_UpdateReplace) { |
| AutofillChangeList changes; |
| // Add a form field. This will be replaced. |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValue(field, &changes)); |
| |
| AutofillEntry entry(MakeAutofillEntry("Name", "Superman", 1, 2)); |
| std::vector<AutofillEntry> entries; |
| entries.push_back(entry); |
| ASSERT_TRUE(table_->UpdateAutofillEntries(entries)); |
| |
| std::vector<AutofillEntry> all_entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&all_entries)); |
| ASSERT_EQ(1U, all_entries.size()); |
| EXPECT_EQ(entry, all_entries[0]); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_UpdateDontReplace) { |
| Time t = Time::Now(); |
| AutofillEntry existing( |
| MakeAutofillEntry("Name", "Superman", t.ToTimeT(), -1)); |
| |
| AutofillChangeList changes; |
| // Add a form field. This will NOT be replaced. |
| FormFieldData field; |
| field.name = existing.key().name(); |
| field.value = existing.key().value(); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, t)); |
| AutofillEntry entry(MakeAutofillEntry("Name", "Clark Kent", 1, 2)); |
| std::vector<AutofillEntry> entries; |
| entries.push_back(entry); |
| ASSERT_TRUE(table_->UpdateAutofillEntries(entries)); |
| |
| std::vector<AutofillEntry> all_entries; |
| ASSERT_TRUE(table_->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) { |
| 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<FormFieldData> elements; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("firstname"); |
| field.value = ASCIIToUTF16("Joe"); |
| elements.push_back(field); |
| |
| field.name = ASCIIToUTF16("firstname"); |
| field.value = ASCIIToUTF16("Jane"); |
| elements.push_back(field); |
| |
| field.name = ASCIIToUTF16("lastname"); |
| field.value = ASCIIToUTF16("Smith"); |
| elements.push_back(field); |
| |
| field.name = ASCIIToUTF16("lastname"); |
| field.value = ASCIIToUTF16("Jones"); |
| elements.push_back(field); |
| |
| std::vector<AutofillChange> changes; |
| table_->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(table_->GetAllAutofillEntries(&all_entries)); |
| ASSERT_EQ(2U, all_entries.size()); |
| } |
| |
| TEST_F(AutofillTableTest, |
| Autofill_RemoveFormElementsAddedBetween_UsedOnlyBefore) { |
| // Add an entry used only before the targetted range. |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(10))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(20))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(30))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(40))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(50))); |
| |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| |
| changes.clear(); |
| EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(base::Time::FromTimeT(51), |
| base::Time::FromTimeT(60), |
| &changes)); |
| EXPECT_TRUE(changes.empty()); |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| } |
| |
| TEST_F(AutofillTableTest, |
| Autofill_RemoveFormElementsAddedBetween_UsedOnlyAfter) { |
| // Add an entry used only after the targetted range. |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(50))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(60))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(70))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(80))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(90))); |
| |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| |
| changes.clear(); |
| EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(base::Time::FromTimeT(40), |
| base::Time::FromTimeT(50), |
| &changes)); |
| EXPECT_TRUE(changes.empty()); |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| } |
| |
| TEST_F(AutofillTableTest, |
| Autofill_RemoveFormElementsAddedBetween_UsedOnlyDuring) { |
| // Add an entry used entirely during the targetted range. |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(10))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(20))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(30))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(40))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(50))); |
| |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| |
| changes.clear(); |
| EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(base::Time::FromTimeT(10), |
| base::Time::FromTimeT(51), |
| &changes)); |
| ASSERT_EQ(1U, changes.size()); |
| EXPECT_EQ(AutofillChange(AutofillChange::REMOVE, |
| AutofillKey(field.name, field.value)), |
| changes[0]); |
| EXPECT_EQ(0, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| } |
| |
| TEST_F(AutofillTableTest, |
| Autofill_RemoveFormElementsAddedBetween_UsedBeforeAndDuring) { |
| // Add an entry used both before and during the targetted range. |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(10))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(20))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(30))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(40))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(50))); |
| |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| |
| changes.clear(); |
| EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(base::Time::FromTimeT(40), |
| base::Time::FromTimeT(60), |
| &changes)); |
| ASSERT_EQ(1U, changes.size()); |
| EXPECT_EQ(AutofillChange(AutofillChange::UPDATE, |
| AutofillKey(field.name, field.value)), |
| changes[0]); |
| EXPECT_EQ(4, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| base::Time date_created, date_last_used; |
| EXPECT_TRUE( |
| table_->GetAutofillTimestamps(field.name, field.value, |
| &date_created, &date_last_used)); |
| EXPECT_EQ(base::Time::FromTimeT(10), date_created); |
| EXPECT_EQ(base::Time::FromTimeT(39), date_last_used); |
| } |
| |
| TEST_F(AutofillTableTest, |
| Autofill_RemoveFormElementsAddedBetween_UsedDuringAndAfter) { |
| // Add an entry used both during and after the targetted range. |
| AutofillChangeList changes; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(50))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(60))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(70))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(80))); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| base::Time::FromTimeT(90))); |
| |
| EXPECT_EQ(5, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| |
| changes.clear(); |
| EXPECT_TRUE(table_->RemoveFormElementsAddedBetween(base::Time::FromTimeT(40), |
| base::Time::FromTimeT(80), |
| &changes)); |
| ASSERT_EQ(1U, changes.size()); |
| EXPECT_EQ(AutofillChange(AutofillChange::UPDATE, |
| AutofillKey(field.name, field.value)), |
| changes[0]); |
| EXPECT_EQ(2, GetAutofillEntryCount(field.name, field.value, db_.get())); |
| base::Time date_created, date_last_used; |
| EXPECT_TRUE( |
| table_->GetAutofillTimestamps(field.name, field.value, |
| &date_created, &date_last_used)); |
| EXPECT_EQ(base::Time::FromTimeT(80), date_created); |
| EXPECT_EQ(base::Time::FromTimeT(90), date_last_used); |
| } |
| |
| TEST_F(AutofillTableTest, AutofillProfile) { |
| // Add a 'Home' profile. |
| AutofillProfile home_profile; |
| home_profile.set_origin(std::string()); |
| home_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John")); |
| home_profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q.")); |
| home_profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith")); |
| home_profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz")); |
| home_profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Google")); |
| home_profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way")); |
| home_profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5")); |
| home_profile.SetRawInfo(ADDRESS_HOME_DEPENDENT_LOCALITY, |
| ASCIIToUTF16("Beverly Hills")); |
| home_profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles")); |
| home_profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA")); |
| home_profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025")); |
| home_profile.SetRawInfo(ADDRESS_HOME_SORTING_CODE, ASCIIToUTF16("MAGIC ###")); |
| home_profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US")); |
| home_profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567")); |
| home_profile.set_language_code("en"); |
| |
| Time pre_creation_time = Time::Now(); |
| EXPECT_TRUE(table_->AddAutofillProfile(home_profile)); |
| Time post_creation_time = Time::Now(); |
| |
| // Get the 'Home' profile. |
| std::unique_ptr<AutofillProfile> db_profile = |
| table_->GetAutofillProfile(home_profile.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Add a 'Billing' profile. |
| AutofillProfile billing_profile = home_profile; |
| billing_profile.set_guid(base::GenerateGUID()); |
| billing_profile.set_origin("https://www.example.com/"); |
| billing_profile.SetRawInfo(ADDRESS_HOME_LINE1, |
| ASCIIToUTF16("5678 Bottom Street")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("suite 3")); |
| |
| pre_creation_time = Time::Now(); |
| EXPECT_TRUE(table_->AddAutofillProfile(billing_profile)); |
| post_creation_time = Time::Now(); |
| |
| // Get the 'Billing' profile. |
| db_profile = table_->GetAutofillProfile(billing_profile.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Update the 'Billing' profile, name only. |
| billing_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Jane")); |
| Time pre_modification_time = Time::Now(); |
| EXPECT_TRUE(table_->UpdateAutofillProfile(billing_profile)); |
| Time post_modification_time = Time::Now(); |
| db_profile = table_->GetAutofillProfile(billing_profile.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Update the 'Billing' profile. |
| billing_profile.set_origin(kSettingsOrigin); |
| billing_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Janice")); |
| billing_profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("C.")); |
| billing_profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Joplin")); |
| billing_profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("jane@singer.com")); |
| billing_profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Indy")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("Open Road")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("Route 66")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_DEPENDENT_LOCALITY, |
| ASCIIToUTF16("District 9")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("NFA")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("NY")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("10011")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_SORTING_CODE, ASCIIToUTF16("123456")); |
| billing_profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US")); |
| billing_profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, |
| ASCIIToUTF16("18181230000")); |
| Time pre_modification_time_2 = Time::Now(); |
| EXPECT_TRUE(table_->UpdateAutofillProfile(billing_profile)); |
| Time post_modification_time_2 = Time::Now(); |
| db_profile = table_->GetAutofillProfile(billing_profile.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Remove the 'Billing' profile. |
| EXPECT_TRUE(table_->RemoveAutofillProfile(billing_profile.guid())); |
| db_profile = table_->GetAutofillProfile(billing_profile.guid()); |
| EXPECT_FALSE(db_profile); |
| } |
| |
| TEST_F(AutofillTableTest, AutofillProfileTrash) { |
| std::vector<std::string> guids; |
| table_->GetAutofillProfilesInTrash(&guids); |
| EXPECT_TRUE(guids.empty()); |
| |
| ASSERT_TRUE(table_->AddAutofillGUIDToTrash( |
| "00000000-0000-0000-0000-000000000000")); |
| ASSERT_TRUE(table_->AddAutofillGUIDToTrash( |
| "00000000-0000-0000-0000-000000000001")); |
| ASSERT_TRUE(table_->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(table_->EmptyAutofillProfilesTrash()); |
| ASSERT_TRUE(table_->GetAutofillProfilesInTrash(&guids)); |
| EXPECT_TRUE(guids.empty()); |
| } |
| |
| TEST_F(AutofillTableTest, AutofillProfileTrashInteraction) { |
| std::vector<std::string> guids; |
| table_->GetAutofillProfilesInTrash(&guids); |
| EXPECT_TRUE(guids.empty()); |
| |
| AutofillProfile profile; |
| profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John")); |
| profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q.")); |
| profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith")); |
| profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz")); |
| profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1 Main St")); |
| profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles")); |
| profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA")); |
| profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025")); |
| profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US")); |
| |
| // Mark this profile as in the trash. This stops |AddAutofillProfile| from |
| // adding it. |
| EXPECT_TRUE(table_->AddAutofillGUIDToTrash(profile.guid())); |
| EXPECT_TRUE(table_->AddAutofillProfile(profile)); |
| std::unique_ptr<AutofillProfile> added_profile = |
| table_->GetAutofillProfile(profile.guid()); |
| EXPECT_FALSE(added_profile); |
| |
| // Add the profile for real this time. |
| EXPECT_TRUE(table_->EmptyAutofillProfilesTrash()); |
| EXPECT_TRUE(table_->GetAutofillProfilesInTrash(&guids)); |
| EXPECT_TRUE(guids.empty()); |
| EXPECT_TRUE(table_->AddAutofillProfile(profile)); |
| added_profile = table_->GetAutofillProfile(profile.guid()); |
| EXPECT_TRUE(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(table_->AddAutofillGUIDToTrash(profile.guid())); |
| profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("Jane")); |
| EXPECT_TRUE(table_->UpdateAutofillProfile(profile)); |
| std::unique_ptr<AutofillProfile> updated_profile = |
| table_->GetAutofillProfile(profile.guid()); |
| EXPECT_TRUE(updated_profile); |
| EXPECT_EQ(ASCIIToUTF16("John"), updated_profile->GetRawInfo(NAME_FIRST)); |
| |
| // 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(table_->RemoveAutofillProfile(profile.guid())); |
| std::unique_ptr<AutofillProfile> removed_profile = |
| table_->GetAutofillProfile(profile.guid()); |
| EXPECT_TRUE(removed_profile); |
| EXPECT_FALSE(table_->IsAutofillGUIDInTrash(profile.guid())); |
| |
| // Check that emptying the trash now allows removal to occur. |
| EXPECT_TRUE(table_->EmptyAutofillProfilesTrash()); |
| EXPECT_TRUE(table_->RemoveAutofillProfile(profile.guid())); |
| removed_profile = table_->GetAutofillProfile(profile.guid()); |
| EXPECT_FALSE(removed_profile); |
| } |
| |
| TEST_F(AutofillTableTest, CreditCard) { |
| // Add a 'Work' credit card. |
| CreditCard work_creditcard; |
| work_creditcard.set_origin("https://www.example.com/"); |
| work_creditcard.SetRawInfo(CREDIT_CARD_NAME_FULL, |
| ASCIIToUTF16("Jack Torrance")); |
| work_creditcard.SetRawInfo(CREDIT_CARD_NUMBER, |
| ASCIIToUTF16("1234567890123456")); |
| work_creditcard.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04")); |
| work_creditcard.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, |
| ASCIIToUTF16("2013")); |
| |
| Time pre_creation_time = Time::Now(); |
| EXPECT_TRUE(table_->AddCreditCard(work_creditcard)); |
| Time post_creation_time = Time::Now(); |
| |
| // Get the 'Work' credit card. |
| std::unique_ptr<CreditCard> db_creditcard = |
| table_->GetCreditCard(work_creditcard.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Add a 'Target' credit card. |
| CreditCard target_creditcard; |
| target_creditcard.set_origin(std::string()); |
| target_creditcard.SetRawInfo(CREDIT_CARD_NAME_FULL, |
| ASCIIToUTF16("Jack Torrance")); |
| target_creditcard.SetRawInfo(CREDIT_CARD_NUMBER, |
| ASCIIToUTF16("1111222233334444")); |
| target_creditcard.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("06")); |
| target_creditcard.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, |
| ASCIIToUTF16("2012")); |
| |
| pre_creation_time = Time::Now(); |
| EXPECT_TRUE(table_->AddCreditCard(target_creditcard)); |
| post_creation_time = Time::Now(); |
| db_creditcard = table_->GetCreditCard(target_creditcard.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Update the 'Target' credit card. |
| target_creditcard.set_origin("Interactive Autofill dialog"); |
| target_creditcard.SetRawInfo(CREDIT_CARD_NAME_FULL, |
| ASCIIToUTF16("Charles Grady")); |
| Time pre_modification_time = Time::Now(); |
| EXPECT_TRUE(table_->UpdateCreditCard(target_creditcard)); |
| Time post_modification_time = Time::Now(); |
| db_creditcard = table_->GetCreditCard(target_creditcard.guid()); |
| ASSERT_TRUE(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.is_valid()); |
| 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()); |
| |
| // Remove the 'Target' credit card. |
| EXPECT_TRUE(table_->RemoveCreditCard(target_creditcard.guid())); |
| db_creditcard = table_->GetCreditCard(target_creditcard.guid()); |
| EXPECT_FALSE(db_creditcard); |
| } |
| |
| TEST_F(AutofillTableTest, UpdateAutofillProfile) { |
| // Add a profile to the db. |
| AutofillProfile profile; |
| profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John")); |
| profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q.")); |
| profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith")); |
| profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@example.com")); |
| profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Google")); |
| profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way")); |
| profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5")); |
| profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles")); |
| profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA")); |
| profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025")); |
| profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US")); |
| profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567")); |
| profile.set_language_code("en"); |
| table_->AddAutofillProfile(profile); |
| |
| // Set a mocked value for the profile's creation time. |
| const time_t kMockCreationDate = 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.is_valid()); |
| s_mock_creation_date.BindInt64(0, kMockCreationDate); |
| ASSERT_TRUE(s_mock_creation_date.Run()); |
| |
| // Get the profile. |
| std::unique_ptr<AutofillProfile> db_profile = |
| table_->GetAutofillProfile(profile.guid()); |
| ASSERT_TRUE(db_profile); |
| EXPECT_EQ(profile, *db_profile); |
| sql::Statement s_original(db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified FROM autofill_profiles")); |
| ASSERT_TRUE(s_original.is_valid()); |
| ASSERT_TRUE(s_original.Step()); |
| EXPECT_EQ(kMockCreationDate, 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.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@smith.xyz")); |
| table_->UpdateAutofillProfile(profile); |
| |
| // Get the profile. |
| db_profile = table_->GetAutofillProfile(profile.guid()); |
| ASSERT_TRUE(db_profile); |
| EXPECT_EQ(profile, *db_profile); |
| sql::Statement s_updated(db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified FROM autofill_profiles")); |
| ASSERT_TRUE(s_updated.is_valid()); |
| ASSERT_TRUE(s_updated.Step()); |
| EXPECT_LT(kMockCreationDate, 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.is_valid()); |
| s_mock_modification_date.BindInt64(0, mock_modification_date); |
| ASSERT_TRUE(s_mock_modification_date.Run()); |
| |
| // Finally, call into |UpdateAutofillProfile()| without changing the |
| // profile. The modification date should not change. |
| table_->UpdateAutofillProfile(profile); |
| |
| // Get the profile. |
| db_profile = table_->GetAutofillProfile(profile.guid()); |
| ASSERT_TRUE(db_profile); |
| EXPECT_EQ(profile, *db_profile); |
| sql::Statement s_unchanged(db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified FROM autofill_profiles")); |
| ASSERT_TRUE(s_unchanged.is_valid()); |
| ASSERT_TRUE(s_unchanged.Step()); |
| EXPECT_EQ(mock_modification_date, s_unchanged.ColumnInt64(0)); |
| EXPECT_FALSE(s_unchanged.Step()); |
| } |
| |
| TEST_F(AutofillTableTest, UpdateCreditCard) { |
| // Add a credit card to the db. |
| CreditCard credit_card; |
| credit_card.SetRawInfo(CREDIT_CARD_NAME_FULL, ASCIIToUTF16("Jack Torrance")); |
| credit_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1234567890123456")); |
| credit_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04")); |
| credit_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2013")); |
| table_->AddCreditCard(credit_card); |
| |
| // Set a mocked value for the credit card's creation time. |
| const time_t kMockCreationDate = 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.is_valid()); |
| s_mock_creation_date.BindInt64(0, kMockCreationDate); |
| ASSERT_TRUE(s_mock_creation_date.Run()); |
| |
| // Get the credit card. |
| std::unique_ptr<CreditCard> db_credit_card = |
| table_->GetCreditCard(credit_card.guid()); |
| ASSERT_TRUE(db_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.is_valid()); |
| ASSERT_TRUE(s_original.Step()); |
| EXPECT_EQ(kMockCreationDate, 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.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("01")); |
| table_->UpdateCreditCard(credit_card); |
| |
| // Get the credit card. |
| db_credit_card = table_->GetCreditCard(credit_card.guid()); |
| ASSERT_TRUE(db_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.is_valid()); |
| ASSERT_TRUE(s_updated.Step()); |
| EXPECT_LT(kMockCreationDate, 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.is_valid()); |
| 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. |
| table_->UpdateCreditCard(credit_card); |
| |
| // Get the credit card. |
| db_credit_card = table_->GetCreditCard(credit_card.guid()); |
| ASSERT_TRUE(db_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.is_valid()); |
| ASSERT_TRUE(s_unchanged.Step()); |
| EXPECT_EQ(mock_modification_date, s_unchanged.ColumnInt64(0)); |
| EXPECT_FALSE(s_unchanged.Step()); |
| } |
| |
| TEST_F(AutofillTableTest, UpdateProfileOriginOnly) { |
| // Add a profile to the db. |
| AutofillProfile profile; |
| profile.SetRawInfo(NAME_FIRST, ASCIIToUTF16("John")); |
| profile.SetRawInfo(NAME_MIDDLE, ASCIIToUTF16("Q.")); |
| profile.SetRawInfo(NAME_LAST, ASCIIToUTF16("Smith")); |
| profile.SetRawInfo(EMAIL_ADDRESS, ASCIIToUTF16("js@example.com")); |
| profile.SetRawInfo(COMPANY_NAME, ASCIIToUTF16("Google")); |
| profile.SetRawInfo(ADDRESS_HOME_LINE1, ASCIIToUTF16("1234 Apple Way")); |
| profile.SetRawInfo(ADDRESS_HOME_LINE2, ASCIIToUTF16("unit 5")); |
| profile.SetRawInfo(ADDRESS_HOME_CITY, ASCIIToUTF16("Los Angeles")); |
| profile.SetRawInfo(ADDRESS_HOME_STATE, ASCIIToUTF16("CA")); |
| profile.SetRawInfo(ADDRESS_HOME_ZIP, ASCIIToUTF16("90025")); |
| profile.SetRawInfo(ADDRESS_HOME_COUNTRY, ASCIIToUTF16("US")); |
| profile.SetRawInfo(PHONE_HOME_WHOLE_NUMBER, ASCIIToUTF16("18181234567")); |
| table_->AddAutofillProfile(profile); |
| |
| // Set a mocked value for the profile's creation time. |
| const time_t kMockCreationDate = 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.is_valid()); |
| s_mock_creation_date.BindInt64(0, kMockCreationDate); |
| ASSERT_TRUE(s_mock_creation_date.Run()); |
| |
| // Get the profile. |
| std::unique_ptr<AutofillProfile> db_profile = |
| table_->GetAutofillProfile(profile.guid()); |
| ASSERT_TRUE(db_profile); |
| EXPECT_EQ(profile, *db_profile); |
| sql::Statement s_original(db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified FROM autofill_profiles")); |
| ASSERT_TRUE(s_original.is_valid()); |
| ASSERT_TRUE(s_original.Step()); |
| EXPECT_EQ(kMockCreationDate, s_original.ColumnInt64(0)); |
| EXPECT_FALSE(s_original.Step()); |
| |
| // Now, update just the profile's origin and save the update to the database. |
| // The modification date should change to reflect the update. |
| profile.set_origin("https://www.example.com/"); |
| table_->UpdateAutofillProfile(profile); |
| |
| // Get the profile. |
| db_profile = table_->GetAutofillProfile(profile.guid()); |
| ASSERT_TRUE(db_profile); |
| EXPECT_EQ(profile, *db_profile); |
| sql::Statement s_updated(db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified FROM autofill_profiles")); |
| ASSERT_TRUE(s_updated.is_valid()); |
| ASSERT_TRUE(s_updated.Step()); |
| EXPECT_LT(kMockCreationDate, s_updated.ColumnInt64(0)); |
| EXPECT_FALSE(s_updated.Step()); |
| } |
| |
| TEST_F(AutofillTableTest, UpdateCreditCardOriginOnly) { |
| // Add a credit card to the db. |
| CreditCard credit_card; |
| credit_card.SetRawInfo(CREDIT_CARD_NAME_FULL, ASCIIToUTF16("Jack Torrance")); |
| credit_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1234567890123456")); |
| credit_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("04")); |
| credit_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2013")); |
| table_->AddCreditCard(credit_card); |
| |
| // Set a mocked value for the credit card's creation time. |
| const time_t kMockCreationDate = 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.is_valid()); |
| s_mock_creation_date.BindInt64(0, kMockCreationDate); |
| ASSERT_TRUE(s_mock_creation_date.Run()); |
| |
| // Get the credit card. |
| std::unique_ptr<CreditCard> db_credit_card = |
| table_->GetCreditCard(credit_card.guid()); |
| ASSERT_TRUE(db_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.is_valid()); |
| ASSERT_TRUE(s_original.Step()); |
| EXPECT_EQ(kMockCreationDate, s_original.ColumnInt64(0)); |
| EXPECT_FALSE(s_original.Step()); |
| |
| // Now, update just the credit card's origin and save the update to the |
| // database. The modification date should change to reflect the update. |
| credit_card.set_origin("https://www.example.com/"); |
| table_->UpdateCreditCard(credit_card); |
| |
| // Get the credit card. |
| db_credit_card = table_->GetCreditCard(credit_card.guid()); |
| ASSERT_TRUE(db_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.is_valid()); |
| ASSERT_TRUE(s_updated.Step()); |
| EXPECT_LT(kMockCreationDate, s_updated.ColumnInt64(0)); |
| EXPECT_FALSE(s_updated.Step()); |
| } |
| |
| TEST_F(AutofillTableTest, RemoveAutofillDataModifiedBetween) { |
| // 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; |
| table_->RemoveAutofillDataModifiedBetween( |
| 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.is_valid()); |
| 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.is_valid()); |
| 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). |
| table_->RemoveAutofillDataModifiedBetween( |
| 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.is_valid()); |
| 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.is_valid()); |
| 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. |
| table_->RemoveAutofillDataModifiedBetween( |
| 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.is_valid()); |
| 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.is_valid()); |
| EXPECT_FALSE(s_credit_cards_empty.Step()); |
| } |
| |
| TEST_F(AutofillTableTest, RemoveOriginURLsModifiedBetween) { |
| // Populate the autofill_profiles and credit_cards tables. |
| ASSERT_TRUE(db_->GetSQLConnection()->Execute( |
| "INSERT INTO autofill_profiles (guid, origin, date_modified) " |
| "VALUES('00000000-0000-0000-0000-000000000000', '', 11);" |
| "INSERT INTO autofill_profiles (guid, origin, date_modified) " |
| "VALUES('00000000-0000-0000-0000-000000000001', " |
| " 'https://www.example.com/', 21);" |
| "INSERT INTO autofill_profiles (guid, origin, date_modified) " |
| "VALUES('00000000-0000-0000-0000-000000000002', 'Chrome settings', 31);" |
| "INSERT INTO credit_cards (guid, origin, date_modified) " |
| "VALUES('00000000-0000-0000-0000-000000000003', '', 17);" |
| "INSERT INTO credit_cards (guid, origin, date_modified) " |
| "VALUES('00000000-0000-0000-0000-000000000004', " |
| " 'https://www.example.com/', 27);" |
| "INSERT INTO credit_cards (guid, origin, date_modified) " |
| "VALUES('00000000-0000-0000-0000-000000000005', 'Chrome settings', " |
| " 37);")); |
| |
| // Remove all origin URLs set in the bounded time range [21,27). |
| ScopedVector<AutofillProfile> profiles; |
| table_->RemoveOriginURLsModifiedBetween( |
| Time::FromTimeT(21), Time::FromTimeT(27), &profiles); |
| ASSERT_EQ(1UL, profiles.size()); |
| EXPECT_EQ("00000000-0000-0000-0000-000000000001", profiles[0]->guid()); |
| sql::Statement s_autofill_profiles_bounded( |
| db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified, origin FROM autofill_profiles")); |
| ASSERT_TRUE(s_autofill_profiles_bounded.is_valid()); |
| ASSERT_TRUE(s_autofill_profiles_bounded.Step()); |
| EXPECT_EQ(11, s_autofill_profiles_bounded.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_autofill_profiles_bounded.ColumnString(1)); |
| ASSERT_TRUE(s_autofill_profiles_bounded.Step()); |
| EXPECT_EQ(21, s_autofill_profiles_bounded.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_autofill_profiles_bounded.ColumnString(1)); |
| ASSERT_TRUE(s_autofill_profiles_bounded.Step()); |
| EXPECT_EQ(31, s_autofill_profiles_bounded.ColumnInt64(0)); |
| EXPECT_EQ(kSettingsOrigin, s_autofill_profiles_bounded.ColumnString(1)); |
| sql::Statement s_credit_cards_bounded( |
| db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified, origin FROM credit_cards")); |
| ASSERT_TRUE(s_credit_cards_bounded.is_valid()); |
| ASSERT_TRUE(s_credit_cards_bounded.Step()); |
| EXPECT_EQ(17, s_credit_cards_bounded.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_credit_cards_bounded.ColumnString(1)); |
| ASSERT_TRUE(s_credit_cards_bounded.Step()); |
| EXPECT_EQ(27, s_credit_cards_bounded.ColumnInt64(0)); |
| EXPECT_EQ("https://www.example.com/", |
| s_credit_cards_bounded.ColumnString(1)); |
| ASSERT_TRUE(s_credit_cards_bounded.Step()); |
| EXPECT_EQ(37, s_credit_cards_bounded.ColumnInt64(0)); |
| EXPECT_EQ(kSettingsOrigin, s_credit_cards_bounded.ColumnString(1)); |
| |
| // Remove all origin URLS. |
| profiles.clear(); |
| table_->RemoveOriginURLsModifiedBetween(Time(), Time(), &profiles); |
| EXPECT_EQ(0UL, profiles.size()); |
| sql::Statement s_autofill_profiles_all( |
| db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified, origin FROM autofill_profiles")); |
| ASSERT_TRUE(s_autofill_profiles_all.is_valid()); |
| ASSERT_TRUE(s_autofill_profiles_all.Step()); |
| EXPECT_EQ(11, s_autofill_profiles_all.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_autofill_profiles_all.ColumnString(1)); |
| ASSERT_TRUE(s_autofill_profiles_all.Step()); |
| EXPECT_EQ(21, s_autofill_profiles_all.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_autofill_profiles_all.ColumnString(1)); |
| ASSERT_TRUE(s_autofill_profiles_all.Step()); |
| EXPECT_EQ(31, s_autofill_profiles_all.ColumnInt64(0)); |
| EXPECT_EQ(kSettingsOrigin, s_autofill_profiles_all.ColumnString(1)); |
| sql::Statement s_credit_cards_all( |
| db_->GetSQLConnection()->GetUniqueStatement( |
| "SELECT date_modified, origin FROM credit_cards")); |
| ASSERT_TRUE(s_credit_cards_all.is_valid()); |
| ASSERT_TRUE(s_credit_cards_all.Step()); |
| EXPECT_EQ(17, s_credit_cards_all.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_credit_cards_all.ColumnString(1)); |
| ASSERT_TRUE(s_credit_cards_all.Step()); |
| EXPECT_EQ(27, s_credit_cards_all.ColumnInt64(0)); |
| EXPECT_EQ(std::string(), s_credit_cards_all.ColumnString(1)); |
| ASSERT_TRUE(s_credit_cards_all.Step()); |
| EXPECT_EQ(37, s_credit_cards_all.ColumnInt64(0)); |
| EXPECT_EQ(kSettingsOrigin, s_credit_cards_all.ColumnString(1)); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_NoResults) { |
| std::vector<AutofillEntry> entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&entries)); |
| |
| EXPECT_EQ(0U, entries.size()); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_OneResult) { |
| AutofillChangeList changes; |
| std::map<std::string, std::vector<Time> > name_value_times_map; |
| |
| time_t start = 0; |
| std::vector<Time> timestamps1; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &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.front(), timestamps1.back()); |
| |
| expected_entries.insert(ae1); |
| |
| std::vector<AutofillEntry> entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&entries)); |
| AutofillEntrySet entry_set(entries.begin(), entries.end(), |
| CompareAutofillEntries); |
| |
| CompareAutofillEntrySets(entry_set, expected_entries); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_TwoDistinct) { |
| AutofillChangeList changes; |
| std::map<std::string, std::vector<Time> > name_value_times_map; |
| time_t start = 0; |
| |
| std::vector<Time> timestamps1; |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &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; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Clark Kent"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &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.front(), timestamps1.back()); |
| AutofillEntry ae2(ak2, timestamps2.front(), timestamps2.back()); |
| |
| expected_entries.insert(ae1); |
| expected_entries.insert(ae2); |
| |
| std::vector<AutofillEntry> entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&entries)); |
| AutofillEntrySet entry_set(entries.begin(), entries.end(), |
| CompareAutofillEntries); |
| |
| CompareAutofillEntrySets(entry_set, expected_entries); |
| } |
| |
| TEST_F(AutofillTableTest, Autofill_GetAllAutofillEntries_TwoSame) { |
| AutofillChangeList changes; |
| std::map<std::string, std::vector<Time> > name_value_times_map; |
| |
| std::vector<Time> timestamps; |
| time_t start = 0; |
| for (int i = 0; i < 2; ++i, ++start) { |
| FormFieldData field; |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16("Superman"); |
| EXPECT_TRUE(table_->AddFormFieldValueTime(field, &changes, |
| Time::FromTimeT(start))); |
| timestamps.push_back(Time::FromTimeT(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.front(), timestamps.back()); |
| |
| expected_entries.insert(ae1); |
| |
| std::vector<AutofillEntry> entries; |
| ASSERT_TRUE(table_->GetAllAutofillEntries(&entries)); |
| AutofillEntrySet entry_set(entries.begin(), entries.end(), |
| CompareAutofillEntries); |
| |
| CompareAutofillEntrySets(entry_set, expected_entries); |
| } |
| |
| TEST_F(AutofillTableTest, SetGetServerCards) { |
| std::vector<CreditCard> inputs; |
| inputs.push_back(CreditCard(CreditCard::FULL_SERVER_CARD, "a123")); |
| inputs[0].SetRawInfo(CREDIT_CARD_NAME_FULL, ASCIIToUTF16("Paul F. Tompkins")); |
| inputs[0].SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("1")); |
| inputs[0].SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2020")); |
| inputs[0].SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("4111111111111111")); |
| |
| inputs.push_back( |
| CreditCard(CreditCard::MASKED_SERVER_CARD, "b456")); |
| inputs[1].SetRawInfo(CREDIT_CARD_NAME_FULL, ASCIIToUTF16("Rick Roman")); |
| inputs[1].SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("12")); |
| inputs[1].SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("1997")); |
| inputs[1].SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1111")); |
| inputs[1].SetTypeForMaskedCard(kVisaCard); |
| inputs[1].SetServerStatus(CreditCard::EXPIRED); |
| |
| test::SetServerCreditCards(table_.get(), inputs); |
| |
| std::vector<CreditCard*> outputs; |
| ASSERT_TRUE(table_->GetServerCreditCards(&outputs)); |
| ASSERT_EQ(inputs.size(), outputs.size()); |
| |
| // Ordering isn't guaranteed, so fix the ordering if it's backwards. |
| if (outputs[1]->server_id() == inputs[0].server_id()) |
| std::swap(outputs[0], outputs[1]); |
| |
| // GUIDs for server cards are dynamically generated so will be different |
| // after reading from the DB. Check they're valid, but otherwise don't count |
| // them in the comparison. |
| inputs[0].set_guid(std::string()); |
| inputs[1].set_guid(std::string()); |
| outputs[0]->set_guid(std::string()); |
| outputs[1]->set_guid(std::string()); |
| |
| EXPECT_EQ(inputs[0], *outputs[0]); |
| EXPECT_EQ(inputs[1], *outputs[1]); |
| |
| EXPECT_EQ(CreditCard::OK, outputs[0]->GetServerStatus()); |
| EXPECT_EQ(CreditCard::EXPIRED, outputs[1]->GetServerStatus()); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| } |
| |
| TEST_F(AutofillTableTest, MaskUnmaskServerCards) { |
| base::string16 masked_number(ASCIIToUTF16("1111")); |
| std::vector<CreditCard> inputs; |
| inputs.push_back(CreditCard(CreditCard::MASKED_SERVER_CARD, "a123")); |
| inputs[0].SetRawInfo(CREDIT_CARD_NAME_FULL, ASCIIToUTF16("Jay Johnson")); |
| inputs[0].SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("1")); |
| inputs[0].SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2020")); |
| inputs[0].SetRawInfo(CREDIT_CARD_NUMBER, masked_number); |
| inputs[0].SetTypeForMaskedCard(kVisaCard); |
| test::SetServerCreditCards(table_.get(), inputs); |
| |
| // Unmask the number. The full number should be available. |
| base::string16 full_number(ASCIIToUTF16("4111111111111111")); |
| ASSERT_TRUE(table_->UnmaskServerCreditCard(inputs[0], |
| full_number)); |
| |
| std::vector<CreditCard*> outputs; |
| table_->GetServerCreditCards(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_TRUE(CreditCard::FULL_SERVER_CARD == outputs[0]->record_type()); |
| EXPECT_EQ(full_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| |
| // Re-mask the number, we should only get the last 4 digits out. |
| ASSERT_TRUE(table_->MaskServerCreditCard(inputs[0].server_id())); |
| table_->GetServerCreditCards(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_TRUE(CreditCard::MASKED_SERVER_CARD == outputs[0]->record_type()); |
| EXPECT_EQ(masked_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| } |
| |
| // Calling SetServerCreditCards should replace all existing cards, but unmasked |
| // cards should not be re-masked. |
| TEST_F(AutofillTableTest, SetServerCardModify) { |
| // Add a masked card. |
| CreditCard masked_card(CreditCard::MASKED_SERVER_CARD, "a123"); |
| masked_card.SetRawInfo(CREDIT_CARD_NAME_FULL, |
| ASCIIToUTF16("Paul F. Tompkins")); |
| masked_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("1")); |
| masked_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2020")); |
| masked_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1111")); |
| masked_card.SetTypeForMaskedCard(kVisaCard); |
| |
| std::vector<CreditCard> inputs; |
| inputs.push_back(masked_card); |
| test::SetServerCreditCards(table_.get(), inputs); |
| |
| // Now unmask it. |
| base::string16 full_number = ASCIIToUTF16("4111111111111111"); |
| table_->UnmaskServerCreditCard(masked_card, full_number); |
| |
| // The card should now be unmasked. |
| std::vector<CreditCard*> outputs; |
| table_->GetServerCreditCards(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_TRUE(outputs[0]->record_type() == CreditCard::FULL_SERVER_CARD); |
| EXPECT_EQ(full_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| |
| // Call set again with the masked number. |
| inputs[0] = masked_card; |
| test::SetServerCreditCards(table_.get(), inputs); |
| |
| // The card should stay unmasked. |
| table_->GetServerCreditCards(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_TRUE(outputs[0]->record_type() == CreditCard::FULL_SERVER_CARD); |
| EXPECT_EQ(full_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| |
| // Set inputs that do not include our old card. |
| CreditCard random_card(CreditCard::MASKED_SERVER_CARD, "b456"); |
| random_card.SetRawInfo(CREDIT_CARD_NAME_FULL, ASCIIToUTF16("Rick Roman")); |
| random_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("12")); |
| random_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("1997")); |
| random_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("2222")); |
| random_card.SetTypeForMaskedCard(kVisaCard); |
| inputs[0] = random_card; |
| test::SetServerCreditCards(table_.get(), inputs); |
| |
| // We should have only the new card, the other one should have been deleted. |
| table_->GetServerCreditCards(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_TRUE(outputs[0]->record_type() == CreditCard::MASKED_SERVER_CARD); |
| EXPECT_EQ(random_card.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(ASCIIToUTF16("2222"), outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| |
| // Putting back the original card masked should make it masked (this tests |
| // that the unmasked data was really deleted). |
| inputs[0] = masked_card; |
| test::SetServerCreditCards(table_.get(), inputs); |
| table_->GetServerCreditCards(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_TRUE(outputs[0]->record_type() == CreditCard::MASKED_SERVER_CARD); |
| EXPECT_EQ(masked_card.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(ASCIIToUTF16("1111"), outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| } |
| |
| TEST_F(AutofillTableTest, SetServerCardUpdateUsageStats) { |
| // Add a masked card. |
| CreditCard masked_card(CreditCard::MASKED_SERVER_CARD, "a123"); |
| masked_card.SetRawInfo(CREDIT_CARD_NAME_FULL, |
| ASCIIToUTF16("Paul F. Tompkins")); |
| masked_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("1")); |
| masked_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2020")); |
| masked_card.SetRawInfo(CREDIT_CARD_NUMBER, ASCIIToUTF16("1111")); |
| masked_card.SetTypeForMaskedCard(kVisaCard); |
| |
| std::vector<CreditCard> inputs; |
| inputs.push_back(masked_card); |
| test::SetServerCreditCards(table_.get(), inputs); |
| |
| ScopedVector<CreditCard> outputs; |
| table_->GetServerCreditCards(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(masked_card.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(1U, outputs[0]->use_count()); |
| EXPECT_NE(base::Time(), outputs[0]->use_date()); |
| // We don't track modification date for server cards. It should always be |
| // base::Time(). |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| |
| // Update the usage stats; make sure they're reflected in GetServerProfiles. |
| inputs.back().set_use_count(4U); |
| inputs.back().set_use_date(base::Time()); |
| table_->UpdateServerCardUsageStats(inputs.back()); |
| table_->GetServerCreditCards(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(masked_card.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(4U, outputs[0]->use_count()); |
| EXPECT_EQ(base::Time(), outputs[0]->use_date()); |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| |
| // Setting the cards again shouldn't delete the usage stats. |
| table_->SetServerCreditCards(inputs); |
| table_->GetServerCreditCards(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(masked_card.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(4U, outputs[0]->use_count()); |
| EXPECT_EQ(base::Time(), outputs[0]->use_date()); |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| |
| // Set a card list where the card is missing --- this should clear metadata. |
| CreditCard masked_card2(CreditCard::MASKED_SERVER_CARD, "b456"); |
| inputs.back() = masked_card2; |
| table_->SetServerCreditCards(inputs); |
| |
| // Back to the original card list. |
| inputs.back() = masked_card; |
| table_->SetServerCreditCards(inputs); |
| table_->GetServerCreditCards(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(masked_card.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(1U, outputs[0]->use_count()); |
| EXPECT_NE(base::Time(), outputs[0]->use_date()); |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| } |
| |
| TEST_F(AutofillTableTest, SetServerProfile) { |
| AutofillProfile one(AutofillProfile::SERVER_PROFILE, "a123"); |
| std::vector<AutofillProfile> inputs; |
| inputs.push_back(one); |
| table_->SetServerProfiles(inputs); |
| |
| std::vector<AutofillProfile*> outputs; |
| table_->GetServerProfiles(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(one.server_id(), outputs[0]->server_id()); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| |
| // Set a different profile. |
| AutofillProfile two(AutofillProfile::SERVER_PROFILE, "b456"); |
| inputs[0] = two; |
| table_->SetServerProfiles(inputs); |
| |
| // The original one should have been replaced. |
| table_->GetServerProfiles(&outputs); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(two.server_id(), outputs[0]->server_id()); |
| |
| STLDeleteContainerPointers(outputs.begin(), outputs.end()); |
| outputs.clear(); |
| } |
| |
| TEST_F(AutofillTableTest, SetServerProfileUpdateUsageStats) { |
| AutofillProfile one(AutofillProfile::SERVER_PROFILE, "a123"); |
| std::vector<AutofillProfile> inputs; |
| inputs.push_back(one); |
| table_->SetServerProfiles(inputs); |
| |
| ScopedVector<AutofillProfile> outputs; |
| table_->GetServerProfiles(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(one.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(0U, outputs[0]->use_count()); |
| EXPECT_EQ(base::Time(), outputs[0]->use_date()); |
| // We don't track modification date for server profiles. It should always be |
| // base::Time(). |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| |
| // Update the usage stats; make sure they're reflected in GetServerProfiles. |
| inputs.back().set_use_count(4U); |
| inputs.back().set_use_date(base::Time::Now()); |
| table_->UpdateServerAddressUsageStats(inputs.back()); |
| table_->GetServerProfiles(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(one.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(4U, outputs[0]->use_count()); |
| EXPECT_NE(base::Time(), outputs[0]->use_date()); |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| |
| // Setting the profiles again shouldn't delete the usage stats. |
| table_->SetServerProfiles(inputs); |
| table_->GetServerProfiles(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(one.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(4U, outputs[0]->use_count()); |
| EXPECT_NE(base::Time(), outputs[0]->use_date()); |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| |
| // Set a null profile list --- this should clear metadata. |
| table_->SetServerProfiles(std::vector<AutofillProfile>()); |
| // Reset the old profile list and see the metadata is reset. |
| table_->SetServerProfiles(inputs); |
| table_->GetServerProfiles(&outputs.get()); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(one.server_id(), outputs[0]->server_id()); |
| EXPECT_EQ(0U, outputs[0]->use_count()); |
| EXPECT_EQ(base::Time(), outputs[0]->use_date()); |
| EXPECT_EQ(base::Time(), outputs[0]->modification_date()); |
| outputs.clear(); |
| } |
| |
| // Tests that deleting time ranges re-masks server credit cards that were |
| // unmasked in that time. |
| TEST_F(AutofillTableTest, DeleteUnmaskedCard) { |
| // This isn't the exact unmasked time, since the database will use the |
| // current time that it is called. The code below has to be approximate. |
| base::Time unmasked_time = base::Time::Now(); |
| |
| // Add a masked card. |
| base::string16 masked_number = ASCIIToUTF16("1111"); |
| CreditCard masked_card(CreditCard::MASKED_SERVER_CARD, "a123"); |
| masked_card.SetRawInfo(CREDIT_CARD_NAME_FULL, |
| ASCIIToUTF16("Paul F. Tompkins")); |
| masked_card.SetRawInfo(CREDIT_CARD_EXP_MONTH, ASCIIToUTF16("1")); |
| masked_card.SetRawInfo(CREDIT_CARD_EXP_4_DIGIT_YEAR, ASCIIToUTF16("2020")); |
| masked_card.SetRawInfo(CREDIT_CARD_NUMBER, masked_number); |
| masked_card.SetTypeForMaskedCard(kVisaCard); |
| |
| std::vector<CreditCard> inputs; |
| inputs.push_back(masked_card); |
| table_->SetServerCreditCards(inputs); |
| |
| // Unmask it. |
| base::string16 full_number = ASCIIToUTF16("4111111111111111"); |
| table_->UnmaskServerCreditCard(masked_card, full_number); |
| |
| // Delete data in a range a year in the future. |
| std::vector<std::string> profile_guids; |
| std::vector<std::string> credit_card_guids; |
| ASSERT_TRUE(table_->RemoveAutofillDataModifiedBetween( |
| unmasked_time + base::TimeDelta::FromDays(365), |
| unmasked_time + base::TimeDelta::FromDays(530), |
| &profile_guids, &credit_card_guids)); |
| |
| // This should not affect the unmasked card (should be unmasked). |
| ScopedVector<CreditCard> outputs; |
| ASSERT_TRUE(table_->GetServerCreditCards(&outputs.get())); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(CreditCard::FULL_SERVER_CARD, outputs[0]->record_type()); |
| EXPECT_EQ(full_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| outputs.clear(); |
| |
| // Delete data in the range of the last 24 hours. |
| // Fudge |now| to make sure it's strictly greater than the |now| that |
| // the database uses. |
| base::Time now = base::Time::Now() + base::TimeDelta::FromSeconds(1); |
| ASSERT_TRUE(table_->RemoveAutofillDataModifiedBetween( |
| now - base::TimeDelta::FromDays(1), now, |
| &profile_guids, &credit_card_guids)); |
| |
| // This should re-mask. |
| ASSERT_TRUE(table_->GetServerCreditCards(&outputs.get())); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(CreditCard::MASKED_SERVER_CARD, outputs[0]->record_type()); |
| EXPECT_EQ(masked_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| outputs.clear(); |
| |
| // Unmask again, the card should be back. |
| table_->UnmaskServerCreditCard(masked_card, full_number); |
| ASSERT_TRUE(table_->GetServerCreditCards(&outputs.get())); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(CreditCard::FULL_SERVER_CARD, outputs[0]->record_type()); |
| EXPECT_EQ(full_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| outputs.clear(); |
| |
| // Delete all data. |
| ASSERT_TRUE(table_->RemoveAutofillDataModifiedBetween( |
| base::Time(), base::Time::Max(), &profile_guids, &credit_card_guids)); |
| |
| // Should be masked again. |
| ASSERT_TRUE(table_->GetServerCreditCards(&outputs.get())); |
| ASSERT_EQ(1u, outputs.size()); |
| EXPECT_EQ(CreditCard::MASKED_SERVER_CARD, outputs[0]->record_type()); |
| EXPECT_EQ(masked_number, outputs[0]->GetRawInfo(CREDIT_CARD_NUMBER)); |
| outputs.clear(); |
| } |
| |
| TEST_F(AutofillTableTest, GetFormValuesForElementName_SubstringMatchEnabled) { |
| // Token matching is currently behind a flag. |
| base::CommandLine::ForCurrentProcess()->AppendSwitch( |
| switches::kEnableSuggestionsWithSubstringMatch); |
| |
| const size_t kMaxCount = 2; |
| const struct { |
| const char* const field_suggestion[kMaxCount]; |
| const char* const field_contents; |
| size_t expected_suggestion_count; |
| const char* const expected_suggestion[kMaxCount]; |
| } kTestCases[] = { |
| {{"user.test", "test_user"}, "TEST", 2, {"test_user", "user.test"}}, |
| {{"user test", "test-user"}, "user", 2, {"user test", "test-user"}}, |
| {{"user test", "test-rest"}, "user", 1, {"user test", nullptr}}, |
| {{"user@test", "test_user"}, "user@t", 1, {"user@test", nullptr}}, |
| {{"user.test", "test_user"}, "er.tes", 0, {nullptr, nullptr}}, |
| {{"user test", "test_user"}, "_ser", 0, {nullptr, nullptr}}, |
| {{"user.test", "test_user"}, "%ser", 0, {nullptr, nullptr}}, |
| {{"user.test", "test_user"}, |
| "; DROP TABLE autofill;", |
| 0, |
| {nullptr, nullptr}}, |
| }; |
| |
| for (size_t i = 0; i < arraysize(kTestCases); ++i) { |
| SCOPED_TRACE(testing::Message() |
| << "suggestion = " << kTestCases[i].field_suggestion[0] |
| << ", contents = " << kTestCases[i].field_contents); |
| |
| Time t1 = Time::Now(); |
| |
| // Simulate the submission of a handful of entries in a field called "Name". |
| AutofillChangeList changes; |
| FormFieldData field; |
| for (size_t k = 0; k < kMaxCount; ++k) { |
| field.name = ASCIIToUTF16("Name"); |
| field.value = ASCIIToUTF16(kTestCases[i].field_suggestion[k]); |
| table_->AddFormFieldValue(field, &changes); |
| } |
| |
| std::vector<base::string16> v; |
| table_->GetFormValuesForElementName( |
| ASCIIToUTF16("Name"), ASCIIToUTF16(kTestCases[i].field_contents), &v, |
| 6); |
| |
| EXPECT_EQ(kTestCases[i].expected_suggestion_count, v.size()); |
| for (size_t j = 0; j < kTestCases[i].expected_suggestion_count; ++j) { |
| EXPECT_EQ(ASCIIToUTF16(kTestCases[i].expected_suggestion[j]), v[j]); |
| } |
| |
| changes.clear(); |
| table_->RemoveFormElementsAddedBetween(t1, Time(), &changes); |
| } |
| } |
| |
| } // namespace autofill |