chrome/installer/util/logging_installer_unittest.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <string>
   6
   7 #include "base/file_util.h"
   8 #include "base/files/file_path.h"
   9 #include "base/files/scoped_temp_dir.h"
  10 #include "base/platform_file.h"
  11 #include "base/win/scoped_handle.h"
  12 #include "chrome/installer/util/logging_installer.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14
  15 TEST(LoggingInstallerTest, TestTruncate) {
  16   const std::string test_data(installer::kMaxInstallerLogFileSize + 1, 'a');
  17
  18   base::ScopedTempDir temp_dir;
  19   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
  20
  21   base::FilePath temp_file = temp_dir.path().Append(L"temp");
  22   EXPECT_EQ(test_data.size(),
  23             file_util::WriteFile(temp_file, &test_data[0], test_data.size()));
  24   ASSERT_TRUE(base::PathExists(temp_file));
  25
  26   int64 file_size = 0;
  27   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
  28   EXPECT_EQ(test_data.size(), file_size);
  29
  30   EXPECT_EQ(installer::LOGFILE_TRUNCATED,
  31             installer::TruncateLogFileIfNeeded(temp_file));
  32
  33   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
  34   EXPECT_EQ(installer::kTruncatedInstallerLogFileSize , file_size);
  35
  36   // Check that the temporary file was deleted.
  37   EXPECT_FALSE(base::PathExists(temp_file.Append(L".tmp")));
  38 }
  39
  40 TEST(LoggingInstallerTest, TestTruncationNotNeeded) {
  41   const std::string test_data(installer::kMaxInstallerLogFileSize, 'a');
  42
  43   base::ScopedTempDir temp_dir;
  44   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
  45
  46   base::FilePath temp_file = temp_dir.path().Append(L"temp");
  47   EXPECT_EQ(test_data.size(),
  48             file_util::WriteFile(temp_file, &test_data[0], test_data.size()));
  49   ASSERT_TRUE(base::PathExists(temp_file));
  50
  51   int64 file_size = 0;
  52   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
  53   EXPECT_EQ(test_data.size(), file_size);
  54
  55   EXPECT_EQ(installer::LOGFILE_UNTOUCHED,
  56             installer::TruncateLogFileIfNeeded(temp_file));
  57   EXPECT_TRUE(base::PathExists(temp_file));
  58   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
  59   EXPECT_EQ(test_data.size(), file_size);
  60 }
  61
  62 TEST(LoggingInstallerTest, TestInUseNeedsTruncation) {
  63   const std::string test_data(installer::kMaxInstallerLogFileSize + 1, 'a');
  64
  65   base::ScopedTempDir temp_dir;
  66   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
  67
  68   base::FilePath temp_file = temp_dir.path().Append(L"temp");
  69   EXPECT_EQ(test_data.size(),
  70             file_util::WriteFile(temp_file, &test_data[0], test_data.size()));
  71   ASSERT_TRUE(base::PathExists(temp_file));
  72   int64 file_size = 0;
  73   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
  74   EXPECT_EQ(test_data.size(), file_size);
  75
  76   // Prevent the log file from being moved or deleted.
  77   const int file_flags = base::PLATFORM_FILE_OPEN |
  78                          base::PLATFORM_FILE_READ |
  79                          base::PLATFORM_FILE_EXCLUSIVE_READ;
  80   base::win::ScopedHandle temp_platform_file(
  81       base::CreatePlatformFile(temp_file, file_flags, NULL, NULL));
  82   ASSERT_TRUE(temp_platform_file.IsValid());
  83
  84   EXPECT_EQ(installer::LOGFILE_UNTOUCHED,
  85             installer::TruncateLogFileIfNeeded(temp_file));
  86   EXPECT_TRUE(base::PathExists(temp_file));
  87   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
  88   EXPECT_EQ(test_data.size(), file_size);
  89 }
  90
  91 TEST(LoggingInstallerTest, TestMoveFailsNeedsTruncation) {
  92   const std::string test_data(installer::kMaxInstallerLogFileSize + 1, 'a');
  93
  94   base::ScopedTempDir temp_dir;
  95   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
  96
  97   base::FilePath temp_file = temp_dir.path().Append(L"temp");
  98   EXPECT_EQ(test_data.size(),
  99             file_util::WriteFile(temp_file, &test_data[0], test_data.size()));
 100   ASSERT_TRUE(base::PathExists(temp_file));
 101   int64 file_size = 0;
 102   EXPECT_TRUE(base::GetFileSize(temp_file, &file_size));
 103   EXPECT_EQ(test_data.size(), file_size);
 104
 105   // Create an inconvenient, non-deletable file in the location that
 106   // TruncateLogFileIfNeeded would like to move the log file to.
 107   const int file_flags = base::PLATFORM_FILE_CREATE |
 108                          base::PLATFORM_FILE_READ |
 109                          base::PLATFORM_FILE_EXCLUSIVE_READ;
 110   base::FilePath temp_file_move_dest(
 111       temp_file.value() + FILE_PATH_LITERAL(".tmp"));
 112   base::win::ScopedHandle temp_move_destination_file(
 113       base::CreatePlatformFile(temp_file_move_dest, file_flags, NULL, NULL));
 114   ASSERT_TRUE(temp_move_destination_file.IsValid());
 115
 116   EXPECT_EQ(installer::LOGFILE_DELETED,
 117             installer::TruncateLogFileIfNeeded(temp_file));
 118   EXPECT_FALSE(base::PathExists(temp_file));
 119 }