1 files changed, 8 insertions, 21 deletions
diff --git a/tests/planner-mock.h b/tests/planner-mock.h
index 67e550c..53109cf 100644
--- a/tests/planner-mock.h
+++ b/tests/planner-mock.h
@@ -26,23 +26,6 @@ namespace unity {
 namespace indicator {
 namespace datetime {
 
-#if 0
-/**
- * \brief Planner which does nothing on its own.
- *        It requires its client must set its appointments property.
- */
-class MockPlanner: public Planner
-{
-public:
-    MockPlanner() =default;
-    virtual ~MockPlanner() =default;
-    core::Property<std::vector<Appointment>>& appointments() { return m_appointments; }
-
-private:
-    core::Property<std::vector<Appointment>> m_appointments;
-};
-#endif
-
 /**
  * \brief #RangePlanner which does nothing on its own.
  *        Its controller must set its appointments property.
@@ -50,15 +33,19 @@ private:
 class MockRangePlanner: public RangePlanner
 {
 public:
-    MockRangePlanner() =default;
+    MockRangePlanner():
+        m_range(std::pair<DateTime,DateTime>(DateTime::NowLocal(), DateTime::NowLocal()))
+    {
+    }
+
     ~MockRangePlanner() =default;
-    core::Property<std::vector<Appointment>>& appointments() { return m_appointments; }
 
-protected:
-    void rebuild_now(){}
+    core::Property<std::vector<Appointment>>& appointments() { return m_appointments; }
+    core::Property<std::pair<DateTime,DateTime>>& range() { return m_range; }
 
 private:
     core::Property<std::vector<Appointment>> m_appointments;
+    core::Property<std::pair<DateTime,DateTime>> m_range;
 };