Update FuturePlanner.php

pakomp · Nov 21, 2017 · fe3f442 · fe3f442
1 parent 3c04322
commit fe3f442
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/FuturePlanner.php b/src/FuturePlanner.php
@@ -35,11 +35,15 @@ class FuturePlanner
      *
      * @param ModelFuture $model
      */
-    public function __construct(ModelFuture $model)
+    public function __construct(ModelFuture $model, string $newFuture = null)
     {
         $this->model = $model;
         $this->futures = $this->model->uncommittedFutures();
-        $this->newFuture = new Future;
+        if ($newFuture == null) {
+            $this->newFuture = new Future;
+        } else {
+            $this->newFuture = new $newFuture;
+        }
     }
 
     /**