OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/run_loop.h" | |
6 #include "chrome/browser/extensions/api/image_writer_private/destroy_partitions_
operation.h" | |
7 #include "chrome/browser/extensions/api/image_writer_private/error_messages.h" | |
8 #include "chrome/browser/extensions/api/image_writer_private/test_utils.h" | |
9 | |
10 namespace extensions { | |
11 namespace image_writer { | |
12 | |
13 using testing::_; | |
14 using testing::AnyNumber; | |
15 | |
16 namespace { | |
17 | |
18 class ImageWriterDestroyPartitionsOperationTest | |
19 : public ImageWriterUnitTestBase { | |
20 }; | |
21 | |
22 // Tests that the DestroyPartitionsOperation can successfully zero the first | |
23 // kPartitionTableSize bytes of an image. | |
24 TEST_F(ImageWriterDestroyPartitionsOperationTest, DestroyPartitions) { | |
25 MockOperationManager manager; | |
26 base::RunLoop loop; | |
27 | |
28 scoped_refptr<DestroyPartitionsOperation> operation( | |
29 new DestroyPartitionsOperation(manager.AsWeakPtr(), | |
30 kDummyExtensionId, | |
31 test_device_path_.AsUTF8Unsafe())); | |
32 | |
33 #if defined(OS_LINUX) || defined(OS_CHROMEOS) | |
34 EXPECT_CALL(manager, OnProgress(kDummyExtensionId, _, _)).Times(0); | |
35 EXPECT_CALL(manager, OnProgress(kDummyExtensionId, | |
36 image_writer_api::STAGE_WRITE, | |
37 _)).Times(AnyNumber()); | |
38 EXPECT_CALL(manager, OnComplete(kDummyExtensionId)).Times(1); | |
39 EXPECT_CALL(manager, OnError(kDummyExtensionId, _, _, _)).Times(0); | |
40 #else | |
41 EXPECT_CALL(manager, OnProgress(kDummyExtensionId, _, _)).Times(0); | |
42 EXPECT_CALL(manager, OnComplete(kDummyExtensionId)).Times(0); | |
43 EXPECT_CALL(manager, OnError(kDummyExtensionId, | |
44 _, | |
45 _, | |
46 error::kUnsupportedOperation)).Times(1); | |
47 #endif | |
48 | |
49 operation->Start(); | |
50 | |
51 loop.RunUntilIdle(); | |
52 | |
53 #if defined(OS_LINUX) && !defined(OS_CHROMEOS) | |
54 scoped_ptr<char[]> image_data(new char[kPartitionTableSize]); | |
55 scoped_ptr<char[]> zeroes(new char[kPartitionTableSize]); | |
56 memset(zeroes.get(), 0, kPartitionTableSize); | |
57 ASSERT_EQ(kPartitionTableSize, base::ReadFile(test_device_path_, | |
58 image_data.get(), | |
59 kPartitionTableSize)); | |
60 EXPECT_EQ(0, memcmp(image_data.get(), zeroes.get(), kPartitionTableSize)); | |
61 #endif | |
62 } | |
63 | |
64 } // namespace | |
65 } // namespace image_writer | |
66 } // namespace extensions | |
OLD | NEW |