Index: apps/saved_files_service_unittest.cc |
diff --git a/apps/saved_files_service_unittest.cc b/apps/saved_files_service_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..2b3def752e712d7c76714b84b055ec5fbaaefb19 |
--- /dev/null |
+++ b/apps/saved_files_service_unittest.cc |
@@ -0,0 +1,202 @@ |
+// 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 <algorithm> |
+ |
+#include "apps/saved_files_service.h" |
+#include "base/files/file_path.h" |
+#include "base/string_number_conversions.h" |
+#include "base/test/values_test_util.h" |
+#include "base/values.h" |
+#include "chrome/browser/extensions/extension_prefs.h" |
+#include "chrome/browser/extensions/extension_service.h" |
+#include "chrome/browser/extensions/extension_system.h" |
+#include "chrome/browser/extensions/test_extension_environment.h" |
+#include "chrome/common/extensions/extension.h" |
+#include "chrome/test/base/testing_profile.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+#define TRACE_CALL(expression) \ |
+ do { \ |
+ SCOPED_TRACE(#expression); \ |
+ expression; \ |
+ } while (0) |
+ |
+using apps::SavedFileEntry; |
+using apps::SavedFilesService; |
+ |
+namespace { |
+ |
+std::string GenerateId(int i) { |
+ return base::IntToString(i) + ":filename.ext"; |
+} |
+ |
+} // namespace |
+ |
+class SavedFilesServiceUnitTest : public testing::Test { |
+ protected: |
+ virtual void SetUp() OVERRIDE { |
+ testing::Test::SetUp(); |
+ extension_ = env_.MakeExtension(*base::test::ParseJson( |
+ "{\"app\": {\"background\": {\"scripts\": [\"background.js\"]}}," |
+ "\"permissions\": [{\"fileSystem\": [\"retainFiles\"]}]}")); |
+ service_ = SavedFilesService::Get(env_.profile()); |
+ path_ = base::FilePath(FILE_PATH_LITERAL("filename.ext")); |
+ } |
+ |
+ virtual void TearDown() OVERRIDE { |
+ SavedFilesService::ClearMaxSequenceNumberForTest(); |
+ SavedFilesService::ClearLruSizeForTest(); |
+ testing::Test::TearDown(); |
+ } |
+ |
+ void CheckSavedFileEntry(const std::string& id, int sequence_number) { |
+ SCOPED_TRACE(id); |
+ SavedFileEntry entry; |
+ EXPECT_TRUE(service_->IsRetained(extension_->id(), id)); |
+ ASSERT_TRUE(service_->GetFileEntry(extension_->id(), id, &entry)); |
+ EXPECT_EQ(id, entry.id); |
+ EXPECT_EQ(path_, entry.path); |
+ EXPECT_TRUE(entry.writable); |
+ EXPECT_EQ(sequence_number, entry.sequence_number); |
+ } |
+ |
+ void CheckSavedFileEntries(int start, int end) { |
+ SavedFileEntry entry; |
+ for (int i = start; i < end; i++) { |
+ CheckSavedFileEntry(GenerateId(i), i + 1); |
+ } |
+ } |
+ extensions::TestExtensionEnvironment env_; |
+ const extensions::Extension* extension_; |
+ SavedFilesService* service_; |
+ base::FilePath path_; |
+}; |
+ |
+TEST_F(SavedFilesServiceUnitTest, RetainTwoFilesTest) { |
+ service_->RetainFileEntry(extension_->id(), GenerateId(1), path_, true); |
+ service_->RetainFileEntry(extension_->id(), GenerateId(2), path_, true); |
+ service_->RetainFileEntry(extension_->id(), GenerateId(3), path_, true); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 0)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 0)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(3), 0)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 0)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 0)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 2)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 2)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 3)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(3), 0)); |
+ EXPECT_FALSE(service_->IsRetained(extension_->id(), "another id")); |
+ SavedFileEntry entry; |
+ EXPECT_FALSE(service_->GetFileEntry(extension_->id(), "another id", &entry)); |
+ std::vector<SavedFileEntry> entries; |
+ service_->ClearRetainedFiles(extension_); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 3)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 2)); |
+ EXPECT_FALSE(service_->IsRetained(extension_->id(), GenerateId(3))); |
+} |
+ |
+TEST_F(SavedFilesServiceUnitTest, NoRetainFilesPermissionTest) { |
+ extension_ = env_.MakeExtension(*base::test::ParseJson( |
+ "{\"app\": {\"background\": {\"scripts\": [\"background.js\"]}}," |
+ "\"permissions\": [\"fileSystem\"]}")); |
+ service_->RetainFileEntry(extension_->id(), GenerateId(1), path_, true); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 0)); |
+ SavedFileEntry entry; |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 1)); |
+ EXPECT_FALSE(service_->IsRetained(extension_->id(), "another id")); |
+ EXPECT_FALSE(service_->GetFileEntry(extension_->id(), "another id", &entry)); |
+ service_->ClearRetainedFiles(extension_); |
+ std::vector<SavedFileEntry> entries = |
+ service_->GetAllFileEntries(extension_->id()); |
+ EXPECT_TRUE(entries.empty()); |
+} |
+ |
+TEST_F(SavedFilesServiceUnitTest, EvictionTest) { |
+ SavedFilesService::SetLruSizeForTest(10); |
+ for (int i = 0; i < 10; i++) { |
+ service_->RetainFileEntry(extension_->id(), GenerateId(i), path_, true); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(i)); |
+ } |
+ service_->RetainFileEntry(extension_->id(), GenerateId(10), path_, true); |
+ TRACE_CALL(CheckSavedFileEntries(0, 10)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(10), 0)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(10)); |
+ |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(0), 0)); |
+ TRACE_CALL(CheckSavedFileEntries(1, 11)); |
+ service_->ClearRetainedFiles(extension_); |
+ SavedFileEntry entry; |
+ EXPECT_FALSE(service_->GetFileEntry(extension_->id(), GenerateId(0), &entry)); |
+ TRACE_CALL(CheckSavedFileEntries(1, 11)); |
+ |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 12)); |
+ TRACE_CALL(CheckSavedFileEntries(2, 11)); |
+ service_->ClearRetainedFiles(extension_); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 12)); |
+ TRACE_CALL(CheckSavedFileEntries(2, 11)); |
+} |
+ |
+TEST_F(SavedFilesServiceUnitTest, SequenceNumberCompactionTest) { |
+ SavedFilesService::SetMaxSequenceNumberForTest(8); |
+ for (int i = 0; i < 4; i++) { |
+ service_->RetainFileEntry(extension_->id(), GenerateId(i), path_, true); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(i)); |
+ } |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(2)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(3)); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(0), 1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 7)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(3), 6)); |
+ service_->ClearRetainedFiles(extension_); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(0), 1)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 7)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(3), 6)); |
+ |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(3)); |
+ TRACE_CALL(CheckSavedFileEntries(0, 4)); |
+ service_->ClearRetainedFiles(extension_); |
+ TRACE_CALL(CheckSavedFileEntries(0, 4)); |
+} |
+ |
+TEST_F(SavedFilesServiceUnitTest, SequenceNumberCompactionFirstAboveOneTest) { |
+ SavedFilesService::SetMaxSequenceNumberForTest(10); |
+ for (int i = 0; i < 4; i++) { |
+ service_->RetainFileEntry(extension_->id(), GenerateId(i), path_, true); |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(i)); |
+ } |
+ for (int i = 0; i < 4; i++) { |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(i)); |
+ } |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(2)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(0), 5)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 6)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 9)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(3), 8)); |
+ service_->ClearRetainedFiles(extension_); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(0), 5)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(1), 6)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(2), 9)); |
+ TRACE_CALL(CheckSavedFileEntry(GenerateId(3), 8)); |
+ |
+ service_->MoveEntryToFrontOfQueue(extension_->id(), GenerateId(3)); |
+ TRACE_CALL(CheckSavedFileEntries(0, 4)); |
+ service_->ClearRetainedFiles(extension_); |
+ TRACE_CALL(CheckSavedFileEntries(0, 4)); |
+} |