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