Index: chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation_unittest.cc |
diff --git a/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation_unittest.cc b/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation_unittest.cc |
index af09c427f621484858a7a1ce677fd10305849602..9363a1ad7bc0c4813ed10b3d8d44dc0003077b54 100644 |
--- a/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation_unittest.cc |
+++ b/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation_unittest.cc |
@@ -22,16 +22,12 @@ class ImageWriterDestroyPartitionsOperationTest |
TEST_F(ImageWriterDestroyPartitionsOperationTest, EndToEnd) { |
TestingProfile profile; |
MockOperationManager manager(&profile); |
- scoped_refptr<FakeImageWriterClient> client = FakeImageWriterClient::Create(); |
scoped_refptr<DestroyPartitionsOperation> operation( |
- new DestroyPartitionsOperation(manager.AsWeakPtr(), |
- kDummyExtensionId, |
- test_device_path_.AsUTF8Unsafe())); |
- |
-#if !defined(OS_CHROMEOS) |
- operation->SetUtilityClientForTesting(client); |
-#endif |
+ new DestroyPartitionsOperation( |
+ manager.AsWeakPtr(), |
+ kDummyExtensionId, |
+ test_utils_.GetDevicePath().AsUTF8Unsafe())); |
EXPECT_CALL( |
manager, |
@@ -54,10 +50,10 @@ TEST_F(ImageWriterDestroyPartitionsOperationTest, EndToEnd) { |
base::RunLoop().RunUntilIdle(); |
#if !defined(OS_CHROMEOS) |
- client->Progress(0); |
- client->Progress(50); |
- client->Progress(100); |
- client->Success(); |
+ test_utils_.GetUtilityClient()->Progress(0); |
+ test_utils_.GetUtilityClient()->Progress(50); |
+ test_utils_.GetUtilityClient()->Progress(100); |
+ test_utils_.GetUtilityClient()->Success(); |
base::RunLoop().RunUntilIdle(); |
#endif |