commit dcba8098a905018e6fe2e8e4e3b121a255f0d99e
author David Zeuthen <zeuthen@chromium.org> Tue Aug 06 12:16:35 2013 -0700
committer chrome-internal-fetch <chrome-internal-fetch@google.com> Tue Aug 27 23:20:12 2013 +0000
tree 1b335eb36e37643ccd82ae9614ee0fa9fc72c04c
parent 8a3e88b436f9fb8bcafd439aa405f71b900d8984

p2p: Add accounting for p2p usage to PayloadStateInterface

This machinery is needed in order to stop using p2p if the device has
failed to update itself using p2p in a timely manner.

BUG=chromium:260426
TEST=New unit tests + unit tests pass
Change-Id: I9f33309368f8cd5399b9d67c9072a1f78385abc0
Reviewed-on: https://chromium-review.googlesource.com/64827
Reviewed-by: David Zeuthen <zeuthen@chromium.org>
Commit-Queue: David Zeuthen <zeuthen@chromium.org>
Tested-by: David Zeuthen <zeuthen@chromium.org>

mock_payload_state.h

diff --git a/mock_payload_state.h b/mock_payload_state.h
index 0415a80..2e8536a 100644
--- a/mock_payload_state.h
+++ b/mock_payload_state.h
@@ -31,6 +31,8 @@
   MOCK_METHOD0(Rollback, void());
   MOCK_METHOD1(ExpectRebootInNewVersion,
                void(const std::string& target_version_uid));
+  MOCK_METHOD0(GetP2PNumAttempts, int());
+  MOCK_METHOD0(GetP2PFirstAttemptTimestamp, base::Time());
 
   // Getters.
   MOCK_METHOD0(GetResponseSignature, std::string());
@@ -47,6 +49,8 @@
   MOCK_METHOD1(GetTotalBytesDownloaded, uint64_t(DownloadSource source));
   MOCK_METHOD0(GetNumReboots, uint32_t());
   MOCK_METHOD0(GetRollbackVersion, std::string());
+  MOCK_METHOD0(P2PNewAttempt, void());
+  MOCK_METHOD0(P2PAttemptAllowed, bool());
 };
 
 }  // namespace chromeos_update_engine