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 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
60 | 60 |
61 class ImageWriterOperationManagerTest | 61 class ImageWriterOperationManagerTest |
62 : public ImageWriterUnitTestBase { | 62 : public ImageWriterUnitTestBase { |
63 public: | 63 public: |
64 void StartCallback(bool success, const std::string& error) { | 64 void StartCallback(bool success, const std::string& error) { |
65 started_ = true; | 65 started_ = true; |
66 start_success_ = success; | 66 start_success_ = success; |
67 start_error_ = error; | 67 start_error_ = error; |
68 } | 68 } |
69 | 69 |
| 70 void CancelCallback(bool success, const std::string& error) { |
| 71 cancelled_ = true; |
| 72 cancel_success_ = true; |
| 73 cancel_error_ = error; |
| 74 } |
| 75 |
70 protected: | 76 protected: |
71 ImageWriterOperationManagerTest() | 77 ImageWriterOperationManagerTest() |
72 : started_(false), | 78 : started_(false), |
73 start_success_(false) { | 79 start_success_(false) { |
74 } | 80 } |
75 | 81 |
76 virtual void SetUp() OVERRIDE { | 82 virtual void SetUp() OVERRIDE { |
77 ImageWriterUnitTestBase::SetUp(); | 83 ImageWriterUnitTestBase::SetUp(); |
78 extension_system_ = static_cast<FakeExtensionSystem*>( | 84 extension_system_ = static_cast<FakeExtensionSystem*>( |
79 ExtensionSystemFactory::GetInstance()-> | 85 ExtensionSystemFactory::GetInstance()-> |
80 SetTestingFactoryAndUse(&test_profile_, &BuildFakeExtensionSystem)); | 86 SetTestingFactoryAndUse(&test_profile_, &BuildFakeExtensionSystem)); |
81 event_router_ = static_cast<FakeEventRouter*>( | 87 event_router_ = static_cast<FakeEventRouter*>( |
82 extension_system_->event_router()); | 88 extension_system_->event_router()); |
83 } | 89 } |
84 | 90 |
85 bool started_; | 91 bool started_; |
86 bool start_success_; | 92 bool start_success_; |
87 std::string start_error_; | 93 std::string start_error_; |
88 | 94 |
| 95 bool cancelled_; |
| 96 bool cancel_success_; |
| 97 std::string cancel_error_; |
| 98 |
89 TestingProfile test_profile_; | 99 TestingProfile test_profile_; |
90 FakeExtensionSystem* extension_system_; | 100 FakeExtensionSystem* extension_system_; |
91 FakeEventRouter* event_router_; | 101 FakeEventRouter* event_router_; |
92 | 102 |
93 #if defined(OS_CHROMEOS) | 103 #if defined(OS_CHROMEOS) |
94 chromeos::ScopedTestDeviceSettingsService test_device_settings_service_; | 104 chromeos::ScopedTestDeviceSettingsService test_device_settings_service_; |
95 chromeos::ScopedTestCrosSettings test_cros_settings_; | 105 chromeos::ScopedTestCrosSettings test_cros_settings_; |
96 chromeos::ScopedTestUserManager test_user_manager_; | 106 chromeos::ScopedTestUserManager test_user_manager_; |
97 #endif | 107 #endif |
98 }; | 108 }; |
99 | 109 |
100 TEST_F(ImageWriterOperationManagerTest, WriteFromFile) { | 110 TEST_F(ImageWriterOperationManagerTest, WriteFromFile) { |
101 OperationManager manager(&test_profile_); | 111 OperationManager manager(&test_profile_); |
102 | 112 |
103 manager.StartWriteFromFile( | 113 manager.StartWriteFromFile( |
104 kDummyExtensionId, | 114 kDummyExtensionId, |
105 test_image_path_, | 115 test_image_path_, |
106 test_device_path_.AsUTF8Unsafe(), | 116 test_device_path_.AsUTF8Unsafe(), |
107 base::Bind(&ImageWriterOperationManagerTest::StartCallback, | 117 base::Bind(&ImageWriterOperationManagerTest::StartCallback, |
108 base::Unretained(this))); | 118 base::Unretained(this))); |
109 | 119 |
110 EXPECT_TRUE(started_); | 120 EXPECT_TRUE(started_); |
111 EXPECT_TRUE(start_success_); | 121 EXPECT_TRUE(start_success_); |
112 EXPECT_EQ("", start_error_); | 122 EXPECT_EQ("", start_error_); |
113 | 123 |
| 124 manager.CancelWrite( |
| 125 kDummyExtensionId, |
| 126 base::Bind(&ImageWriterOperationManagerTest::CancelCallback, |
| 127 base::Unretained(this))); |
| 128 |
| 129 EXPECT_TRUE(cancelled_); |
| 130 EXPECT_TRUE(cancel_success_); |
| 131 EXPECT_EQ("", cancel_error_); |
| 132 |
114 base::RunLoop().RunUntilIdle(); | 133 base::RunLoop().RunUntilIdle(); |
115 } | 134 } |
116 | 135 |
117 TEST_F(ImageWriterOperationManagerTest, DestroyPartitions) { | 136 TEST_F(ImageWriterOperationManagerTest, DestroyPartitions) { |
118 OperationManager manager(&test_profile_); | 137 OperationManager manager(&test_profile_); |
119 | 138 |
120 manager.DestroyPartitions( | 139 manager.DestroyPartitions( |
121 kDummyExtensionId, | 140 kDummyExtensionId, |
122 test_device_path_.AsUTF8Unsafe(), | 141 test_device_path_.AsUTF8Unsafe(), |
123 base::Bind(&ImageWriterOperationManagerTest::StartCallback, | 142 base::Bind(&ImageWriterOperationManagerTest::StartCallback, |
124 base::Unretained(this))); | 143 base::Unretained(this))); |
125 | 144 |
126 EXPECT_TRUE(started_); | 145 EXPECT_TRUE(started_); |
127 EXPECT_TRUE(start_success_); | 146 EXPECT_TRUE(start_success_); |
128 EXPECT_EQ("", start_error_); | 147 EXPECT_EQ("", start_error_); |
129 | 148 |
| 149 manager.CancelWrite( |
| 150 kDummyExtensionId, |
| 151 base::Bind(&ImageWriterOperationManagerTest::CancelCallback, |
| 152 base::Unretained(this))); |
| 153 |
| 154 EXPECT_TRUE(cancelled_); |
| 155 EXPECT_TRUE(cancel_success_); |
| 156 EXPECT_EQ("", cancel_error_); |
| 157 |
130 base::RunLoop().RunUntilIdle(); | 158 base::RunLoop().RunUntilIdle(); |
131 } | 159 } |
132 | 160 |
133 } // namespace | 161 } // namespace |
134 } // namespace image_writer | 162 } // namespace image_writer |
135 } // namespace extensions | 163 } // namespace extensions |
OLD | NEW |