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 2b0fb93d9808582a14d86a3f3d62d41b81b8e56c..68a669aaad9a6ccf5856a346680ec9a337bc8371 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 |
@@ -12,6 +12,7 @@ namespace image_writer { |
using testing::_; |
using testing::AnyNumber; |
+using testing::AtLeast; |
namespace { |
@@ -21,7 +22,7 @@ class ImageWriterDestroyPartitionsOperationTest |
// Tests that the DestroyPartitionsOperation can successfully zero the first |
// kPartitionTableSize bytes of an image. |
-TEST_F(ImageWriterDestroyPartitionsOperationTest, DestroyPartitions) { |
+TEST_F(ImageWriterDestroyPartitionsOperationTest, DestroyPartitionsEndToEnd) { |
MockOperationManager manager; |
base::RunLoop loop; |
@@ -31,10 +32,15 @@ TEST_F(ImageWriterDestroyPartitionsOperationTest, DestroyPartitions) { |
test_device_path_.AsUTF8Unsafe())); |
#if defined(OS_LINUX) || defined(OS_CHROMEOS) |
- EXPECT_CALL(manager, OnProgress(kDummyExtensionId, _, _)).Times(0); |
EXPECT_CALL(manager, OnProgress(kDummyExtensionId, |
image_writer_api::STAGE_WRITE, |
_)).Times(AnyNumber()); |
+ EXPECT_CALL(manager, |
+ OnProgress(kDummyExtensionId, image_writer_api::STAGE_WRITE, 0)) |
+ .Times(AtLeast(1)); |
+ EXPECT_CALL(manager, |
+ OnProgress(kDummyExtensionId, image_writer_api::STAGE_WRITE, 100)) |
+ .Times(AtLeast(1)); |
EXPECT_CALL(manager, OnComplete(kDummyExtensionId)).Times(1); |
EXPECT_CALL(manager, OnError(kDummyExtensionId, _, _, _)).Times(0); |
#else |