Index: trunk/src/chrome/browser/extensions/api/image_writer_private/write_from_file_operation_unittest.cc |
=================================================================== |
--- trunk/src/chrome/browser/extensions/api/image_writer_private/write_from_file_operation_unittest.cc (revision 282139) |
+++ trunk/src/chrome/browser/extensions/api/image_writer_private/write_from_file_operation_unittest.cc (working copy) |
@@ -27,10 +27,10 @@ |
scoped_refptr<WriteFromFileOperation> op = |
new WriteFromFileOperation(manager_.AsWeakPtr(), |
kDummyExtensionId, |
- test_utils_.GetImagePath(), |
- test_utils_.GetDevicePath().AsUTF8Unsafe()); |
+ test_image_path_, |
+ test_device_path_.AsUTF8Unsafe()); |
- base::DeleteFile(test_utils_.GetImagePath(), false); |
+ base::DeleteFile(test_image_path_, false); |
EXPECT_CALL(manager_, OnProgress(kDummyExtensionId, _, _)).Times(0); |
EXPECT_CALL(manager_, OnComplete(kDummyExtensionId)).Times(0); |
@@ -50,8 +50,13 @@ |
scoped_refptr<WriteFromFileOperation> op = |
new WriteFromFileOperation(manager_.AsWeakPtr(), |
kDummyExtensionId, |
- test_utils_.GetImagePath(), |
- test_utils_.GetDevicePath().AsUTF8Unsafe()); |
+ test_image_path_, |
+ test_device_path_.AsUTF8Unsafe()); |
+#if !defined(OS_CHROMEOS) |
+ scoped_refptr<FakeImageWriterClient> client = FakeImageWriterClient::Create(); |
+ op->SetUtilityClientForTesting(client); |
+#endif |
+ |
EXPECT_CALL(manager_, |
OnProgress(kDummyExtensionId, image_writer_api::STAGE_WRITE, _)) |
.Times(AnyNumber()); |
@@ -85,15 +90,15 @@ |
base::RunLoop().RunUntilIdle(); |
#if !defined(OS_CHROMEOS) |
- test_utils_.GetUtilityClient()->Progress(0); |
- test_utils_.GetUtilityClient()->Progress(50); |
- test_utils_.GetUtilityClient()->Progress(100); |
- test_utils_.GetUtilityClient()->Success(); |
+ client->Progress(0); |
+ client->Progress(50); |
+ client->Progress(100); |
+ client->Success(); |
base::RunLoop().RunUntilIdle(); |
- test_utils_.GetUtilityClient()->Progress(0); |
- test_utils_.GetUtilityClient()->Progress(50); |
- test_utils_.GetUtilityClient()->Progress(100); |
- test_utils_.GetUtilityClient()->Success(); |
+ client->Progress(0); |
+ client->Progress(50); |
+ client->Progress(100); |
+ client->Success(); |
base::RunLoop().RunUntilIdle(); |
#endif |
} |