OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/command_line.h" | 5 #include "base/command_line.h" |
6 #include "chrome/browser/chromeos/login/user_manager.h" | 6 #include "chrome/browser/chromeos/login/user_manager.h" |
7 #include "chrome/browser/extensions/api/image_writer_private/error_messages.h" | 7 #include "chrome/browser/extensions/api/image_writer_private/error_messages.h" |
8 #include "chrome/browser/extensions/api/image_writer_private/operation_manager.h
" | 8 #include "chrome/browser/extensions/api/image_writer_private/operation_manager.h
" |
9 #include "chrome/browser/extensions/api/image_writer_private/test_utils.h" | 9 #include "chrome/browser/extensions/api/image_writer_private/test_utils.h" |
10 #include "chrome/browser/extensions/extension_service.h" | 10 #include "chrome/browser/extensions/extension_service.h" |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
61 | 61 |
62 class ImageWriterOperationManagerTest | 62 class ImageWriterOperationManagerTest |
63 : public ImageWriterUnitTestBase { | 63 : public ImageWriterUnitTestBase { |
64 public: | 64 public: |
65 void StartCallback(bool success, const std::string& error) { | 65 void StartCallback(bool success, const std::string& error) { |
66 started_ = true; | 66 started_ = true; |
67 start_success_ = success; | 67 start_success_ = success; |
68 start_error_ = error; | 68 start_error_ = error; |
69 } | 69 } |
70 | 70 |
| 71 void CancelCallback(bool success, const std::string& error) { |
| 72 cancelled_ = true; |
| 73 cancel_success_ = true; |
| 74 cancel_error_ = error; |
| 75 } |
| 76 |
71 protected: | 77 protected: |
72 ImageWriterOperationManagerTest() | 78 ImageWriterOperationManagerTest() |
73 : started_(false), | 79 : started_(false), |
74 start_success_(false) { | 80 start_success_(false) { |
75 } | 81 } |
76 | 82 |
77 virtual void SetUp() OVERRIDE { | 83 virtual void SetUp() OVERRIDE { |
78 ImageWriterUnitTestBase::SetUp(); | 84 ImageWriterUnitTestBase::SetUp(); |
79 extension_system_ = static_cast<FakeExtensionSystem*>( | 85 extension_system_ = static_cast<FakeExtensionSystem*>( |
80 ExtensionSystemFactory::GetInstance()-> | 86 ExtensionSystemFactory::GetInstance()-> |
81 SetTestingFactoryAndUse(&test_profile_, &BuildFakeExtensionSystem)); | 87 SetTestingFactoryAndUse(&test_profile_, &BuildFakeExtensionSystem)); |
82 event_router_ = static_cast<FakeEventRouter*>( | 88 event_router_ = static_cast<FakeEventRouter*>( |
83 extension_system_->event_router()); | 89 extension_system_->event_router()); |
84 } | 90 } |
85 | 91 |
86 bool started_; | 92 bool started_; |
87 bool start_success_; | 93 bool start_success_; |
88 std::string start_error_; | 94 std::string start_error_; |
89 | 95 |
| 96 bool cancelled_; |
| 97 bool cancel_success_; |
| 98 std::string cancel_error_; |
| 99 |
90 TestingProfile test_profile_; | 100 TestingProfile test_profile_; |
91 FakeExtensionSystem* extension_system_; | 101 FakeExtensionSystem* extension_system_; |
92 FakeEventRouter* event_router_; | 102 FakeEventRouter* event_router_; |
93 | 103 |
94 #if defined(OS_CHROMEOS) | 104 #if defined(OS_CHROMEOS) |
95 chromeos::ScopedTestDeviceSettingsService test_device_settings_service_; | 105 chromeos::ScopedTestDeviceSettingsService test_device_settings_service_; |
96 chromeos::ScopedTestCrosSettings test_cros_settings_; | 106 chromeos::ScopedTestCrosSettings test_cros_settings_; |
97 chromeos::ScopedTestUserManager test_user_manager_; | 107 chromeos::ScopedTestUserManager test_user_manager_; |
98 #endif | 108 #endif |
99 }; | 109 }; |
100 | 110 |
101 TEST_F(ImageWriterOperationManagerTest, WriteFromFile) { | 111 TEST_F(ImageWriterOperationManagerTest, WriteFromFile) { |
102 OperationManager manager(&test_profile_); | 112 OperationManager manager(&test_profile_); |
103 | 113 |
104 manager.StartWriteFromFile( | 114 manager.StartWriteFromFile( |
105 kDummyExtensionId, | 115 kDummyExtensionId, |
106 test_image_path_, | 116 test_image_path_, |
107 test_device_path_.AsUTF8Unsafe(), | 117 test_device_path_.AsUTF8Unsafe(), |
108 base::Bind(&ImageWriterOperationManagerTest::StartCallback, | 118 base::Bind(&ImageWriterOperationManagerTest::StartCallback, |
109 base::Unretained(this))); | 119 base::Unretained(this))); |
110 | 120 |
111 EXPECT_TRUE(started_); | 121 EXPECT_TRUE(started_); |
112 EXPECT_TRUE(start_success_); | 122 EXPECT_TRUE(start_success_); |
113 EXPECT_EQ("", start_error_); | 123 EXPECT_EQ("", start_error_); |
114 | 124 |
| 125 manager.CancelWrite( |
| 126 kDummyExtensionId, |
| 127 base::Bind(&ImageWriterOperationManagerTest::CancelCallback, |
| 128 base::Unretained(this))); |
| 129 |
| 130 EXPECT_TRUE(cancelled_); |
| 131 EXPECT_TRUE(cancel_success_); |
| 132 EXPECT_EQ("", cancel_error_); |
| 133 |
115 base::RunLoop().RunUntilIdle(); | 134 base::RunLoop().RunUntilIdle(); |
116 } | 135 } |
117 | 136 |
118 TEST_F(ImageWriterOperationManagerTest, DestroyPartitions) { | 137 TEST_F(ImageWriterOperationManagerTest, DestroyPartitions) { |
119 OperationManager manager(&test_profile_); | 138 OperationManager manager(&test_profile_); |
120 | 139 |
121 manager.DestroyPartitions( | 140 manager.DestroyPartitions( |
122 kDummyExtensionId, | 141 kDummyExtensionId, |
123 test_device_path_.AsUTF8Unsafe(), | 142 test_device_path_.AsUTF8Unsafe(), |
124 base::Bind(&ImageWriterOperationManagerTest::StartCallback, | 143 base::Bind(&ImageWriterOperationManagerTest::StartCallback, |
125 base::Unretained(this))); | 144 base::Unretained(this))); |
126 | 145 |
127 EXPECT_TRUE(started_); | 146 EXPECT_TRUE(started_); |
128 EXPECT_TRUE(start_success_); | 147 EXPECT_TRUE(start_success_); |
129 EXPECT_EQ("", start_error_); | 148 EXPECT_EQ("", start_error_); |
130 | 149 |
| 150 manager.CancelWrite( |
| 151 kDummyExtensionId, |
| 152 base::Bind(&ImageWriterOperationManagerTest::CancelCallback, |
| 153 base::Unretained(this))); |
| 154 |
| 155 EXPECT_TRUE(cancelled_); |
| 156 EXPECT_TRUE(cancel_success_); |
| 157 EXPECT_EQ("", cancel_error_); |
| 158 |
131 base::RunLoop().RunUntilIdle(); | 159 base::RunLoop().RunUntilIdle(); |
132 } | 160 } |
133 | 161 |
134 } // namespace | 162 } // namespace |
135 } // namespace image_writer | 163 } // namespace image_writer |
136 } // namespace extensions | 164 } // namespace extensions |
OLD | NEW |