OLD | NEW |
| (Empty) |
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_win.h" | |
6 | |
7 #include <windows.h> | |
8 | |
9 #include <stddef.h> | |
10 | |
11 #include <memory> | |
12 | |
13 #include "base/file_version_info.h" | |
14 #include "base/files/file_path.h" | |
15 #include "base/macros.h" | |
16 #include "base/memory/ptr_util.h" | |
17 #include "base/path_service.h" | |
18 #include "base/scoped_native_library.h" | |
19 #include "testing/gtest/include/gtest/gtest.h" | |
20 | |
21 using base::FilePath; | |
22 | |
23 namespace { | |
24 | |
25 FilePath GetTestDataPath() { | |
26 FilePath path; | |
27 PathService::Get(base::DIR_SOURCE_ROOT, &path); | |
28 path = path.AppendASCII("base"); | |
29 path = path.AppendASCII("test"); | |
30 path = path.AppendASCII("data"); | |
31 path = path.AppendASCII("file_version_info_unittest"); | |
32 return path; | |
33 } | |
34 | |
35 class FileVersionInfoFactory { | |
36 public: | |
37 explicit FileVersionInfoFactory(const FilePath& path) : path_(path) {} | |
38 | |
39 std::unique_ptr<FileVersionInfo> Create() const { | |
40 return base::WrapUnique(FileVersionInfo::CreateFileVersionInfo(path_)); | |
41 } | |
42 | |
43 private: | |
44 const FilePath path_; | |
45 | |
46 DISALLOW_COPY_AND_ASSIGN(FileVersionInfoFactory); | |
47 }; | |
48 | |
49 class FileVersionInfoForModuleFactory { | |
50 public: | |
51 explicit FileVersionInfoForModuleFactory(const FilePath& path) | |
52 // Load the library with LOAD_LIBRARY_AS_IMAGE_RESOURCE since it shouldn't | |
53 // be executed. | |
54 : library_(::LoadLibraryEx(path.value().c_str(), | |
55 nullptr, | |
56 LOAD_LIBRARY_AS_IMAGE_RESOURCE)) { | |
57 EXPECT_TRUE(library_.is_valid()); | |
58 } | |
59 | |
60 std::unique_ptr<FileVersionInfo> Create() const { | |
61 return base::WrapUnique( | |
62 FileVersionInfo::CreateFileVersionInfoForModule(library_.get())); | |
63 } | |
64 | |
65 private: | |
66 const base::ScopedNativeLibrary library_; | |
67 | |
68 DISALLOW_COPY_AND_ASSIGN(FileVersionInfoForModuleFactory); | |
69 }; | |
70 | |
71 template <typename T> | |
72 class FileVersionInfoTest : public testing::Test {}; | |
73 | |
74 using FileVersionInfoFactories = | |
75 ::testing::Types<FileVersionInfoFactory, FileVersionInfoForModuleFactory>; | |
76 | |
77 } // namespace | |
78 | |
79 TYPED_TEST_CASE(FileVersionInfoTest, FileVersionInfoFactories); | |
80 | |
81 TYPED_TEST(FileVersionInfoTest, HardCodedProperties) { | |
82 const wchar_t kDLLName[] = {L"FileVersionInfoTest1.dll"}; | |
83 | |
84 const wchar_t* const kExpectedValues[15] = { | |
85 // FileVersionInfoTest.dll | |
86 L"Goooooogle", // company_name | |
87 L"Google", // company_short_name | |
88 L"This is the product name", // product_name | |
89 L"This is the product short name", // product_short_name | |
90 L"The Internal Name", // internal_name | |
91 L"4.3.2.1", // product_version | |
92 L"Private build property", // private_build | |
93 L"Special build property", // special_build | |
94 L"This is a particularly interesting comment", // comments | |
95 L"This is the original filename", // original_filename | |
96 L"This is my file description", // file_description | |
97 L"1.2.3.4", // file_version | |
98 L"This is the legal copyright", // legal_copyright | |
99 L"This is the legal trademarks", // legal_trademarks | |
100 L"This is the last change", // last_change | |
101 }; | |
102 | |
103 FilePath dll_path = GetTestDataPath(); | |
104 dll_path = dll_path.Append(kDLLName); | |
105 | |
106 TypeParam factory(dll_path); | |
107 std::unique_ptr<FileVersionInfo> version_info(factory.Create()); | |
108 ASSERT_TRUE(version_info); | |
109 | |
110 int j = 0; | |
111 EXPECT_EQ(kExpectedValues[j++], version_info->company_name()); | |
112 EXPECT_EQ(kExpectedValues[j++], version_info->company_short_name()); | |
113 EXPECT_EQ(kExpectedValues[j++], version_info->product_name()); | |
114 EXPECT_EQ(kExpectedValues[j++], version_info->product_short_name()); | |
115 EXPECT_EQ(kExpectedValues[j++], version_info->internal_name()); | |
116 EXPECT_EQ(kExpectedValues[j++], version_info->product_version()); | |
117 EXPECT_EQ(kExpectedValues[j++], version_info->private_build()); | |
118 EXPECT_EQ(kExpectedValues[j++], version_info->special_build()); | |
119 EXPECT_EQ(kExpectedValues[j++], version_info->comments()); | |
120 EXPECT_EQ(kExpectedValues[j++], version_info->original_filename()); | |
121 EXPECT_EQ(kExpectedValues[j++], version_info->file_description()); | |
122 EXPECT_EQ(kExpectedValues[j++], version_info->file_version()); | |
123 EXPECT_EQ(kExpectedValues[j++], version_info->legal_copyright()); | |
124 EXPECT_EQ(kExpectedValues[j++], version_info->legal_trademarks()); | |
125 EXPECT_EQ(kExpectedValues[j++], version_info->last_change()); | |
126 } | |
127 | |
128 TYPED_TEST(FileVersionInfoTest, IsOfficialBuild) { | |
129 constexpr struct { | |
130 const wchar_t* const dll_name; | |
131 const bool is_official_build; | |
132 } kTestItems[]{ | |
133 {L"FileVersionInfoTest1.dll", true}, {L"FileVersionInfoTest2.dll", false}, | |
134 }; | |
135 | |
136 for (const auto& test_item : kTestItems) { | |
137 const FilePath dll_path = GetTestDataPath().Append(test_item.dll_name); | |
138 | |
139 TypeParam factory(dll_path); | |
140 std::unique_ptr<FileVersionInfo> version_info(factory.Create()); | |
141 ASSERT_TRUE(version_info); | |
142 | |
143 EXPECT_EQ(test_item.is_official_build, version_info->is_official_build()); | |
144 } | |
145 } | |
146 | |
147 TYPED_TEST(FileVersionInfoTest, CustomProperties) { | |
148 FilePath dll_path = GetTestDataPath(); | |
149 dll_path = dll_path.AppendASCII("FileVersionInfoTest1.dll"); | |
150 | |
151 TypeParam factory(dll_path); | |
152 std::unique_ptr<FileVersionInfo> version_info(factory.Create()); | |
153 ASSERT_TRUE(version_info); | |
154 | |
155 // Test few existing properties. | |
156 std::wstring str; | |
157 FileVersionInfoWin* version_info_win = | |
158 static_cast<FileVersionInfoWin*>(version_info.get()); | |
159 EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 1", &str)); | |
160 EXPECT_EQ(L"Un", str); | |
161 EXPECT_EQ(L"Un", version_info_win->GetStringValue(L"Custom prop 1")); | |
162 | |
163 EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 2", &str)); | |
164 EXPECT_EQ(L"Deux", str); | |
165 EXPECT_EQ(L"Deux", version_info_win->GetStringValue(L"Custom prop 2")); | |
166 | |
167 EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 3", &str)); | |
168 EXPECT_EQ(L"1600 Amphitheatre Parkway Mountain View, CA 94043", str); | |
169 EXPECT_EQ(L"1600 Amphitheatre Parkway Mountain View, CA 94043", | |
170 version_info_win->GetStringValue(L"Custom prop 3")); | |
171 | |
172 // Test an non-existing property. | |
173 EXPECT_FALSE(version_info_win->GetValue(L"Unknown property", &str)); | |
174 EXPECT_EQ(L"", version_info_win->GetStringValue(L"Unknown property")); | |
175 } | |
OLD | NEW |