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 |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1e1e48ed5ee0867d16510775416c187dfb8c20fa |
--- /dev/null |
+++ b/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation_unittest.cc |
@@ -0,0 +1,66 @@ |
+// Copyright 2013 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "base/run_loop.h" |
+#include "chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation.h" |
+#include "chrome/browser/extensions/api/image_writer_private/error_messages.h" |
+#include "chrome/browser/extensions/api/image_writer_private/test_utils.h" |
+ |
+namespace extensions { |
+namespace image_writer { |
+ |
+using testing::_; |
+using testing::AnyNumber; |
+ |
+namespace { |
+ |
+class ImageWriterDestroyPartitionsOperationTest |
+ : public ImageWriterUnitTestBase { |
+}; |
+ |
+// Tests that the DestroyPartitionsOperation can successfully zero the first |
+// kPartitionTableSize bytes of an image. |
+TEST_F(ImageWriterDestroyPartitionsOperationTest, DestroyPartitions) { |
+ MockOperationManager manager; |
+ base::RunLoop loop; |
+ |
+ scoped_refptr<DestroyPartitionsOperation> operation( |
+ new DestroyPartitionsOperation(manager.AsWeakPtr(), |
+ kDummyExtensionId, |
+ 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, OnComplete(kDummyExtensionId)).Times(1); |
+ EXPECT_CALL(manager, OnError(kDummyExtensionId, _, _, _)).Times(0); |
+#else |
+ EXPECT_CALL(manager, OnProgress(kDummyExtensionId, _, _)).Times(0); |
+ EXPECT_CALL(manager, OnComplete(kDummyExtensionId)).Times(0); |
+ EXPECT_CALL(manager, OnError(kDummyExtensionId, |
+ _, |
+ _, |
+ error::kUnsupportedOperation)).Times(1); |
+#endif |
+ |
+ operation->Start(); |
+ |
+ loop.RunUntilIdle(); |
+ |
+#if defined(OS_LINUX) && !defined(OS_CHROMEOS) |
+ scoped_ptr<char[]> image_data(new char[kPartitionTableSize]); |
+ scoped_ptr<char[]> zeroes(new char[kPartitionTableSize]); |
+ memset(zeroes.get(), 0, kPartitionTableSize); |
+ ASSERT_EQ(kPartitionTableSize, base::ReadFile(test_device_path_, |
+ image_data.get(), |
+ kPartitionTableSize)); |
+ EXPECT_EQ(0, memcmp(image_data.get(), zeroes.get(), kPartitionTableSize)); |
+#endif |
+} |
+ |
+} // namespace |
+} // namespace image_writer |
+} // namespace extensions |