base/file_version_info_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 "base/file_version_info.h"
   6 #include "base/files/file_path.h"
   7 #include "base/memory/scoped_ptr.h"
   8 #include "base/path_service.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 #if defined(OS_WIN)
  12 #include "base/file_version_info_win.h"
  13 #endif
  14
  15 using base::FilePath;
  16
  17 namespace {
  18
  19 #if defined(OS_WIN)
  20 FilePath GetTestDataPath() {
  21   FilePath path;
  22   PathService::Get(base::DIR_SOURCE_ROOT, &path);
  23   path = path.AppendASCII("base");
  24   path = path.AppendASCII("test");
  25   path = path.AppendASCII("data");
  26   path = path.AppendASCII("file_version_info_unittest");
  27   return path;
  28 }
  29 #endif
  30
  31 }  // namespace
  32
  33 #if defined(OS_WIN)
  34 TEST(FileVersionInfoTest, HardCodedProperties) {
  35   const wchar_t* kDLLNames[] = {
  36     L"FileVersionInfoTest1.dll"
  37   };
  38
  39   const wchar_t* kExpectedValues[1][15] = {
  40       // FileVersionInfoTest.dll
  41       L"Goooooogle",                      // company_name
  42       L"Google",                          // company_short_name
  43       L"This is the product name",        // product_name
  44       L"This is the product short name",  // product_short_name
  45       L"The Internal Name",               // internal_name
  46       L"4.3.2.1",                         // product_version
  47       L"Private build property",          // private_build
  48       L"Special build property",          // special_build
  49       L"This is a particularly interesting comment",  // comments
  50       L"This is the original filename",   // original_filename
  51       L"This is my file description",     // file_description
  52       L"1.2.3.4",                         // file_version
  53       L"This is the legal copyright",     // legal_copyright
  54       L"This is the legal trademarks",    // legal_trademarks
  55       L"This is the last change",         // last_change
  56   };
  57
  58   for (int i = 0; i < arraysize(kDLLNames); ++i) {
  59     FilePath dll_path = GetTestDataPath();
  60     dll_path = dll_path.Append(kDLLNames[i]);
  61
  62     scoped_ptr<FileVersionInfo> version_info(
  63         FileVersionInfo::CreateFileVersionInfo(dll_path));
  64
  65     int j = 0;
  66     EXPECT_EQ(kExpectedValues[i][j++], version_info->company_name());
  67     EXPECT_EQ(kExpectedValues[i][j++], version_info->company_short_name());
  68     EXPECT_EQ(kExpectedValues[i][j++], version_info->product_name());
  69     EXPECT_EQ(kExpectedValues[i][j++], version_info->product_short_name());
  70     EXPECT_EQ(kExpectedValues[i][j++], version_info->internal_name());
  71     EXPECT_EQ(kExpectedValues[i][j++], version_info->product_version());
  72     EXPECT_EQ(kExpectedValues[i][j++], version_info->private_build());
  73     EXPECT_EQ(kExpectedValues[i][j++], version_info->special_build());
  74     EXPECT_EQ(kExpectedValues[i][j++], version_info->comments());
  75     EXPECT_EQ(kExpectedValues[i][j++], version_info->original_filename());
  76     EXPECT_EQ(kExpectedValues[i][j++], version_info->file_description());
  77     EXPECT_EQ(kExpectedValues[i][j++], version_info->file_version());
  78     EXPECT_EQ(kExpectedValues[i][j++], version_info->legal_copyright());
  79     EXPECT_EQ(kExpectedValues[i][j++], version_info->legal_trademarks());
  80     EXPECT_EQ(kExpectedValues[i][j++], version_info->last_change());
  81   }
  82 }
  83 #endif
  84
  85 #if defined(OS_WIN)
  86 TEST(FileVersionInfoTest, IsOfficialBuild) {
  87   const wchar_t* kDLLNames[] = {
  88     L"FileVersionInfoTest1.dll",
  89     L"FileVersionInfoTest2.dll"
  90   };
  91
  92   const bool kExpected[] = {
  93     true,
  94     false,
  95   };
  96
  97   // Test consistency check.
  98   ASSERT_EQ(arraysize(kDLLNames), arraysize(kExpected));
  99
 100   for (int i = 0; i < arraysize(kDLLNames); ++i) {
 101     FilePath dll_path = GetTestDataPath();
 102     dll_path = dll_path.Append(kDLLNames[i]);
 103
 104     scoped_ptr<FileVersionInfo> version_info(
 105         FileVersionInfo::CreateFileVersionInfo(dll_path));
 106
 107     EXPECT_EQ(kExpected[i], version_info->is_official_build());
 108   }
 109 }
 110 #endif
 111
 112 #if defined(OS_WIN)
 113 TEST(FileVersionInfoTest, CustomProperties) {
 114   FilePath dll_path = GetTestDataPath();
 115   dll_path = dll_path.AppendASCII("FileVersionInfoTest1.dll");
 116
 117   scoped_ptr<FileVersionInfo> version_info(
 118       FileVersionInfo::CreateFileVersionInfo(dll_path));
 119
 120   // Test few existing properties.
 121   std::wstring str;
 122   FileVersionInfoWin* version_info_win =
 123       static_cast<FileVersionInfoWin*>(version_info.get());
 124   EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 1",  &str));
 125   EXPECT_EQ(L"Un", str);
 126   EXPECT_EQ(L"Un", version_info_win->GetStringValue(L"Custom prop 1"));
 127
 128   EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 2",  &str));
 129   EXPECT_EQ(L"Deux", str);
 130   EXPECT_EQ(L"Deux", version_info_win->GetStringValue(L"Custom prop 2"));
 131
 132   EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 3",  &str));
 133   EXPECT_EQ(L"1600 Amphitheatre Parkway Mountain View, CA 94043", str);
 134   EXPECT_EQ(L"1600 Amphitheatre Parkway Mountain View, CA 94043",
 135             version_info_win->GetStringValue(L"Custom prop 3"));
 136
 137   // Test an non-existing property.
 138   EXPECT_FALSE(version_info_win->GetValue(L"Unknown property",  &str));
 139   EXPECT_EQ(L"", version_info_win->GetStringValue(L"Unknown property"));
 140 }
 141 #endif