1a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)// Copyright 2013 The Chromium Authors. All rights reserved. 2a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 3a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)// found in the LICENSE file. 4a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 5a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "chrome/browser/extensions/api/image_writer_private/test_utils.h" 6a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#if defined(OS_CHROMEOS) 85d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "chromeos/dbus/dbus_thread_manager.h" 95d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "chromeos/dbus/fake_image_burner_client.h" 105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#endif 115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 12a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)namespace extensions { 13a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)namespace image_writer { 14a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#if defined(OS_CHROMEOS) 165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)namespace { 17a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)class ImageWriterFakeImageBurnerClient 195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : public chromeos::FakeImageBurnerClient { 205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) public: 215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ImageWriterFakeImageBurnerClient() {} 225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual ~ImageWriterFakeImageBurnerClient() {} 235d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 245d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual void SetEventHandlers( 255d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const BurnFinishedHandler& burn_finished_handler, 265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const BurnProgressUpdateHandler& burn_progress_update_handler) OVERRIDE { 275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) burn_finished_handler_ = burn_finished_handler; 285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) burn_progress_update_handler_ = burn_progress_update_handler; 295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) virtual void BurnImage(const std::string& from_path, 325d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const std::string& to_path, 335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const ErrorCallback& error_callback) OVERRIDE { 345d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::MessageLoop::current()->PostTask(FROM_HERE, 355d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind(burn_progress_update_handler_, to_path, 0, 100)); 365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::MessageLoop::current()->PostTask(FROM_HERE, 375d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind(burn_progress_update_handler_, to_path, 50, 100)); 385d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::MessageLoop::current()->PostTask(FROM_HERE, 395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind(burn_progress_update_handler_, to_path, 100, 100)); 405d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::MessageLoop::current()->PostTask(FROM_HERE, 415d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::Bind(burn_finished_handler_, to_path, true, "")); 425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 435d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) private: 455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) BurnFinishedHandler burn_finished_handler_; 465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) BurnProgressUpdateHandler burn_progress_update_handler_; 475d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)}; 485d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} // namespace 505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#endif 515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)MockOperationManager::MockOperationManager() : OperationManager(NULL) {} 53a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)MockOperationManager::MockOperationManager(content::BrowserContext* context) 54a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) : OperationManager(context) {} 555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)MockOperationManager::~MockOperationManager() {} 565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 57cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#if defined(OS_CHROMEOS) 58cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)FakeDiskMountManager::FakeDiskMountManager() {} 59cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)FakeDiskMountManager::~FakeDiskMountManager() {} 60cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 61cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)void FakeDiskMountManager::UnmountDeviceRecursively( 62cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) const std::string& device_path, 63cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) const UnmountDeviceRecursivelyCallbackType& callback) { 64cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) base::MessageLoop::current()->PostTask(FROM_HERE, base::Bind(callback, true)); 65cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)} 66cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#endif 67cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 68a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)FakeImageWriterClient::FakeImageWriterClient() {} 69a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)FakeImageWriterClient::~FakeImageWriterClient() {} 70a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 71a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void FakeImageWriterClient::Write(const ProgressCallback& progress_callback, 72a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const SuccessCallback& success_callback, 73a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const ErrorCallback& error_callback, 74a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const base::FilePath& source, 75a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const base::FilePath& target) { 76a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) progress_callback_ = progress_callback; 77a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) success_callback_ = success_callback; 78a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) error_callback_ = error_callback; 79116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 80116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!write_callback_.is_null()) 81116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch write_callback_.Run(); 82a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 83a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 84a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void FakeImageWriterClient::Verify(const ProgressCallback& progress_callback, 85a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const SuccessCallback& success_callback, 86a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const ErrorCallback& error_callback, 87a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const base::FilePath& source, 88a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const base::FilePath& target) { 89a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) progress_callback_ = progress_callback; 90a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) success_callback_ = success_callback; 91a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) error_callback_ = error_callback; 92116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 93116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!verify_callback_.is_null()) 94116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch verify_callback_.Run(); 95a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 96a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 97a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void FakeImageWriterClient::Cancel(const CancelCallback& cancel_callback) { 98a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) cancel_callback_ = cancel_callback; 99a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 100a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 101a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void FakeImageWriterClient::Shutdown() { 102a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) // Clear handlers to not hold any reference to the caller. 103116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch success_callback_.Reset(); 104116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch progress_callback_.Reset(); 105116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch error_callback_.Reset(); 106116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch cancel_callback_.Reset(); 107116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 108116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch write_callback_.Reset(); 109116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch verify_callback_.Reset(); 110116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 111116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 112116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid FakeImageWriterClient::SetWriteCallback( 113116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch const base::Closure& write_callback) { 114116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch write_callback_ = write_callback; 115116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 116116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 117116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid FakeImageWriterClient::SetVerifyCallback( 118116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch const base::Closure& verify_callback) { 119116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch verify_callback_ = verify_callback; 120a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 121a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 122a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void FakeImageWriterClient::Progress(int64 progress) { 123116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!progress_callback_.is_null()) 124116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch progress_callback_.Run(progress); 125a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 126a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 127116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid FakeImageWriterClient::Success() { 128116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!success_callback_.is_null()) 129116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch success_callback_.Run(); 130116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 131a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 132a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void FakeImageWriterClient::Error(const std::string& message) { 133116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!error_callback_.is_null()) 134116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch error_callback_.Run(message); 135a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 136a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 137116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid FakeImageWriterClient::Cancel() { 138116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!cancel_callback_.is_null()) 139116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch cancel_callback_.Run(); 140a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)} 141a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 142116680a4aac90f2aa7413d9095a592090648e557Ben MurdochImageWriterTestUtils::ImageWriterTestUtils() { 143116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 144116680a4aac90f2aa7413d9095a592090648e557Ben MurdochImageWriterTestUtils::~ImageWriterTestUtils() { 145116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 146a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 147116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid ImageWriterTestUtils::SetUp() { 148116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch SetUp(false); 149116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 1505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 151116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid ImageWriterTestUtils::SetUp(bool is_browser_test) { 152a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); 1535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), 1545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) &test_image_path_)); 1555d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), 1565d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) &test_device_path_)); 157a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 1585d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(FillFile(test_image_path_, kImagePattern, kTestFileSize)); 1595d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) ASSERT_TRUE(FillFile(test_device_path_, kDevicePattern, kTestFileSize)); 1605d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 1615d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#if defined(OS_CHROMEOS) 1625d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (!chromeos::DBusThreadManager::IsInitialized()) { 1631320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<chromeos::DBusThreadManagerSetter> dbus_setter = 1641320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci chromeos::DBusThreadManager::GetSetterForTesting(); 1655d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<chromeos::ImageBurnerClient> 1665d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) image_burner_fake(new ImageWriterFakeImageBurnerClient()); 1671320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci dbus_setter->SetImageBurnerClient(image_burner_fake.Pass()); 1685d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 169116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 170cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) FakeDiskMountManager* disk_manager = new FakeDiskMountManager(); 171cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) chromeos::disks::DiskMountManager::InitializeForTesting(disk_manager); 172cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 173cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) // Adds a disk entry for test_device_path_ with the same device and file path. 174cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) disk_manager->CreateDiskEntryForMountDevice( 175cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) chromeos::disks::DiskMountManager::MountPointInfo( 176cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) test_device_path_.value(), 177cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) "/dummy/mount", 178cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) chromeos::MOUNT_TYPE_DEVICE, 179cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) chromeos::disks::MOUNT_CONDITION_NONE), 180cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) "device_id", 181cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) "device_label", 182cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) "Vendor", 183cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) "Product", 184cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) chromeos::DEVICE_TYPE_USB, 185cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) kTestFileSize, 186cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) true, 187cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) true, 1885f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) true, 189cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) false); 190cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) disk_manager->SetupDefaultReplies(); 191116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#else 192116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch client_ = new FakeImageWriterClient(); 193116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch image_writer::Operation::SetUtilityClientForTesting(client_); 1945d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#endif 195a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)} 196a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 197116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid ImageWriterTestUtils::TearDown() { 1985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#if defined(OS_CHROMEOS) 199116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (chromeos::DBusThreadManager::IsInitialized()) { 200116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch chromeos::DBusThreadManager::Shutdown(); 201116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch } 202cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) chromeos::disks::DiskMountManager::Shutdown(); 203116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#else 204116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch image_writer::Operation::SetUtilityClientForTesting(NULL); 205116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch client_->Shutdown(); 2065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#endif 2075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 2085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 209116680a4aac90f2aa7413d9095a592090648e557Ben Murdochconst base::FilePath& ImageWriterTestUtils::GetTempDir() { 210116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return temp_dir_.path(); 211116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 212116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 213116680a4aac90f2aa7413d9095a592090648e557Ben Murdochconst base::FilePath& ImageWriterTestUtils::GetImagePath() { 214116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return test_image_path_; 215116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 216116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 217116680a4aac90f2aa7413d9095a592090648e557Ben Murdochconst base::FilePath& ImageWriterTestUtils::GetDevicePath() { 218116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return test_device_path_; 219116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 220116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 221116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#if !defined(OS_CHROMEOS) 222116680a4aac90f2aa7413d9095a592090648e557Ben MurdochFakeImageWriterClient* ImageWriterTestUtils::GetUtilityClient() { 223116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return client_.get(); 224116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 225116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#endif 226116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 227116680a4aac90f2aa7413d9095a592090648e557Ben Murdochbool ImageWriterTestUtils::ImageWrittenToDevice() { 2285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<char[]> image_buffer(new char[kTestFileSize]); 2295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<char[]> device_buffer(new char[kTestFileSize]); 2305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 231a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int image_bytes_read = 232116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch ReadFile(test_image_path_, image_buffer.get(), kTestFileSize); 2335d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 234a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) if (image_bytes_read < 0) 235a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return false; 2365d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 237a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) int device_bytes_read = 238116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch ReadFile(test_device_path_, device_buffer.get(), kTestFileSize); 2395d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 240a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) if (image_bytes_read != device_bytes_read) 241a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return false; 2425d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 243a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return memcmp(image_buffer.get(), device_buffer.get(), image_bytes_read) == 0; 2445d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 2455d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 246116680a4aac90f2aa7413d9095a592090648e557Ben Murdochbool ImageWriterTestUtils::FillFile(const base::FilePath& file, 247116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch const int pattern, 248116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch const int length) { 2495d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<char[]> buffer(new char[length]); 2505d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) memset(buffer.get(), pattern, length); 2515d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 252a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return base::WriteFile(file, buffer.get(), length) == length; 253a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)} 254a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) 255116680a4aac90f2aa7413d9095a592090648e557Ben MurdochImageWriterUnitTestBase::ImageWriterUnitTestBase() 256116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) { 257116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 258116680a4aac90f2aa7413d9095a592090648e557Ben MurdochImageWriterUnitTestBase::~ImageWriterUnitTestBase() { 259116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 260116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 261116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid ImageWriterUnitTestBase::SetUp() { 262116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch testing::Test::SetUp(); 263116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch test_utils_.SetUp(); 264116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 265116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 266116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid ImageWriterUnitTestBase::TearDown() { 267116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch testing::Test::TearDown(); 268116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch test_utils_.TearDown(); 269116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 270116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 271a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)} // namespace image_writer 272a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)} // namespace extensions 273