Show chart only until current date when milestone due date is bigger

app/models/burndown.rb

@@ -2,11 +2,12 @@ class Burndown
   attr_accessor :start_date, :end_date, :issues_count, :issues_weight
 
   def initialize(milestone)
-    @milestone     = milestone
-    @start_date    = @milestone.start_date
-    @end_date      = @milestone.due_date
+    @milestone = milestone
+    @start_date = @milestone.start_date
+    @end_date = @milestone.due_date
+    @end_date = Date.today if @end_date.present? && @end_date > Date.today
 
-    issues         = @milestone.issues
+    issues = @milestone.issues
 
     @issues_count  = issues.count
     @issues_weight = issues.sum(:weight)

spec/models/burndown_spec.rb

@@ -50,6 +50,12 @@ describe Burndown, models: true do
     expect(subject).to eq([])
   end
 
+  it "it counts until today if milestone due date > Date.today" do
+    Timecop.travel(milestone.due_date - 1.day)
+
+    expect(subject.last[0]).to eq(Time.now.strftime("%Y-%m-%d"))
+  end
+
   # Creates, closes and reopens issues only for odd days numbers
   def build_sample
     milestone.start_date.upto(milestone.due_date) do |date|