Add use_count and use_date to unmasked server cards as well.

components/webdata/common/web_database_migration_unittest.cc

 // Copyright (c) 2012 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 <string>
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/guid.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 215 matching lines @@
 
   static int VersionFromConnection(sql::Connection* connection) {
     // Get version.
     sql::Statement s(connection->GetUniqueStatement(
         "SELECT value FROM meta WHERE key='version'"));
     if (!s.Step())
       return 0;
     return s.ColumnInt(0);
   }
 
-  // The sql files located in "chrome/test/data/web_database" were generated by
-  // launching the Chromium application prior to schema change, then using the
-  // sqlite3 command-line application to dump the contents of the "Web Data"
+  // The sql files located in "components/test/data/web_database" were generated
+  // by launching the Chromium application prior to schema change, then using
+  // the sqlite3 command-line application to dump the contents of the "Web Data"
   // database.
   // Like this:
   //   > .output version_nn.sql
   //   > .dump
   void LoadDatabase(const base::FilePath::StringType& file);
 
  private:
   base::ScopedTempDir temp_dir_;
 
   DISALLOW_COPY_AND_ASSIGN(WebDatabaseMigrationTest);
 };
 
-const int WebDatabaseMigrationTest::kCurrentTestedVersionNumber = 61;
+const int WebDatabaseMigrationTest::kCurrentTestedVersionNumber = 62;
 
 void WebDatabaseMigrationTest::LoadDatabase(
     const base::FilePath::StringType& file) {
   std::string contents;
   ASSERT_TRUE(GetWebDatabaseData(base::FilePath(file), &contents));
 
   sql::Connection connection;
   ASSERT_TRUE(connection.Open(GetDatabasePath()));
   ASSERT_TRUE(connection.Execute(contents.data()));
 }
@@ skipping 2541 matching lines @@
     ASSERT_TRUE(sql::MetaTable::DoesTableExist(&connection));
 
     // Check version.
     EXPECT_EQ(kCurrentTestedVersionNumber, VersionFromConnection(&connection));
 
     ASSERT_TRUE(connection.DoesTableExist("masked_credit_cards"));
     ASSERT_TRUE(connection.DoesTableExist("unmasked_credit_cards"));
     ASSERT_TRUE(connection.DoesTableExist("server_addresses"));
   }
 }
+
+// Tests addition of use_count and use_date fields to autofill profiles and
+// credit cards.
+TEST_F(WebDatabaseMigrationTest, MigrateVersion60ToCurrent) {
+  ASSERT_NO_FATAL_FAILURE(LoadDatabase(FILE_PATH_LITERAL("version_60.sql")));
+
+  // Verify pre-conditions.
+  {
+    sql::Connection connection;
+    ASSERT_TRUE(connection.Open(GetDatabasePath()));
+    ASSERT_TRUE(sql::MetaTable::DoesTableExist(&connection));
+
+    sql::MetaTable meta_table;
+    ASSERT_TRUE(meta_table.Init(&connection, 60, 60));
+
+    EXPECT_FALSE(connection.DoesColumnExist("credit_cards", "use_count"));
+    EXPECT_FALSE(connection.DoesColumnExist("credit_cards", "use_date"));
+    EXPECT_FALSE(connection.DoesColumnExist("autofill_profiles", "use_count"));
+    EXPECT_FALSE(connection.DoesColumnExist("autofill_profiles", "use_date"));
+  }
+
+  DoMigration();
+
+  // Verify post-conditions.
+  {
+    sql::Connection connection;
+    ASSERT_TRUE(connection.Open(GetDatabasePath()));
+    ASSERT_TRUE(sql::MetaTable::DoesTableExist(&connection));
+
+    // Check version.
+    EXPECT_EQ(kCurrentTestedVersionNumber, VersionFromConnection(&connection));
+
+    EXPECT_TRUE(connection.DoesColumnExist("credit_cards", "use_count"));
+    EXPECT_TRUE(connection.DoesColumnExist("credit_cards", "use_date"));
+    EXPECT_TRUE(connection.DoesColumnExist("autofill_profiles", "use_count"));
+    EXPECT_TRUE(connection.DoesColumnExist("autofill_profiles", "use_date"));
+  }
+}
+
+// Tests addition of use_count and use_date fields to unmasked server cards.
+TEST_F(WebDatabaseMigrationTest, MigrateVersion61ToCurrent) {
+  ASSERT_NO_FATAL_FAILURE(LoadDatabase(FILE_PATH_LITERAL("version_61.sql")));
+
+  // Verify pre-conditions.
+  {
+    sql::Connection connection;
+    ASSERT_TRUE(connection.Open(GetDatabasePath()));
+    ASSERT_TRUE(sql::MetaTable::DoesTableExist(&connection));
+
+    sql::MetaTable meta_table;
+    ASSERT_TRUE(meta_table.Init(&connection, 61, 61));
+
+    EXPECT_FALSE(connection.DoesColumnExist("unmasked_credit_cards",
+                                            "use_count"));
+    EXPECT_FALSE(connection.DoesColumnExist("unmasked_credit_cards",
+                                            "use_date"));
+  }
+
+  DoMigration();
+
+  // Verify post-conditions.
+  {
+    sql::Connection connection;
+    ASSERT_TRUE(connection.Open(GetDatabasePath()));
+    ASSERT_TRUE(sql::MetaTable::DoesTableExist(&connection));
+
+    // Check version.
+    EXPECT_EQ(kCurrentTestedVersionNumber, VersionFromConnection(&connection));
+
+    EXPECT_TRUE(connection.DoesColumnExist("unmasked_credit_cards",
+                                           "use_count"));
+    EXPECT_TRUE(connection.DoesColumnExist("unmasked_credit_cards",
+                                           "use_date"));
+  }
+}