Review fixes.

1 files changed, 21 insertions, 0 deletions

diff --git a/traffic/traffic_info.hpp b/traffic/traffic_info.hpp
index a07e4d8cc4..99d5a8d974 100644
--- a/traffic/traffic_info.hpp
+++ b/traffic/traffic_info.hpp
@@ -6,6 +6,7 @@
 
 #include "std/cstdint.hpp"
 #include "std/map.hpp"
+#include "std/shared_ptr.hpp"
 #include "std/vector.hpp"
 
 namespace traffic
@@ -67,6 +68,8 @@ public:
 
   TrafficInfo(MwmSet::MwmId const & mwmId, int64_t currentDataVersion);
 
+  void SetColoringForTesting(Coloring & coloring) { m_coloring = coloring; }
+
   // Fetches the latest traffic data from the server and updates the coloring.
   // Construct the url by passing an MwmId.
   // *NOTE* This method must not be called on the UI thread.
@@ -90,4 +93,22 @@ private:
   Availability m_availability = Availability::Unknown;
   int64_t m_currentDataVersion = 0;
 };
+
+class RoutingObserver
+{
+public:
+  virtual ~RoutingObserver() = default;
+
+  virtual void OnTrafficEnabled(bool enable) = 0;
+  virtual void OnTrafficInfoAdded(traffic::TrafficInfo const & info) = 0;
+  virtual void OnTrafficInfoRemoved(MwmSet::MwmId const & mwmId) = 0;
+};
+
+class TrafficInfoGetter
+{
+public:
+  virtual ~TrafficInfoGetter() = default;
+
+  virtual shared_ptr<traffic::TrafficInfo>GetTrafficInfo(MwmSet::MwmId const & mwmId) const = 0;
+};
 }  // namespace traffic