| Index: trunk/src/chrome/browser/extensions/api/image_writer_private/test_utils.cc
|
| ===================================================================
|
| --- trunk/src/chrome/browser/extensions/api/image_writer_private/test_utils.cc (revision 282139)
|
| +++ trunk/src/chrome/browser/extensions/api/image_writer_private/test_utils.cc (working copy)
|
| @@ -77,9 +77,6 @@
|
| progress_callback_ = progress_callback;
|
| success_callback_ = success_callback;
|
| error_callback_ = error_callback;
|
| -
|
| - if (!write_callback_.is_null())
|
| - write_callback_.Run();
|
| }
|
|
|
| void FakeImageWriterClient::Verify(const ProgressCallback& progress_callback,
|
| @@ -90,9 +87,6 @@
|
| progress_callback_ = progress_callback;
|
| success_callback_ = success_callback;
|
| error_callback_ = error_callback;
|
| -
|
| - if (!verify_callback_.is_null())
|
| - verify_callback_.Run();
|
| }
|
|
|
| void FakeImageWriterClient::Cancel(const CancelCallback& cancel_callback) {
|
| @@ -101,55 +95,35 @@
|
|
|
| void FakeImageWriterClient::Shutdown() {
|
| // Clear handlers to not hold any reference to the caller.
|
| - success_callback_.Reset();
|
| - progress_callback_.Reset();
|
| - error_callback_.Reset();
|
| - cancel_callback_.Reset();
|
| -
|
| - write_callback_.Reset();
|
| - verify_callback_.Reset();
|
| + success_callback_ = base::Closure();
|
| + progress_callback_ = base::Callback<void(int64)>();
|
| + error_callback_ = base::Callback<void(const std::string&)>();
|
| + cancel_callback_ = base::Closure();
|
| }
|
|
|
| -void FakeImageWriterClient::SetWriteCallback(
|
| - const base::Closure& write_callback) {
|
| - write_callback_ = write_callback;
|
| -}
|
| -
|
| -void FakeImageWriterClient::SetVerifyCallback(
|
| - const base::Closure& verify_callback) {
|
| - verify_callback_ = verify_callback;
|
| -}
|
| -
|
| void FakeImageWriterClient::Progress(int64 progress) {
|
| - if (!progress_callback_.is_null())
|
| - progress_callback_.Run(progress);
|
| + progress_callback_.Run(progress);
|
| }
|
|
|
| -void FakeImageWriterClient::Success() {
|
| - if (!success_callback_.is_null())
|
| - success_callback_.Run();
|
| -}
|
| +void FakeImageWriterClient::Success() { success_callback_.Run(); }
|
|
|
| void FakeImageWriterClient::Error(const std::string& message) {
|
| - if (!error_callback_.is_null())
|
| - error_callback_.Run(message);
|
| + error_callback_.Run(message);
|
| }
|
|
|
| -void FakeImageWriterClient::Cancel() {
|
| - if (!cancel_callback_.is_null())
|
| - cancel_callback_.Run();
|
| -}
|
| +void FakeImageWriterClient::Cancel() { cancel_callback_.Run(); }
|
|
|
| -ImageWriterTestUtils::ImageWriterTestUtils() {
|
| +scoped_refptr<FakeImageWriterClient> FakeImageWriterClient::Create() {
|
| + return scoped_refptr<FakeImageWriterClient>(new FakeImageWriterClient());
|
| }
|
| -ImageWriterTestUtils::~ImageWriterTestUtils() {
|
| -}
|
|
|
| -void ImageWriterTestUtils::SetUp() {
|
| - SetUp(false);
|
| -}
|
| +ImageWriterUnitTestBase::ImageWriterUnitTestBase()
|
| + : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {}
|
| +ImageWriterUnitTestBase::~ImageWriterUnitTestBase() {}
|
|
|
| -void ImageWriterTestUtils::SetUp(bool is_browser_test) {
|
| +void ImageWriterUnitTestBase::SetUp() {
|
| + testing::Test::SetUp();
|
| +
|
| ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
|
| ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(),
|
| &test_image_path_));
|
| @@ -163,20 +137,11 @@
|
| if (!chromeos::DBusThreadManager::IsInitialized()) {
|
| chromeos::FakeDBusThreadManager* fake_dbus_thread_manager =
|
| new chromeos::FakeDBusThreadManager;
|
| - fake_dbus_thread_manager->SetFakeClients();
|
| scoped_ptr<chromeos::ImageBurnerClient>
|
| image_burner_fake(new ImageWriterFakeImageBurnerClient());
|
| fake_dbus_thread_manager->SetImageBurnerClient(image_burner_fake.Pass());
|
| -
|
| - if (is_browser_test) {
|
| - chromeos::DBusThreadManager::SetInstanceForTesting(
|
| - fake_dbus_thread_manager);
|
| - } else {
|
| - chromeos::DBusThreadManager::InitializeForTesting(
|
| - fake_dbus_thread_manager);
|
| - }
|
| + chromeos::DBusThreadManager::InitializeForTesting(fake_dbus_thread_manager);
|
| }
|
| -
|
| FakeDiskMountManager* disk_manager = new FakeDiskMountManager();
|
| chromeos::disks::DiskMountManager::InitializeForTesting(disk_manager);
|
|
|
| @@ -197,53 +162,30 @@
|
| true,
|
| false);
|
| disk_manager->SetupDefaultReplies();
|
| -#else
|
| - client_ = new FakeImageWriterClient();
|
| - image_writer::Operation::SetUtilityClientForTesting(client_);
|
| #endif
|
| }
|
|
|
| -void ImageWriterTestUtils::TearDown() {
|
| +void ImageWriterUnitTestBase::TearDown() {
|
| #if defined(OS_CHROMEOS)
|
| - if (chromeos::DBusThreadManager::IsInitialized()) {
|
| - chromeos::DBusThreadManager::Shutdown();
|
| - }
|
| + chromeos::DBusThreadManager::Shutdown();
|
| chromeos::disks::DiskMountManager::Shutdown();
|
| -#else
|
| - image_writer::Operation::SetUtilityClientForTesting(NULL);
|
| #endif
|
| }
|
|
|
| -const base::FilePath& ImageWriterTestUtils::GetTempDir() {
|
| - return temp_dir_.path();
|
| -}
|
| -
|
| -const base::FilePath& ImageWriterTestUtils::GetImagePath() {
|
| - return test_image_path_;
|
| -}
|
| -
|
| -const base::FilePath& ImageWriterTestUtils::GetDevicePath() {
|
| - return test_device_path_;
|
| -}
|
| -
|
| -#if !defined(OS_CHROMEOS)
|
| -FakeImageWriterClient* ImageWriterTestUtils::GetUtilityClient() {
|
| - return client_.get();
|
| -}
|
| -#endif
|
| -
|
| -bool ImageWriterTestUtils::ImageWrittenToDevice() {
|
| +bool ImageWriterUnitTestBase::ImageWrittenToDevice(
|
| + const base::FilePath& image_path,
|
| + const base::FilePath& device_path) {
|
| scoped_ptr<char[]> image_buffer(new char[kTestFileSize]);
|
| scoped_ptr<char[]> device_buffer(new char[kTestFileSize]);
|
|
|
| int image_bytes_read =
|
| - ReadFile(test_image_path_, image_buffer.get(), kTestFileSize);
|
| + ReadFile(image_path, image_buffer.get(), kTestFileSize);
|
|
|
| if (image_bytes_read < 0)
|
| return false;
|
|
|
| int device_bytes_read =
|
| - ReadFile(test_device_path_, device_buffer.get(), kTestFileSize);
|
| + ReadFile(device_path, device_buffer.get(), kTestFileSize);
|
|
|
| if (image_bytes_read != device_bytes_read)
|
| return false;
|
| @@ -251,30 +193,14 @@
|
| return memcmp(image_buffer.get(), device_buffer.get(), image_bytes_read) == 0;
|
| }
|
|
|
| -bool ImageWriterTestUtils::FillFile(const base::FilePath& file,
|
| - const int pattern,
|
| - const int length) {
|
| +bool ImageWriterUnitTestBase::FillFile(const base::FilePath& file,
|
| + const int pattern,
|
| + const int length) {
|
| scoped_ptr<char[]> buffer(new char[length]);
|
| memset(buffer.get(), pattern, length);
|
|
|
| return base::WriteFile(file, buffer.get(), length) == length;
|
| }
|
|
|
| -ImageWriterUnitTestBase::ImageWriterUnitTestBase()
|
| - : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {
|
| -}
|
| -ImageWriterUnitTestBase::~ImageWriterUnitTestBase() {
|
| -}
|
| -
|
| -void ImageWriterUnitTestBase::SetUp() {
|
| - testing::Test::SetUp();
|
| - test_utils_.SetUp();
|
| -}
|
| -
|
| -void ImageWriterUnitTestBase::TearDown() {
|
| - testing::Test::TearDown();
|
| - test_utils_.TearDown();
|
| -}
|
| -
|
| } // namespace image_writer
|
| } // namespace extensions
|
|
|