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