ServiceWorker: Migrate the script cache backend from BlockFile to Simple

content/browser/service_worker/service_worker_database_unittest.cc

 // Copyright 2014 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 "content/browser/service_worker/service_worker_database.h"
 
 #include <string>
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
@@ skipping 119 matching lines @@
   data.resources_total_size_bytes = 10;
   ASSERT_EQ(ServiceWorkerDatabase::STATUS_OK,
             database->WriteRegistration(
                 data, resources, &deleted_version, &newly_purgeable_resources));
 
   EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK,
             database->ReadDatabaseVersion(&db_version));
   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());
   scoped_ptr<ServiceWorkerDatabase> database(
       CreateDatabase(database_dir.path()));
 
   GURL origin("http://example.com");
 
   // Add a registration to the database.
   std::vector<ServiceWorkerDatabase::ResourceRecord> resources;
@@ skipping 1422 matching lines @@
       CreateDatabase(database_dir.path()));
 
   EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK, database->LazyOpen(true));
   ASSERT_TRUE(base::DirectoryExists(database_dir.path()));
 
   EXPECT_EQ(ServiceWorkerDatabase::STATUS_OK, database->DestroyDatabase());
   ASSERT_FALSE(base::DirectoryExists(database_dir.path()));
 }
 
 }  // namespace content