chrome/browser/chromeos/imageburner/burn_manager_unittest.cc

   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