| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "cloud_print/virtual_driver/win/port_monitor/port_monitor.h" | 5 #include "cloud_print/virtual_driver/win/port_monitor/port_monitor.h" |
| 6 #include <winspool.h> | 6 #include <winspool.h> |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/path_service.h" | 8 #include "base/path_service.h" |
| 9 #include "base/string16.h" | 9 #include "base/string16.h" |
| 10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
| (...skipping 21 matching lines...) Expand all Loading... |
| 32 public: | 32 public: |
| 33 PortMonitorTest() {} | 33 PortMonitorTest() {} |
| 34 protected: | 34 protected: |
| 35 // Creates a registry entry pointing at a chrome | 35 // Creates a registry entry pointing at a chrome |
| 36 virtual void SetUpChromeExeRegistry() { | 36 virtual void SetUpChromeExeRegistry() { |
| 37 // Create a temporary chrome.exe location value. | 37 // Create a temporary chrome.exe location value. |
| 38 base::win::RegKey key(HKEY_CURRENT_USER, | 38 base::win::RegKey key(HKEY_CURRENT_USER, |
| 39 cloud_print::kCloudPrintRegKey, | 39 cloud_print::kCloudPrintRegKey, |
| 40 KEY_ALL_ACCESS); | 40 KEY_ALL_ACCESS); |
| 41 | 41 |
| 42 FilePath path; | 42 base::FilePath path; |
| 43 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); | 43 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); |
| 44 path = path.Append(kAlternateChromeExePath); | 44 path = path.Append(kAlternateChromeExePath); |
| 45 ASSERT_EQ(ERROR_SUCCESS, | 45 ASSERT_EQ(ERROR_SUCCESS, |
| 46 key.WriteValue(cloud_print::kChromeExePathRegValue, | 46 key.WriteValue(cloud_print::kChromeExePathRegValue, |
| 47 path.value().c_str())); | 47 path.value().c_str())); |
| 48 FilePath temp; | 48 base::FilePath temp; |
| 49 PathService::Get(base::DIR_TEMP, &temp); | 49 PathService::Get(base::DIR_TEMP, &temp); |
| 50 // Write any dir here. | 50 // Write any dir here. |
| 51 ASSERT_EQ(ERROR_SUCCESS, | 51 ASSERT_EQ(ERROR_SUCCESS, |
| 52 key.WriteValue(cloud_print::kChromeProfilePathRegValue, | 52 key.WriteValue(cloud_print::kChromeProfilePathRegValue, |
| 53 temp.value().c_str())); | 53 temp.value().c_str())); |
| 54 } | 54 } |
| 55 // Deletes the registry entry created in SetUpChromeExeRegistry | 55 // Deletes the registry entry created in SetUpChromeExeRegistry |
| 56 virtual void DeleteChromeExeRegistry() { | 56 virtual void DeleteChromeExeRegistry() { |
| 57 base::win::RegKey key(HKEY_CURRENT_USER, | 57 base::win::RegKey key(HKEY_CURRENT_USER, |
| 58 cloud_print::kCloudPrintRegKey, | 58 cloud_print::kCloudPrintRegKey, |
| 59 KEY_ALL_ACCESS); | 59 KEY_ALL_ACCESS); |
| 60 key.DeleteValue(cloud_print::kChromeExePathRegValue); | 60 key.DeleteValue(cloud_print::kChromeExePathRegValue); |
| 61 key.DeleteValue(cloud_print::kChromeProfilePathRegValue); | 61 key.DeleteValue(cloud_print::kChromeProfilePathRegValue); |
| 62 } | 62 } |
| 63 | 63 |
| 64 virtual void CreateTempChromeExeFiles() { | 64 virtual void CreateTempChromeExeFiles() { |
| 65 FilePath path; | 65 base::FilePath path; |
| 66 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); | 66 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); |
| 67 FilePath main_path = path.Append(kChromeExePath); | 67 base::FilePath main_path = path.Append(kChromeExePath); |
| 68 ASSERT_TRUE(file_util::CreateDirectory(main_path)); | 68 ASSERT_TRUE(file_util::CreateDirectory(main_path)); |
| 69 FilePath alternate_path = path.Append(kAlternateChromeExePath); | 69 base::FilePath alternate_path = path.Append(kAlternateChromeExePath); |
| 70 ASSERT_TRUE(file_util::CreateDirectory(alternate_path)); | 70 ASSERT_TRUE(file_util::CreateDirectory(alternate_path)); |
| 71 } | 71 } |
| 72 | 72 |
| 73 virtual void DeleteTempChromeExeFiles() { | 73 virtual void DeleteTempChromeExeFiles() { |
| 74 FilePath path; | 74 base::FilePath path; |
| 75 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); | 75 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); |
| 76 FilePath main_path = path.Append(kChromeExePath); | 76 base::FilePath main_path = path.Append(kChromeExePath); |
| 77 ASSERT_TRUE(file_util::Delete(main_path, true)); | 77 ASSERT_TRUE(file_util::Delete(main_path, true)); |
| 78 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); | 78 PathService::Get(base::DIR_LOCAL_APP_DATA, &path); |
| 79 FilePath alternate_path = path.Append(kAlternateChromeExePath); | 79 base::FilePath alternate_path = path.Append(kAlternateChromeExePath); |
| 80 ASSERT_TRUE(file_util::Delete(alternate_path, true)); | 80 ASSERT_TRUE(file_util::Delete(alternate_path, true)); |
| 81 } | 81 } |
| 82 | 82 |
| 83 protected: | 83 protected: |
| 84 virtual void SetUp() { | 84 virtual void SetUp() { |
| 85 SetUpChromeExeRegistry(); | 85 SetUpChromeExeRegistry(); |
| 86 } | 86 } |
| 87 | 87 |
| 88 virtual void TearDown() { | 88 virtual void TearDown() { |
| 89 DeleteChromeExeRegistry(); | 89 DeleteChromeExeRegistry(); |
| 90 } | 90 } |
| 91 | 91 |
| 92 private: | 92 private: |
| 93 DISALLOW_COPY_AND_ASSIGN(PortMonitorTest); | 93 DISALLOW_COPY_AND_ASSIGN(PortMonitorTest); |
| 94 }; | 94 }; |
| 95 | 95 |
| 96 TEST_F(PortMonitorTest, GetChromeExePathTest) { | 96 TEST_F(PortMonitorTest, GetChromeExePathTest) { |
| 97 CreateTempChromeExeFiles(); | 97 CreateTempChromeExeFiles(); |
| 98 FilePath chrome_path = cloud_print::GetChromeExePath(); | 98 base::FilePath chrome_path = cloud_print::GetChromeExePath(); |
| 99 EXPECT_FALSE(chrome_path.empty()); | 99 EXPECT_FALSE(chrome_path.empty()); |
| 100 EXPECT_TRUE( | 100 EXPECT_TRUE( |
| 101 chrome_path.value().rfind(kAlternateChromeExePath) != std::string::npos); | 101 chrome_path.value().rfind(kAlternateChromeExePath) != std::string::npos); |
| 102 EXPECT_TRUE(file_util::PathExists(chrome_path)); | 102 EXPECT_TRUE(file_util::PathExists(chrome_path)); |
| 103 DeleteChromeExeRegistry(); | 103 DeleteChromeExeRegistry(); |
| 104 chrome_path = cloud_print::GetChromeExePath(); | 104 chrome_path = cloud_print::GetChromeExePath(); |
| 105 // No Chrome or regular chrome path. | 105 // No Chrome or regular chrome path. |
| 106 EXPECT_TRUE(chrome_path.empty() || | 106 EXPECT_TRUE(chrome_path.empty() || |
| 107 chrome_path.value().rfind(kChromeExePath) == std::string::npos); | 107 chrome_path.value().rfind(kChromeExePath) == std::string::npos); |
| 108 } | 108 } |
| 109 | 109 |
| 110 TEST_F(PortMonitorTest, GetChromeProfilePathTest) { | 110 TEST_F(PortMonitorTest, GetChromeProfilePathTest) { |
| 111 FilePath data_path = cloud_print::GetChromeProfilePath(); | 111 base::FilePath data_path = cloud_print::GetChromeProfilePath(); |
| 112 EXPECT_FALSE(data_path.empty()); | 112 EXPECT_FALSE(data_path.empty()); |
| 113 FilePath temp; | 113 base::FilePath temp; |
| 114 PathService::Get(base::DIR_TEMP, &temp); | 114 PathService::Get(base::DIR_TEMP, &temp); |
| 115 EXPECT_EQ(data_path, temp); | 115 EXPECT_EQ(data_path, temp); |
| 116 EXPECT_TRUE(file_util::DirectoryExists(data_path)); | 116 EXPECT_TRUE(file_util::DirectoryExists(data_path)); |
| 117 DeleteChromeExeRegistry(); | 117 DeleteChromeExeRegistry(); |
| 118 data_path = cloud_print::GetChromeProfilePath(); | 118 data_path = cloud_print::GetChromeProfilePath(); |
| 119 EXPECT_TRUE(data_path.empty()); | 119 EXPECT_TRUE(data_path.empty()); |
| 120 } | 120 } |
| 121 | 121 |
| 122 TEST_F(PortMonitorTest, EnumPortsTest) { | 122 TEST_F(PortMonitorTest, EnumPortsTest) { |
| 123 DWORD needed_bytes = 0; | 123 DWORD needed_bytes = 0; |
| (...skipping 141 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 265 EXPECT_TRUE(monitor2->pfnEndDocPort(port_handle)); | 265 EXPECT_TRUE(monitor2->pfnEndDocPort(port_handle)); |
| 266 RevertToSelf(); | 266 RevertToSelf(); |
| 267 EXPECT_TRUE(monitor2->pfnClosePort != NULL); | 267 EXPECT_TRUE(monitor2->pfnClosePort != NULL); |
| 268 EXPECT_TRUE(monitor2->pfnClosePort(port_handle)); | 268 EXPECT_TRUE(monitor2->pfnClosePort(port_handle)); |
| 269 // Shutdown the port monitor. | 269 // Shutdown the port monitor. |
| 270 Monitor2Shutdown(monitor_handle); | 270 Monitor2Shutdown(monitor_handle); |
| 271 } | 271 } |
| 272 | 272 |
| 273 } // namespace cloud_print | 273 } // namespace cloud_print |
| 274 | 274 |
| OLD | NEW |