Create syncable metadata table for Wallet credit cards and addresses.

components/autofill/core/browser/webdata/autofill_table_unittest.cc

 // 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 <utility>
 #include <vector>
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/guid.h"
@@ skipping 1647 matching lines @@
   inputs.push_back(CreditCard(CreditCard::MASKED_SERVER_CARD, "a123"));
   inputs[0].SetRawInfo(CREDIT_CARD_NAME, 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].server_id(),
+  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();
@@ skipping 19 matching lines @@
   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.server_id(), full_number);
+  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();
@@ skipping 38 matching lines @@
   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, 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(0U, outputs[0]->use_count());
+  EXPECT_EQ(base::Time(), outputs[0]->use_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_->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_NE(base::Time(), outputs[0]->use_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_NE(base::Time(), outputs[0]->use_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(0U, outputs[0]->use_count());
+  EXPECT_EQ(base::Time(), outputs[0]->use_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());
+  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());
+  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());
+  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());
+  outputs.clear();
+}
+
 // Tests that deleting time ranges re-masks server credit cards that were
 // unmasked in that time.
 // TODO(brettw) fix flakiness and re-enable: crbug.com/465882
 TEST_F(AutofillTableTest, DISABLED_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, 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.server_id(), full_number);
+  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).
@@ skipping 11 matching lines @@
       &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.server_id(), full_number);
+  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();
 }
 
 }  // namespace autofill