Use existing methods for branch names: Ex use @repository.branch_names instead of @repository.heads.map(&:name)

1 files changed, 12 insertions, 1 deletions

diff --git a/spec/models/project_spec.rb b/spec/models/project_spec.rb
index e6585f78aa8..53388b933ee 100644
--- a/spec/models/project_spec.rb
+++ b/spec/models/project_spec.rb
@@ -233,7 +233,7 @@ describe Project do
 
     it "should be true for projects with external issues tracker if issues enabled" do
       ext_project.can_have_issues_tracker_id?.should be_true
-    end 
+    end
 
     it "should be false for projects with internal issue tracker if issues enabled" do
       project.can_have_issues_tracker_id?.should be_false
@@ -247,4 +247,15 @@ describe Project do
       ext_project.can_have_issues_tracker_id?.should be_false
     end
   end
+
+  describe :open_branches do
+    let(:project) { create(:project) }
+
+    before do
+      project.protected_branches.create(name: 'master')
+    end
+
+    it { project.open_branches.map(&:name).should include('bootstrap') }
+    it { project.open_branches.map(&:name).should_not include('master') }
+  end
 end