Index: update_attempter.h |
diff --git a/update_attempter.h b/update_attempter.h |
index fd9bc2a9b14a59dc25bc2abcf00c87f02c363424..6a81ef2da4eefa268f1467b5ad3c392f855752d6 100644 |
--- a/update_attempter.h |
+++ b/update_attempter.h |
@@ -12,6 +12,7 @@ |
#include <vector> |
#include <glib.h> |
+#include <gtest/gtest_prod.h> // for FRIEND_TEST |
#include "base/time.h" |
#include "update_engine/action_processor.h" |
@@ -104,6 +105,9 @@ class UpdateAttempter : public ActionProcessorDelegate, |
void BytesReceived(uint64_t bytes_received, uint64_t total); |
private: |
+ friend class UpdateAttempterTest; |
+ FRIEND_TEST(UpdateAttempterTest, UpdateTest); |
+ |
// Sets the status to the given status and notifies a status update |
// over dbus. |
void SetStatusAndNotify(UpdateStatus status); |
@@ -144,7 +148,7 @@ class UpdateAttempter : public ActionProcessorDelegate, |
base::TimeTicks last_notify_time_; |
std::vector<std::tr1::shared_ptr<AbstractAction> > actions_; |
- ActionProcessor processor_; |
+ scoped_ptr<ActionProcessor> processor_; |
// If non-null, this UpdateAttempter will send status updates over this |
// dbus service. |