ServiceWorker: Migrate the script cache backend from BlockFile to Simple

content/browser/service_worker/service_worker_database_unittest.cc

Index: content/browser/service_worker/service_worker_database_unittest.cc
diff --git a/content/browser/service_worker/service_worker_database_unittest.cc b/content/browser/service_worker/service_worker_database_unittest.cc
index e38930366318f4323ce62b3a8b05033d9101d652..1d19f2237b81bdcf396a4630e9c711f8efd83dd0 100644
--- a/content/browser/service_worker/service_worker_database_unittest.cc
+++ b/content/browser/service_worker/service_worker_database_unittest.cc
@@ -137,6 +137,75 @@ TEST(ServiceWorkerDatabaseTest, DatabaseVersion) {
   EXPECT_LT(0, db_version);
 }
 
+TEST(ServiceWorkerDatabaseTest, DiskCacheMigrationState) {
+  GURL origin("http://example.com");
+  base::ScopedTempDir database_dir;
+  ASSERT_TRUE(database_dir.CreateUniqueTempDir());
+  scoped_ptr<ServiceWorkerDatabase> database(
+      CreateDatabase(database_dir.path()));
+
+  // An empty database should return false.
+  bool migration_needed = false;
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsDiskCacheMigrationNeeded(&migration_needed));
+  EXPECT_FALSE(migration_needed);
+  bool deletion_needed = false;
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsOldDiskCacheDeletionNeeded(&deletion_needed));
+  EXPECT_FALSE(deletion_needed);
+
+  // Simulate an existing database created before diskcache migration.
+  database->set_skip_writing_diskcache_migration_state_on_init_for_testing();
+  std::vector<ServiceWorkerDatabase::ResourceRecord> resources;
+  ServiceWorkerDatabase::RegistrationData deleted_version;
+  std::vector<int64> newly_purgeable_resources;
+  ServiceWorkerDatabase::RegistrationData data;
+  data.registration_id = 100;
+  data.scope = URL(origin, "/foo");
+  data.script = URL(origin, "/script.js");
+  data.version_id = 200;
+  data.resources_total_size_bytes = 300;
+  resources.push_back(CreateResource(1, data.script, 300));
+  ASSERT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->WriteRegistration(data, resources, &deleted_version,
+                                        &newly_purgeable_resources));
+  migration_needed = false;
+  ASSERT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsDiskCacheMigrationNeeded(&migration_needed));
+  ASSERT_TRUE(migration_needed);
+  deletion_needed = false;
+  ASSERT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsOldDiskCacheDeletionNeeded(&deletion_needed));
+  ASSERT_TRUE(deletion_needed);
+
+  // Opening the existing database should not update the migration states.
+  database.reset(CreateDatabase(database_dir.path()));
+  migration_needed = false;
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsDiskCacheMigrationNeeded(&migration_needed));
+  EXPECT_TRUE(migration_needed);
+  deletion_needed = false;
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsOldDiskCacheDeletionNeeded(&deletion_needed));
+  EXPECT_TRUE(deletion_needed);
+
+  // Test SetDiskCacheMigrationNotNeeded().
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->SetDiskCacheMigrationNotNeeded());
+  migration_needed = false;
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsDiskCacheMigrationNeeded(&migration_needed));
+  EXPECT_FALSE(migration_needed);
+
+  // Test SetOldDiskCacheDeletionNotNeeded().
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->SetOldDiskCacheDeletionNotNeeded());
+  deletion_needed = false;
+  EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
+            database->IsOldDiskCacheDeletionNeeded(&deletion_needed));
+  EXPECT_FALSE(deletion_needed);
+}
+
 TEST(ServiceWorkerDatabaseTest, UpgradeSchemaToVersion2) {
   base::ScopedTempDir database_dir;
   ASSERT_TRUE(database_dir.CreateUniqueTempDir());