chrome/browser/component_updater/cld_component_installer_unittest.cc

   1 // Copyright 2014 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 <stdint.h>
   6 #include <vector>
   7
   8 #include "base/files/file_path.h"
   9 #include "base/files/file_util.h"
  10 #include "base/files/scoped_temp_dir.h"
  11 #include "base/macros.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/strings/string_util.h"
  14 #include "base/strings/utf_string_conversions.h"
  15 #include "base/values.h"
  16 #include "base/version.h"
  17 #include "chrome/browser/component_updater/cld_component_installer.h"
  18 #include "components/translate/content/browser/browser_cld_data_provider.h"
  19 #include "components/translate/content/common/cld_data_source.h"
  20 #include "testing/gtest/include/gtest/gtest.h"
  21 #include "testing/platform_test.h"
  22
  23 using component_updater::CldComponentInstallerTraits;
  24
  25 namespace {
  26
  27 // This has to match what's in cld_component_installer.cc.
  28 const base::FilePath::CharType kTestCldDataFileName[] =
  29     FILE_PATH_LITERAL("cld2_data.bin");
  30
  31 }  // namespace
  32
  33 namespace component_updater {
  34
  35 class CldComponentInstallerTest : public PlatformTest {
  36  public:
  37   CldComponentInstallerTest() {}
  38   void SetUp() override {
  39     PlatformTest::SetUp();
  40     translate::CldDataSource::DisableSanityChecksForTest();
  41
  42     // ScopedTempDir automatically does a recursive delete on the entire
  43     // directory in its destructor, so no cleanup is required in TearDown.
  44     // Note that all files created by this test case are created within the
  45     // directory that is created here, so even though they are not explicitly
  46     // created *as temp files*, they will still get cleaned up automagically.
  47     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
  48
  49     // The "latest CLD data file" is a static piece of information, and thus
  50     // for correctness we empty it before each test.
  51     CldComponentInstallerTraits::SetLatestCldDataFile(base::FilePath());
  52     base::FilePath path_now =
  53         CldComponentInstallerTraits::GetLatestCldDataFile();
  54     ASSERT_TRUE(path_now.empty());
  55   }
  56
  57   void TearDown() override {
  58     // Restore sanity checks.
  59     translate::CldDataSource::EnableSanityChecksForTest();
  60   }
  61
  62  protected:
  63   base::ScopedTempDir temp_dir_;
  64   CldComponentInstallerTraits traits_;
  65
  66  private:
  67   DISALLOW_COPY_AND_ASSIGN(CldComponentInstallerTest);
  68 };
  69
  70 TEST_F(CldComponentInstallerTest, SetLatestCldDataFile) {
  71   const base::FilePath expected(FILE_PATH_LITERAL("test/foo.test"));
  72   CldComponentInstallerTraits::SetLatestCldDataFile(expected);
  73   base::FilePath result = CldComponentInstallerTraits::GetLatestCldDataFile();
  74   ASSERT_EQ(expected, result);
  75 }
  76
  77 TEST_F(CldComponentInstallerTest, VerifyInstallation) {
  78   // All files are created within a ScopedTempDir, which deletes all
  79   // children when its destructor is called (at the end of each test).
  80   const base::DictionaryValue manifest;
  81   ASSERT_FALSE(traits_.VerifyInstallation(manifest, temp_dir_.path()));
  82   const base::FilePath data_file_dir =
  83       temp_dir_.path()
  84           .Append(FILE_PATH_LITERAL("_platform_specific"))
  85           .Append(FILE_PATH_LITERAL("all"));
  86   ASSERT_TRUE(base::CreateDirectory(data_file_dir));
  87   const base::FilePath data_file = data_file_dir.Append(kTestCldDataFileName);
  88   const std::string test_data("fake cld2 data file content here :)");
  89   ASSERT_EQ(static_cast<int32>(test_data.length()),
  90             base::WriteFile(data_file, test_data.c_str(), test_data.length()));
  91   ASSERT_TRUE(traits_.VerifyInstallation(manifest, temp_dir_.path()));
  92 }
  93
  94 TEST_F(CldComponentInstallerTest, OnCustomInstall) {
  95   const base::DictionaryValue manifest;
  96   const base::FilePath install_dir;
  97   // Sanity: shouldn't crash.
  98   ASSERT_TRUE(traits_.OnCustomInstall(manifest, install_dir));
  99 }
 100
 101 TEST_F(CldComponentInstallerTest, GetInstalledPath) {
 102   const base::FilePath base_dir;
 103   const base::FilePath result =
 104       CldComponentInstallerTraits::GetInstalledPath(base_dir);
 105   ASSERT_TRUE(base::EndsWith(result.value(), kTestCldDataFileName,
 106                              base::CompareCase::SENSITIVE));
 107 }
 108
 109 TEST_F(CldComponentInstallerTest, GetBaseDirectory) {
 110   const base::FilePath result = traits_.GetBaseDirectory();
 111   ASSERT_FALSE(result.empty());
 112 }
 113
 114 TEST_F(CldComponentInstallerTest, GetHash) {
 115   std::vector<uint8_t> hash;
 116   traits_.GetHash(&hash);
 117   ASSERT_EQ(static_cast<size_t>(32), hash.size());
 118 }
 119
 120 TEST_F(CldComponentInstallerTest, GetName) {
 121   ASSERT_FALSE(traits_.GetName().empty());
 122 }
 123
 124 TEST_F(CldComponentInstallerTest, ComponentReady) {
 125   scoped_ptr<base::DictionaryValue> manifest;
 126   const base::FilePath install_dir(FILE_PATH_LITERAL("/foo"));
 127   const base::Version version("1.2.3.4");
 128   traits_.ComponentReady(version, install_dir, manifest.Pass());
 129   base::FilePath result = CldComponentInstallerTraits::GetLatestCldDataFile();
 130   ASSERT_TRUE(base::StartsWith(result.AsUTF16Unsafe(),
 131                                install_dir.AsUTF16Unsafe(),
 132                                base::CompareCase::SENSITIVE));
 133   ASSERT_TRUE(base::EndsWith(result.value(), kTestCldDataFileName,
 134                              base::CompareCase::SENSITIVE));
 135 }
 136
 137 }  // namespace component_updater