diff options
author | Tao Bao <tbao@google.com> | 2018-07-13 19:16:43 +0200 |
---|---|---|
committer | android-build-merger <android-build-merger@google.com> | 2018-07-13 19:16:43 +0200 |
commit | d8e96f4b4d38b3ef7385d4f4e022e58439a21b47 (patch) | |
tree | 3e6a81eb61f07c1dd70dfed0b3cfbaf38071b17e /tests/unit/applypatch_test.cpp | |
parent | Merge "applypatch: Fix the return type of FreeSpaceForFile()." (diff) | |
parent | Merge "tests: Clean up the temporary dirs post-run." (diff) | |
download | android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.tar android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.tar.gz android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.tar.bz2 android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.tar.lz android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.tar.xz android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.tar.zst android_bootable_recovery-d8e96f4b4d38b3ef7385d4f4e022e58439a21b47.zip |
Diffstat (limited to '')
-rw-r--r-- | tests/unit/applypatch_test.cpp | 38 |
1 files changed, 19 insertions, 19 deletions
diff --git a/tests/unit/applypatch_test.cpp b/tests/unit/applypatch_test.cpp index 4b757feab..0d4123b5e 100644 --- a/tests/unit/applypatch_test.cpp +++ b/tests/unit/applypatch_test.cpp @@ -164,27 +164,37 @@ class FreeCacheTest : public ::testing::Test { return file_list; } - static void AddFilesToDir(const std::string& dir, const std::vector<std::string>& files) { + void AddFilesToDir(const std::string& dir, const std::vector<std::string>& files) { std::string zeros(4096, 0); for (const auto& file : files) { - std::string path = dir + "/" + file; - ASSERT_TRUE(android::base::WriteStringToFile(zeros, path)); + temporary_files_.push_back(dir + "/" + file); + ASSERT_TRUE(android::base::WriteStringToFile(zeros, temporary_files_.back())); } } void SetUp() override { Paths::Get().set_cache_log_directory(mock_log_dir.path); + temporary_files_.clear(); + } + + void TearDown() override { + for (const auto& file : temporary_files_) { + ASSERT_TRUE(android::base::RemoveFileIfExists(file)); + } } // A mock method to calculate the free space. It assumes the partition has a total size of 40960 // bytes and all files are 4096 bytes in size. - int64_t MockFreeSpaceChecker(const std::string& dirname) { + static size_t MockFreeSpaceChecker(const std::string& dirname) { std::vector<std::string> files = FindFilesInDir(dirname); return PARTITION_SIZE - 4096 * files.size(); } TemporaryDir mock_cache; TemporaryDir mock_log_dir; + + private: + std::vector<std::string> temporary_files_; }; TEST_F(FreeCacheTest, FreeCacheSmoke) { @@ -193,9 +203,7 @@ TEST_F(FreeCacheTest, FreeCacheSmoke) { ASSERT_EQ(files, FindFilesInDir(mock_cache.path)); ASSERT_EQ(4096 * 7, MockFreeSpaceChecker(mock_cache.path)); - ASSERT_TRUE(RemoveFilesInDirectory(4096 * 9, mock_cache.path, [this](const std::string& dir) { - return this->MockFreeSpaceChecker(dir); - })); + ASSERT_TRUE(RemoveFilesInDirectory(4096 * 9, mock_cache.path, MockFreeSpaceChecker)); ASSERT_EQ(std::vector<std::string>{ "file3" }, FindFilesInDir(mock_cache.path)); ASSERT_EQ(4096 * 9, MockFreeSpaceChecker(mock_cache.path)); @@ -221,9 +229,7 @@ TEST_F(FreeCacheTest, FreeCacheOpenFile) { android::base::unique_fd fd(open(file1_path.c_str(), O_RDONLY)); // file1 can't be deleted as it's opened by us. - ASSERT_FALSE(RemoveFilesInDirectory(4096 * 10, mock_cache.path, [this](const std::string& dir) { - return this->MockFreeSpaceChecker(dir); - })); + ASSERT_FALSE(RemoveFilesInDirectory(4096 * 10, mock_cache.path, MockFreeSpaceChecker)); ASSERT_EQ(std::vector<std::string>{ "file1" }, FindFilesInDir(mock_cache.path)); } @@ -234,9 +240,7 @@ TEST_F(FreeCacheTest, FreeCacheLogsSmoke) { AddFilesToDir(mock_log_dir.path, log_files); ASSERT_EQ(4096 * 5, MockFreeSpaceChecker(mock_log_dir.path)); - ASSERT_TRUE(RemoveFilesInDirectory(4096 * 8, mock_log_dir.path, [this](const std::string& dir) { - return this->MockFreeSpaceChecker(dir); - })); + ASSERT_TRUE(RemoveFilesInDirectory(4096 * 8, mock_log_dir.path, MockFreeSpaceChecker)); // Logs with a higher index will be deleted first std::vector<std::string> expected = { "last_log", "last_log.1" }; @@ -250,9 +254,7 @@ TEST_F(FreeCacheTest, FreeCacheLogsStringComparison) { AddFilesToDir(mock_log_dir.path, log_files); ASSERT_EQ(4096 * 6, MockFreeSpaceChecker(mock_log_dir.path)); - ASSERT_TRUE(RemoveFilesInDirectory(4096 * 9, mock_log_dir.path, [this](const std::string& dir) { - return this->MockFreeSpaceChecker(dir); - })); + ASSERT_TRUE(RemoveFilesInDirectory(4096 * 9, mock_log_dir.path, MockFreeSpaceChecker)); // Logs with incorrect format will be deleted first; and the last_kmsg with the same index is // deleted before last_log. @@ -267,9 +269,7 @@ TEST_F(FreeCacheTest, FreeCacheLogsOtherFiles) { AddFilesToDir(mock_log_dir.path, log_files); ASSERT_EQ(4096 * 5, MockFreeSpaceChecker(mock_log_dir.path)); - ASSERT_FALSE(RemoveFilesInDirectory(4096 * 8, mock_log_dir.path, [this](const std::string& dir) { - return this->MockFreeSpaceChecker(dir); - })); + ASSERT_FALSE(RemoveFilesInDirectory(4096 * 8, mock_log_dir.path, MockFreeSpaceChecker)); // Non log files in /cache/recovery won't be deleted. std::vector<std::string> expected = { "block.map", "command", "last_install" }; |