mock_payload_state.h revision 4243291a6655972fe89d4747a0089834c9e47c20 
1// Copyright (c) 2012 The Chromium OS 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#ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_MOCK_PAYLOAD_STATE_H__
6#define CHROMEOS_PLATFORM_UPDATE_ENGINE_MOCK_PAYLOAD_STATE_H__
7
8#include "gmock/gmock.h"
9#include "update_engine/omaha_request_action.h"
10#include "update_engine/payload_state_interface.h"
11
12namespace chromeos_update_engine {
13
14class MockPayloadState: public PayloadStateInterface {
15 public:
16  bool Initialize(SystemState* system_state) {
17    return true;
18  }
19
20  // Significant methods.
21  MOCK_METHOD1(SetResponse, void(const OmahaResponse& response));
22  MOCK_METHOD0(DownloadComplete, void());
23  MOCK_METHOD1(DownloadProgress, void(size_t count));
24  MOCK_METHOD0(UpdateResumed, void());
25  MOCK_METHOD0(UpdateRestarted, void());
26  MOCK_METHOD0(UpdateSucceeded, void());
27  MOCK_METHOD1(UpdateFailed, void(ErrorCode error));
28  MOCK_METHOD0(ResetUpdateStatus, void());
29  MOCK_METHOD0(ShouldBackoffDownload, bool());
30  MOCK_METHOD0(UpdateEngineStarted, void());
31  MOCK_METHOD0(Rollback, void());
32  MOCK_METHOD1(ExpectRebootInNewVersion,
33               void(const std::string& target_version_uid));
34
35  // Getters.
36  MOCK_METHOD0(GetResponseSignature, std::string());
37  MOCK_METHOD0(GetPayloadAttemptNumber, int());
38  MOCK_METHOD0(GetFullPayloadAttemptNumber, int());
39  MOCK_METHOD0(GetCurrentUrl, std::string());
40  MOCK_METHOD0(GetUrlFailureCount, uint32_t());
41  MOCK_METHOD0(GetUrlSwitchCount, uint32_t());
42  MOCK_METHOD0(GetNumResponsesSeen, int());
43  MOCK_METHOD0(GetBackoffExpiryTime, base::Time());
44  MOCK_METHOD0(GetUpdateDuration, base::TimeDelta());
45  MOCK_METHOD0(GetUpdateDurationUptime, base::TimeDelta());
46  MOCK_METHOD1(GetCurrentBytesDownloaded, uint64_t(DownloadSource source));
47  MOCK_METHOD1(GetTotalBytesDownloaded, uint64_t(DownloadSource source));
48  MOCK_METHOD0(GetNumReboots, uint32_t());
49  MOCK_METHOD0(GetRollbackVersion, std::string());
50};
51
52}  // namespace chromeos_update_engine
53
54#endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_MOCK_PAYLOAD_STATE_H__
55