Ensure milestone counts work with no datagitlab-ce-milestone-tab-badges

Commit originally written by @smcgivern

1 files changed, 17 insertions, 19 deletions

diff --git a/spec/helpers/milestones_helper_spec.rb b/spec/helpers/milestones_helper_spec.rb
index d65610723b0..28c2268f8d0 100644
--- a/spec/helpers/milestones_helper_spec.rb
+++ b/spec/helpers/milestones_helper_spec.rb
@@ -3,33 +3,31 @@ require 'spec_helper'
 describe MilestonesHelper do
   describe '#milestone_counts' do
     let(:project) { FactoryGirl.create(:project) }
-    let!(:milestone_1) { FactoryGirl.create(:active_milestone, project: project) }
-    let!(:milestone_2) { FactoryGirl.create(:active_milestone, project: project) }
-    let!(:milestone_3) { FactoryGirl.create(:closed_milestone, project: project) }
-
     let(:counts) { helper.milestone_counts(project.milestones) }
 
-    it 'returns a hash containing three items' do
-      expect(counts.length).to eq 3
-    end
+    context 'when there are milestones' do
+      let!(:milestone_1) { FactoryGirl.create(:active_milestone, project: project) }
+      let!(:milestone_2) { FactoryGirl.create(:active_milestone, project: project) }
+      let!(:milestone_3) { FactoryGirl.create(:closed_milestone, project: project) }
 
-    it 'returns a hash containing "opened" key' do
-      expect(counts.has_key?(:opened)).to eq true
+      it 'returns the correct counts' do
+        expect(counts).to eq(opened: 2, closed: 1, all: 3)
+      end
     end
 
-    it 'returns a hash containing "closed" key' do
-      expect(counts.has_key?(:closed)).to eq true
-    end
+    context 'when there are only milestones of one type' do
+      let!(:milestone_1) { FactoryGirl.create(:active_milestone, project: project) }
+      let!(:milestone_2) { FactoryGirl.create(:active_milestone, project: project) }
 
-    it 'returns a hash containing "all" key' do
-      expect(counts.has_key?(:all)).to eq true
+      it 'returns the correct counts' do
+        expect(counts).to eq(opened: 2, closed: 0, all: 2)
+      end
     end
 
-    it 'shows "all" object is the sum of "opened" and "closed" objects' do
-      puts counts.as_json
-      total = counts[:opened] + counts[:closed]
-      expect(counts[:all]).to eq total
+    context 'when there are no milestones' do
+      it 'returns the correct counts' do
+        expect(counts).to eq(opened: 0, closed: 0, all: 0)
+      end
     end
   end
 end
-  
\ No newline at end of file