// Copyright (c) 2011 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "base/file_version_info.h" #include "base/files/file_path.h" #include "base/memory/scoped_ptr.h" #include "base/path_service.h" #include "testing/gtest/include/gtest/gtest.h" #if defined(OS_WIN) #include "base/file_version_info_win.h" #endif using base::FilePath; namespace { #if defined(OS_WIN) FilePath GetTestDataPath() { FilePath path; PathService::Get(base::DIR_SOURCE_ROOT, &path); path = path.AppendASCII("base"); path = path.AppendASCII("test"); path = path.AppendASCII("data"); path = path.AppendASCII("file_version_info_unittest"); return path; } #endif } // namespace #if defined(OS_WIN) TEST(FileVersionInfoTest, HardCodedProperties) { const wchar_t kDLLName[] = {L"FileVersionInfoTest1.dll"}; const wchar_t* const kExpectedValues[15] = { // FileVersionInfoTest.dll L"Goooooogle", // company_name L"Google", // company_short_name L"This is the product name", // product_name L"This is the product short name", // product_short_name L"The Internal Name", // internal_name L"4.3.2.1", // product_version L"Private build property", // private_build L"Special build property", // special_build L"This is a particularly interesting comment", // comments L"This is the original filename", // original_filename L"This is my file description", // file_description L"1.2.3.4", // file_version L"This is the legal copyright", // legal_copyright L"This is the legal trademarks", // legal_trademarks L"This is the last change", // last_change }; FilePath dll_path = GetTestDataPath(); dll_path = dll_path.Append(kDLLName); scoped_ptr version_info( FileVersionInfo::CreateFileVersionInfo(dll_path)); int j = 0; EXPECT_EQ(kExpectedValues[j++], version_info->company_name()); EXPECT_EQ(kExpectedValues[j++], version_info->company_short_name()); EXPECT_EQ(kExpectedValues[j++], version_info->product_name()); EXPECT_EQ(kExpectedValues[j++], version_info->product_short_name()); EXPECT_EQ(kExpectedValues[j++], version_info->internal_name()); EXPECT_EQ(kExpectedValues[j++], version_info->product_version()); EXPECT_EQ(kExpectedValues[j++], version_info->private_build()); EXPECT_EQ(kExpectedValues[j++], version_info->special_build()); EXPECT_EQ(kExpectedValues[j++], version_info->comments()); EXPECT_EQ(kExpectedValues[j++], version_info->original_filename()); EXPECT_EQ(kExpectedValues[j++], version_info->file_description()); EXPECT_EQ(kExpectedValues[j++], version_info->file_version()); EXPECT_EQ(kExpectedValues[j++], version_info->legal_copyright()); EXPECT_EQ(kExpectedValues[j++], version_info->legal_trademarks()); EXPECT_EQ(kExpectedValues[j++], version_info->last_change()); } #endif #if defined(OS_WIN) TEST(FileVersionInfoTest, IsOfficialBuild) { const wchar_t* kDLLNames[] = { L"FileVersionInfoTest1.dll", L"FileVersionInfoTest2.dll" }; const bool kExpected[] = { true, false, }; // Test consistency check. ASSERT_EQ(arraysize(kDLLNames), arraysize(kExpected)); for (int i = 0; i < arraysize(kDLLNames); ++i) { FilePath dll_path = GetTestDataPath(); dll_path = dll_path.Append(kDLLNames[i]); scoped_ptr version_info( FileVersionInfo::CreateFileVersionInfo(dll_path)); EXPECT_EQ(kExpected[i], version_info->is_official_build()); } } #endif #if defined(OS_WIN) TEST(FileVersionInfoTest, CustomProperties) { FilePath dll_path = GetTestDataPath(); dll_path = dll_path.AppendASCII("FileVersionInfoTest1.dll"); scoped_ptr version_info( FileVersionInfo::CreateFileVersionInfo(dll_path)); // Test few existing properties. std::wstring str; FileVersionInfoWin* version_info_win = static_cast(version_info.get()); EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 1", &str)); EXPECT_EQ(L"Un", str); EXPECT_EQ(L"Un", version_info_win->GetStringValue(L"Custom prop 1")); EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 2", &str)); EXPECT_EQ(L"Deux", str); EXPECT_EQ(L"Deux", version_info_win->GetStringValue(L"Custom prop 2")); EXPECT_TRUE(version_info_win->GetValue(L"Custom prop 3", &str)); EXPECT_EQ(L"1600 Amphitheatre Parkway Mountain View, CA 94043", str); EXPECT_EQ(L"1600 Amphitheatre Parkway Mountain View, CA 94043", version_info_win->GetStringValue(L"Custom prop 3")); // Test an non-existing property. EXPECT_FALSE(version_info_win->GetValue(L"Unknown property", &str)); EXPECT_EQ(L"", version_info_win->GetStringValue(L"Unknown property")); } #endif