15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <gmock/gmock.h> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <gtest/gtest.h> 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/chromeos/imageburner/burn_manager.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace chromeos { 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace imageburner { 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using ::testing::_; 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using ::testing::AnyNumber; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using ::testing::InSequence; 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const std::string kConfigFileWithNoHwidProperty = 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_name\n" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version\n" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=1000\n" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=http://image.bin.zip\n"; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const std::string kConfigFileWithNoNameProperty = 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version\n" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=2000\n" 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=http://some_image.bin.zip\n"; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const std::string kConfigFileWithNoNewLineAtEnd = 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_name\n" 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version\n" 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=1000\n" 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=some_hwid\n" 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=http://image.bin.zip"; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const std::string kSampleConfigFile = 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=aaa\n" 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=block_no_name\n" 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=aaa\n" 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_name1\n" 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version1\n" 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid11\n" 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid12\n" 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid13\n" 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=1000\n" 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=http://image1.bin.zip\n" 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "file=url\n" 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_name2\n" 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version2\n" 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid21\n" 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid22\n" 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid23\n" 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=1200\n" 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=http://image2.bin.zip\n" 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "file=file2" 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_name3\n" 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version3\n" 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid31\n" 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=3\n" 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=http://image3.bin.zip\n" 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "file=file3" 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_block_with_no_hwid\n" 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url=some_url\n" 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\n" 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=some_name_invalid_block\n" // Good line. 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version \n" // Trailing whitespace. 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid41=q\n" // Extra =. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid=hwid42\n" 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "hwid= \n" // Blank property value. 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "=\n" 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "filesize=\n" // Empty property value. 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "url\n" // No =. 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) " =something\n" 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "name=another_block_with_no_hwid\n" 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "version=version\n"; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(BurnManagerTest, ConfigFileTest) { 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<ConfigFile> cf(new ConfigFile()); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cf->empty()); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cf.reset(new ConfigFile("")); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cf->empty()); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cf.reset(new ConfigFile(kConfigFileWithNoNameProperty)); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cf->empty()); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cf.reset(new ConfigFile(kConfigFileWithNoHwidProperty)); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cf->empty()); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cf.reset(new ConfigFile(kConfigFileWithNoNewLineAtEnd)); 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(cf->empty()); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, cf->size()); 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("http://image.bin.zip", cf->GetProperty("url", "some_hwid")); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("some_name", cf->GetProperty("name", "some_hwid")); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cf.reset(new ConfigFile(kSampleConfigFile)); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(cf->empty()); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(4u, cf->size()); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("version", "block_no_name")); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("some_name1", cf->GetProperty("name", "hwid11")); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("version1", cf->GetProperty("version", "hwid12")); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("filesize", "hwid1_non_existent")); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("http://image1.bin.zip", cf->GetProperty("url", "hwid13")); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("hwid", "hwid11")); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("", "hwid12")); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("name", "")); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("some_name", "hwid11")); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("url", cf->GetProperty("file", "hwid11")); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("http://image2.bin.zip", cf->GetProperty("url", "hwid21")); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("some_name2", cf->GetProperty("name", "hwid23")); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("http://image3.bin.zip", cf->GetProperty("url", "hwid31")); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("some_name3", cf->GetProperty("name", "hwid31")); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("some_name_invalid_block", cf->GetProperty("name", "hwid42")); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(tbarzic): make this pass. 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // EXPECT_EQ("version", cf->GetProperty("version", "hwid42")); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("filesize", "hwid42")); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("url", "hwid42")); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(tbarzic): make this pass. 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // EXPECT_EQ("", cf->GetProperty(" ", "hwid42")); 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ("", cf->GetProperty("name", "hwid41")); 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockStateMachineObserver : public StateMachine::Observer { 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_METHOD1(OnBurnStateChanged, void(StateMachine::State)); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_METHOD1(OnError, void(int)); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(BurnManagerTest, StateMachineNormalWorkflow) { 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<StateMachine> state_machine(new StateMachine()); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(StateMachine::INITIAL, state_machine->state()); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockStateMachineObserver observer; 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->AddObserver(&observer); 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::DOWNLOADING)) 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .RetiresOnSaturation(); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::BURNING)) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .RetiresOnSaturation(); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::INITIAL)) 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .RetiresOnSaturation(); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->download_started()); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->download_finished()); 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->new_burn_posible()); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadStarted(); 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(StateMachine::DOWNLOADING, state_machine->state()); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_started()); 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->download_finished()); 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->new_burn_posible()); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadFinished(); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(tbarzic): make this pass. 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // EXPECT_EQ(StateMachine::INITIAL, state_machine->state()); 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_started()); 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_finished()); 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->new_burn_posible()); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnBurnStarted(); 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(StateMachine::BURNING, state_machine->state()); 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_started()); 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_finished()); 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->new_burn_posible()); 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnSuccess(); 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(StateMachine::INITIAL, state_machine->state()); 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_started()); 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_finished()); 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->new_burn_posible()); 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(BurnManagerTest, StateMachineError) { 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<StateMachine> state_machine(new StateMachine()); 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockStateMachineObserver observer; 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We don't want state change to INITIAL due to error to be reported to 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // observers. We use OnError for that. 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnBurnStateChanged(_)) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(AnyNumber()); 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnBurnStateChanged(StateMachine::INITIAL)) 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) InSequence error_calls; 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnError(1234)) 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnError(4321)) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer, OnError(0)) 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->AddObserver(&observer); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadStarted(); 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnError(1234); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If called before download finished, download flags should be reset. 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(state_machine->download_started()); 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(state_machine->state(), StateMachine::INITIAL); 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->new_burn_posible()); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadStarted(); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadFinished(); 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnError(4321); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If called after download finished, download flags should not be changed. 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_started()); 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->download_finished()); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(state_machine->state(), StateMachine::INITIAL); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->new_burn_posible()); 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnBurnStarted(); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnError(0); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(state_machine->state(), StateMachine::INITIAL); 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(state_machine->new_burn_posible()); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(BurnManagerTest, StateMachineObservers) { 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<StateMachine> state_machine(new StateMachine()); 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockStateMachineObserver observer1, observer2; 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer1, OnBurnStateChanged(_)) 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer2, OnBurnStateChanged(_)) 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer1, OnError(_)) 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer2, OnError(_)) 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadStarted(); 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnError(1); 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->AddObserver(&observer1); 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->AddObserver(&observer2); 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer1, OnBurnStateChanged(_)) 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer2, OnBurnStateChanged(_)) 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer1, OnError(_)) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer2, OnError(_)) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadStarted(); 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnError(1); 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->RemoveObserver(&observer1); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer1, OnBurnStateChanged(_)) 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer2, OnBurnStateChanged(_)) 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer1, OnError(_)) 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer2, OnError(_)) 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1); 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnDownloadStarted(); 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_machine->OnError(1); 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace imageburner 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace chromeos 287