summaryrefslogtreecommitdiffstats
path: root/tests/unit/applypatch_test.cpp
diff options
context:
space:
mode:
authorTao Bao <tbao@google.com>2018-07-12 20:49:05 +0200
committerTao Bao <tbao@google.com>2018-07-12 21:54:15 +0200
commitb1c5b62557a7bb3dbae2c5614f3e9bd053c383a3 (patch)
treee2d83b26a49096c0c7ae1899361fa483caf1a266 /tests/unit/applypatch_test.cpp
parentapplypatch: Fix the return type of FreeSpaceForFile(). (diff)
downloadandroid_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.tar
android_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.tar.gz
android_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.tar.bz2
android_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.tar.lz
android_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.tar.xz
android_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.tar.zst
android_bootable_recovery-b1c5b62557a7bb3dbae2c5614f3e9bd053c383a3.zip
Diffstat (limited to 'tests/unit/applypatch_test.cpp')
-rw-r--r--tests/unit/applypatch_test.cpp38
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" };