1 // Copyright (c) 2012 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 <gmock/gmock.h> 6 #include <gtest/gtest.h> 7 8 #include "base/memory/scoped_ptr.h" 9 #include "chrome/browser/chromeos/imageburner/burn_manager.h" 10 11 namespace chromeos { 12 namespace imageburner { 13 14 using ::testing::_; 15 using ::testing::AnyNumber; 16 using ::testing::InSequence; 17 18 const std::string kConfigFileWithNoHwidProperty = 19 "name=some_name\n" 20 "version=version\n" 21 "filesize=1000\n" 22 "url=http://image.bin.zip\n"; 23 24 const std::string kConfigFileWithNoNameProperty = 25 "version=version\n" 26 "filesize=2000\n" 27 "url=http://some_image.bin.zip\n"; 28 29 const std::string kConfigFileWithNoNewLineAtEnd = 30 "name=some_name\n" 31 "version=version\n" 32 "filesize=1000\n" 33 "hwid=some_hwid\n" 34 "url=http://image.bin.zip"; 35 36 const std::string kSampleConfigFile = 37 "version=aaa\n" 38 "hwid=block_no_name\n" 39 "url=aaa\n" 40 "\n" 41 "name=some_name1\n" 42 "version=version1\n" 43 "hwid=hwid11\n" 44 "hwid=hwid12\n" 45 "hwid=hwid13\n" 46 "\n" 47 "filesize=1000\n" 48 "url=http://image1.bin.zip\n" 49 "file=url\n" 50 "name=some_name2\n" 51 "version=version2\n" 52 "hwid=hwid21\n" 53 "hwid=hwid22\n" 54 "hwid=hwid23\n" 55 "\n" 56 "filesize=1200\n" 57 "url=http://image2.bin.zip\n" 58 "file=file2" 59 "\n" 60 "name=some_name3\n" 61 "version=version3\n" 62 "hwid=hwid31\n" 63 "\n" 64 "filesize=3\n" 65 "url=http://image3.bin.zip\n" 66 "file=file3" 67 "\n" 68 "name=some_block_with_no_hwid\n" 69 "url=some_url\n" 70 "\n" 71 "name=some_name_invalid_block\n" // Good line. 72 "version=version \n" // Trailing whitespace. 73 "hwid=hwid41=q\n" // Extra =. 74 "hwid=hwid42\n" 75 "hwid= \n" // Blank property value. 76 "=\n" 77 "filesize=\n" // Empty property value. 78 "url\n" // No =. 79 " =something\n" 80 "name=another_block_with_no_hwid\n" 81 "version=version\n"; 82 83 TEST(BurnManagerTest, ConfigFileTest) { 84 scoped_ptr<ConfigFile> cf(new ConfigFile()); 85 EXPECT_TRUE(cf->empty()); 86 87 cf.reset(new ConfigFile("")); 88 EXPECT_TRUE(cf->empty()); 89 90 cf.reset(new ConfigFile(kConfigFileWithNoNameProperty)); 91 EXPECT_TRUE(cf->empty()); 92 93 cf.reset(new ConfigFile(kConfigFileWithNoHwidProperty)); 94 EXPECT_TRUE(cf->empty()); 95 96 cf.reset(new ConfigFile(kConfigFileWithNoNewLineAtEnd)); 97 EXPECT_FALSE(cf->empty()); 98 EXPECT_EQ(1u, cf->size()); 99 EXPECT_EQ("http://image.bin.zip", cf->GetProperty("url", "some_hwid")); 100 EXPECT_EQ("some_name", cf->GetProperty("name", "some_hwid")); 101 102 cf.reset(new ConfigFile(kSampleConfigFile)); 103 EXPECT_FALSE(cf->empty()); 104 105 EXPECT_EQ(4u, cf->size()); 106 107 EXPECT_EQ("", cf->GetProperty("version", "block_no_name")); 108 109 EXPECT_EQ("some_name1", cf->GetProperty("name", "hwid11")); 110 EXPECT_EQ("version1", cf->GetProperty("version", "hwid12")); 111 EXPECT_EQ("", cf->GetProperty("filesize", "hwid1_non_existent")); 112 EXPECT_EQ("http://image1.bin.zip", cf->GetProperty("url", "hwid13")); 113 EXPECT_EQ("", cf->GetProperty("hwid", "hwid11")); 114 EXPECT_EQ("", cf->GetProperty("", "hwid12")); 115 EXPECT_EQ("", cf->GetProperty("name", "")); 116 EXPECT_EQ("", cf->GetProperty("some_name", "hwid11")); 117 EXPECT_EQ("url", cf->GetProperty("file", "hwid11")); 118 119 EXPECT_EQ("http://image2.bin.zip", cf->GetProperty("url", "hwid21")); 120 EXPECT_EQ("some_name2", cf->GetProperty("name", "hwid23")); 121 122 EXPECT_EQ("http://image3.bin.zip", cf->GetProperty("url", "hwid31")); 123 EXPECT_EQ("some_name3", cf->GetProperty("name", "hwid31")); 124 125 EXPECT_EQ("some_name_invalid_block", cf->GetProperty("name", "hwid42")); 126 // TODO(tbarzic): make this pass. 127 // EXPECT_EQ("version", cf->GetProperty("version", "hwid42")); 128 EXPECT_EQ("", cf->GetProperty("filesize", "hwid42")); 129 EXPECT_EQ("", cf->GetProperty("url", "hwid42")); 130 // TODO(tbarzic): make this pass. 131 // EXPECT_EQ("", cf->GetProperty(" ", "hwid42")); 132 EXPECT_EQ("", cf->GetProperty("name", "hwid41")); 133 } 134 135 class MockStateMachineObserver : public StateMachine::Observer { 136 public: 137 MOCK_METHOD1(OnBurnStateChanged, void(StateMachine::State)); 138 MOCK_METHOD1(OnError, void(int)); 139 }; 140 141 TEST(BurnManagerTest, StateMachineNormalWorkflow) { 142 scoped_ptr<StateMachine> state_machine(new StateMachine()); 143 EXPECT_EQ(StateMachine::INITIAL, state_machine->state()); 144 145 MockStateMachineObserver observer; 146 state_machine->AddObserver(&observer); 147 EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::DOWNLOADING)) 148 .Times(1) 149 .RetiresOnSaturation(); 150 151 EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::BURNING)) 152 .Times(1) 153 .RetiresOnSaturation(); 154 155 EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::INITIAL)) 156 .Times(1) 157 .RetiresOnSaturation(); 158 159 EXPECT_FALSE(state_machine->download_started()); 160 EXPECT_FALSE(state_machine->download_finished()); 161 EXPECT_TRUE(state_machine->new_burn_posible()); 162 163 state_machine->OnDownloadStarted(); 164 165 EXPECT_EQ(StateMachine::DOWNLOADING, state_machine->state()); 166 EXPECT_TRUE(state_machine->download_started()); 167 EXPECT_FALSE(state_machine->download_finished()); 168 EXPECT_FALSE(state_machine->new_burn_posible()); 169 170 state_machine->OnDownloadFinished(); 171 172 // TODO(tbarzic): make this pass. 173 // EXPECT_EQ(StateMachine::INITIAL, state_machine->state()); 174 EXPECT_TRUE(state_machine->download_started()); 175 EXPECT_TRUE(state_machine->download_finished()); 176 EXPECT_FALSE(state_machine->new_burn_posible()); 177 178 state_machine->OnBurnStarted(); 179 180 EXPECT_EQ(StateMachine::BURNING, state_machine->state()); 181 EXPECT_TRUE(state_machine->download_started()); 182 EXPECT_TRUE(state_machine->download_finished()); 183 EXPECT_FALSE(state_machine->new_burn_posible()); 184 185 state_machine->OnSuccess(); 186 187 EXPECT_EQ(StateMachine::INITIAL, state_machine->state()); 188 EXPECT_TRUE(state_machine->download_started()); 189 EXPECT_TRUE(state_machine->download_finished()); 190 EXPECT_TRUE(state_machine->new_burn_posible()); 191 } 192 193 TEST(BurnManagerTest, StateMachineError) { 194 scoped_ptr<StateMachine> state_machine(new StateMachine()); 195 196 MockStateMachineObserver observer; 197 // We don't want state change to INITIAL due to error to be reported to 198 // observers. We use OnError for that. 199 EXPECT_CALL(observer, OnBurnStateChanged(_)) 200 .Times(AnyNumber()); 201 EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::INITIAL)) 202 .Times(0); 203 { 204 InSequence error_calls; 205 EXPECT_CALL(observer, OnError(1234)) 206 .Times(1); 207 EXPECT_CALL(observer, OnError(4321)) 208 .Times(1); 209 EXPECT_CALL(observer, OnError(0)) 210 .Times(1); 211 } 212 state_machine->AddObserver(&observer); 213 214 state_machine->OnDownloadStarted(); 215 216 state_machine->OnError(1234); 217 218 // If called before download finished, download flags should be reset. 219 EXPECT_FALSE(state_machine->download_started()); 220 EXPECT_EQ(state_machine->state(), StateMachine::INITIAL); 221 EXPECT_TRUE(state_machine->new_burn_posible()); 222 223 state_machine->OnDownloadStarted(); 224 state_machine->OnDownloadFinished(); 225 226 state_machine->OnError(4321); 227 228 // If called after download finished, download flags should not be changed. 229 EXPECT_TRUE(state_machine->download_started()); 230 EXPECT_TRUE(state_machine->download_finished()); 231 EXPECT_EQ(state_machine->state(), StateMachine::INITIAL); 232 EXPECT_TRUE(state_machine->new_burn_posible()); 233 234 state_machine->OnBurnStarted(); 235 state_machine->OnError(0); 236 237 EXPECT_EQ(state_machine->state(), StateMachine::INITIAL); 238 EXPECT_TRUE(state_machine->new_burn_posible()); 239 } 240 241 TEST(BurnManagerTest, StateMachineObservers) { 242 scoped_ptr<StateMachine> state_machine(new StateMachine()); 243 244 MockStateMachineObserver observer1, observer2; 245 246 EXPECT_CALL(observer1, OnBurnStateChanged(_)) 247 .Times(0); 248 EXPECT_CALL(observer2, OnBurnStateChanged(_)) 249 .Times(0); 250 EXPECT_CALL(observer1, OnError(_)) 251 .Times(0); 252 EXPECT_CALL(observer2, OnError(_)) 253 .Times(0); 254 255 state_machine->OnDownloadStarted(); 256 state_machine->OnError(1); 257 258 state_machine->AddObserver(&observer1); 259 state_machine->AddObserver(&observer2); 260 EXPECT_CALL(observer1, OnBurnStateChanged(_)) 261 .Times(1); 262 EXPECT_CALL(observer2, OnBurnStateChanged(_)) 263 .Times(1); 264 EXPECT_CALL(observer1, OnError(_)) 265 .Times(1); 266 EXPECT_CALL(observer2, OnError(_)) 267 .Times(1); 268 269 state_machine->OnDownloadStarted(); 270 state_machine->OnError(1); 271 272 state_machine->RemoveObserver(&observer1); 273 EXPECT_CALL(observer1, OnBurnStateChanged(_)) 274 .Times(0); 275 EXPECT_CALL(observer2, OnBurnStateChanged(_)) 276 .Times(1); 277 EXPECT_CALL(observer1, OnError(_)) 278 .Times(0); 279 EXPECT_CALL(observer2, OnError(_)) 280 .Times(1); 281 state_machine->OnDownloadStarted(); 282 state_machine->OnError(1); 283 } 284 285 } // namespace imageburner 286 } // namespace chromeos 287