remove mock for current_branch

thegarage · Sep 19, 2014 · 5abe30f · 5abe30f
1 parent 0886fc5
commit 5abe30f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/spec/thegarage/gitx/cli/integrate_command_spec.rb b/spec/thegarage/gitx/cli/integrate_command_spec.rb
@@ -10,10 +10,12 @@
     }
   end
   let(:cli) { Thegarage::Gitx::Cli::IntegrateCommand.new(args, options, config) }
-  let(:branch) { double('fake branch', name: 'feature-branch') }
+  let(:current_branch) { double('fake branch', name: 'feature-branch', head?: true) }
+  let(:repo) { cli.send(:repo) }
+  let(:branches) { [current_branch] }
 
   before do
-    allow(cli).to receive(:current_branch).and_return(branch)
+    allow(repo).to receive(:branches).and_return(branches)
   end
 
   describe '#integrate' do