commit 526cb58ddecbbc1ba5d2e3faf068d26d0661d464
author David Zeuthen <zeuthen@chromium.org> Tue Aug 06 12:26:18 2013 -0700
committer chrome-internal-fetch <chrome-internal-fetch@google.com> Tue Aug 27 23:20:15 2013 +0000
tree 54900433957d23d4eeafe40e7dfbaf3d6ccf0662
parent dcba8098a905018e6fe2e8e4e3b121a255f0d99e

p2p: Make P2PManager available from the SystemState singleton

BUG=chromium:260426
TEST=Unit tests pass
Change-Id: Iadaa531d631c8d7268f35d512ff33f8d53e1cc5e
Reviewed-on: https://chromium-review.googlesource.com/64828
Reviewed-by: David Zeuthen <zeuthen@chromium.org>
Commit-Queue: David Zeuthen <zeuthen@chromium.org>
Tested-by: David Zeuthen <zeuthen@chromium.org>

mock_system_state.h

diff --git a/mock_system_state.h b/mock_system_state.h
index a666a66..de0caef 100644
--- a/mock_system_state.h
+++ b/mock_system_state.h
@@ -12,6 +12,7 @@
 
 #include "update_engine/mock_dbus_interface.h"
 #include "update_engine/mock_gpio_handler.h"
+#include "update_engine/mock_p2p_manager.h"
 #include "update_engine/mock_payload_state.h"
 #include "update_engine/clock.h"
 #include "update_engine/hardware.h"
@@ -62,7 +63,7 @@
   }
 
   inline virtual PayloadStateInterface* payload_state() {
-    return &mock_payload_state_;
+    return payload_state_;
   }
 
   inline virtual GpioHandler* gpio_handler() const {
@@ -75,6 +76,10 @@
     return request_params_;
   }
 
+  inline virtual P2PManager* p2p_manager() {
+    return p2p_manager_;
+  }
+
   // MockSystemState-specific public method.
   inline void set_connection_manager(ConnectionManager* connection_manager) {
     connection_manager_ = connection_manager;
@@ -116,11 +121,20 @@
     request_params_ = params;
   }
 
+  inline void set_p2p_manager(P2PManager *p2p_manager) {
+    p2p_manager_ = p2p_manager;
+  }
+
+  inline void set_payload_state(PayloadStateInterface *payload_state) {
+    payload_state_ = payload_state;
+  }
+
  private:
   // These are Mock objects or objects we own.
   testing::NiceMock<MetricsLibraryMock> mock_metrics_lib_;
   testing::NiceMock<PrefsMock> mock_prefs_;
   testing::NiceMock<PrefsMock> mock_powerwash_safe_prefs_;
+  testing::NiceMock<MockP2PManager> mock_p2p_manager_;
   testing::NiceMock<MockPayloadState> mock_payload_state_;
   testing::NiceMock<MockGpioHandler>* mock_gpio_handler_;
   testing::NiceMock<UpdateAttempterMock>* mock_update_attempter_;
@@ -138,6 +152,8 @@
   PrefsInterface* powerwash_safe_prefs_;
   ConnectionManager* connection_manager_;
   OmahaRequestParams* request_params_;
+  P2PManager *p2p_manager_;
+  PayloadStateInterface *payload_state_;
 };
 
 } // namespeace chromeos_update_engine