Fix Repository role spec

1 files changed, 6 insertions, 6 deletions

diff --git a/spec/roles/repository_spec.rb b/spec/roles/repository_spec.rb
index 0fda57a3e27..c4e0e9dec81 100644
--- a/spec/roles/repository_spec.rb
+++ b/spec/roles/repository_spec.rb
@@ -21,27 +21,27 @@ describe Project, "Repository" do
   end
 
   describe "#discover_default_branch" do
-    let(:master) { double(name: 'master') }
-    let(:stable) { double(name: 'stable') }
+    let(:master) { 'master' }
+    let(:stable) { 'stable' }
 
     it "returns 'master' when master exists" do
-      project.should_receive(:heads).and_return([stable, master])
+      project.should_receive(:branches).at_least(:once).and_return([stable, master])
       project.discover_default_branch.should == 'master'
     end
 
     it "returns non-master when master exists but default branch is set to something else" do
       project.default_branch = 'stable'
-      project.should_receive(:heads).and_return([stable, master])
+      project.should_receive(:branches).at_least(:once).and_return([stable, master])
       project.discover_default_branch.should == 'stable'
     end
 
     it "returns a non-master branch when only one exists" do
-      project.should_receive(:heads).and_return([stable])
+      project.should_receive(:branches).at_least(:once).and_return([stable])
       project.discover_default_branch.should == 'stable'
     end
 
     it "returns nil when no branch exists" do
-      project.should_receive(:heads).and_return([])
+      project.should_receive(:branches).at_least(:once).and_return([])
       project.discover_default_branch.should be_nil
     end
   end