| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/extensions/api/image_writer_private/test_utils.h" | 5 #include "chrome/browser/extensions/api/image_writer_private/test_utils.h" |
| 6 | 6 |
| 7 #if defined(OS_CHROMEOS) | 7 #if defined(OS_CHROMEOS) |
| 8 #include "chromeos/dbus/dbus_thread_manager.h" | 8 #include "chromeos/dbus/dbus_thread_manager.h" |
| 9 #include "chromeos/dbus/fake_dbus_thread_manager.h" | |
| 10 #include "chromeos/dbus/fake_image_burner_client.h" | 9 #include "chromeos/dbus/fake_image_burner_client.h" |
| 11 #endif | 10 #endif |
| 12 | 11 |
| 13 namespace extensions { | 12 namespace extensions { |
| 14 namespace image_writer { | 13 namespace image_writer { |
| 15 | 14 |
| 16 #if defined(OS_CHROMEOS) | 15 #if defined(OS_CHROMEOS) |
| 17 namespace { | 16 namespace { |
| 18 | 17 |
| 19 class ImageWriterFakeImageBurnerClient | 18 class ImageWriterFakeImageBurnerClient |
| (...skipping 134 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 154 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 153 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
| 155 &test_image_path_)); | 154 &test_image_path_)); |
| 156 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 155 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
| 157 &test_device_path_)); | 156 &test_device_path_)); |
| 158 | 157 |
| 159 ASSERT_TRUE(FillFile(test_image_path_, kImagePattern, kTestFileSize)); | 158 ASSERT_TRUE(FillFile(test_image_path_, kImagePattern, kTestFileSize)); |
| 160 ASSERT_TRUE(FillFile(test_device_path_, kDevicePattern, kTestFileSize)); | 159 ASSERT_TRUE(FillFile(test_device_path_, kDevicePattern, kTestFileSize)); |
| 161 | 160 |
| 162 #if defined(OS_CHROMEOS) | 161 #if defined(OS_CHROMEOS) |
| 163 if (!chromeos::DBusThreadManager::IsInitialized()) { | 162 if (!chromeos::DBusThreadManager::IsInitialized()) { |
| 164 chromeos::FakeDBusThreadManager* fake_dbus_thread_manager = | 163 scoped_ptr<chromeos::DBusThreadManagerSetter> dbus_setter = |
| 165 new chromeos::FakeDBusThreadManager; | 164 chromeos::DBusThreadManager::GetSetterForTesting(); |
| 166 fake_dbus_thread_manager->SetFakeClients(); | |
| 167 scoped_ptr<chromeos::ImageBurnerClient> | 165 scoped_ptr<chromeos::ImageBurnerClient> |
| 168 image_burner_fake(new ImageWriterFakeImageBurnerClient()); | 166 image_burner_fake(new ImageWriterFakeImageBurnerClient()); |
| 169 fake_dbus_thread_manager->SetImageBurnerClient(image_burner_fake.Pass()); | 167 dbus_setter->SetImageBurnerClient(image_burner_fake.Pass()); |
| 170 | |
| 171 if (is_browser_test) { | |
| 172 chromeos::DBusThreadManager::SetInstanceForTesting( | |
| 173 fake_dbus_thread_manager); | |
| 174 } else { | |
| 175 chromeos::DBusThreadManager::InitializeForTesting( | |
| 176 fake_dbus_thread_manager); | |
| 177 } | |
| 178 } | 168 } |
| 179 | 169 |
| 180 FakeDiskMountManager* disk_manager = new FakeDiskMountManager(); | 170 FakeDiskMountManager* disk_manager = new FakeDiskMountManager(); |
| 181 chromeos::disks::DiskMountManager::InitializeForTesting(disk_manager); | 171 chromeos::disks::DiskMountManager::InitializeForTesting(disk_manager); |
| 182 | 172 |
| 183 // Adds a disk entry for test_device_path_ with the same device and file path. | 173 // Adds a disk entry for test_device_path_ with the same device and file path. |
| 184 disk_manager->CreateDiskEntryForMountDevice( | 174 disk_manager->CreateDiskEntryForMountDevice( |
| 185 chromeos::disks::DiskMountManager::MountPointInfo( | 175 chromeos::disks::DiskMountManager::MountPointInfo( |
| 186 test_device_path_.value(), | 176 test_device_path_.value(), |
| 187 "/dummy/mount", | 177 "/dummy/mount", |
| (...skipping 85 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 273 test_utils_.SetUp(); | 263 test_utils_.SetUp(); |
| 274 } | 264 } |
| 275 | 265 |
| 276 void ImageWriterUnitTestBase::TearDown() { | 266 void ImageWriterUnitTestBase::TearDown() { |
| 277 testing::Test::TearDown(); | 267 testing::Test::TearDown(); |
| 278 test_utils_.TearDown(); | 268 test_utils_.TearDown(); |
| 279 } | 269 } |
| 280 | 270 |
| 281 } // namespace image_writer | 271 } // namespace image_writer |
| 282 } // namespace extensions | 272 } // namespace extensions |
| OLD | NEW |