pivotal-integration 1.6.0.1

data/spec/git-pivotal-tracker-integration/util/git_spec.rb

@@ -0,0 +1,235 @@
+# Git Pivotal Tracker Integration
+# Copyright (c) 2013 the original author or authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'spec_helper'
+require 'git-pivotal-tracker-integration/util/git'
+require 'git-pivotal-tracker-integration/util/shell'
+
+describe PivotalIntegration::Util::Git do
+
+  before do
+    $stdout = StringIO.new
+    $stderr = StringIO.new
+  end
+
+  it 'should return the current branch name' do
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git branch').and_return("   master\n * dev_branch")
+
+    current_branch = PivotalIntegration::Util::Git.branch_name
+
+    expect(current_branch).to eq('dev_branch')
+  end
+
+  it 'should return the repository root' do
+    Dir.mktmpdir do |root|
+      child_directory = File.expand_path 'child', root
+      Dir.mkdir child_directory
+
+      git_directory = File.expand_path '.git', root
+      Dir.mkdir git_directory
+
+      Dir.should_receive(:pwd).and_return(child_directory)
+
+      repository_root = PivotalIntegration::Util::Git.repository_root
+
+      expect(repository_root).to eq(root)
+    end
+  end
+
+  it 'should raise an error there is no repository root' do
+    Dir.mktmpdir do |root|
+      child_directory = File.expand_path 'child', root
+      Dir.mkdir child_directory
+
+      Dir.should_receive(:pwd).and_return(child_directory)
+
+      expect { PivotalIntegration::Util::Git.repository_root }.to raise_error
+    end
+  end
+
+  it 'should get configuration when :branch scope is specified' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('test_branch_name')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git config branch.test_branch_name.test_key', false).and_return('test_value')
+
+    value = PivotalIntegration::Util::Git.get_config 'test_key', :branch
+
+    expect(value).to eq('test_value')
+  end
+
+  it 'should get configuration when :inherited scope is specified' do
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git config test_key', false).and_return('test_value')
+
+    value = PivotalIntegration::Util::Git.get_config 'test_key', :inherited
+
+    expect(value).to eq('test_value')
+  end
+
+  it 'should raise an error when an unknown scope is specified (get)' do
+    expect { PivotalIntegration::Util::Git.get_config 'test_key', :unknown }.to raise_error
+  end
+
+  it 'should set configuration when :branch scope is specified' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('test_branch_name')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git config --local branch.test_branch_name.test_key test_value')
+
+    PivotalIntegration::Util::Git.set_config 'test_key', 'test_value', :branch
+  end
+
+  it 'should set configuration when :global scope is specified' do
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git config --global test_key test_value')
+
+    PivotalIntegration::Util::Git.set_config 'test_key', 'test_value', :global
+  end
+
+  it 'should set configuration when :local scope is specified' do
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git config --local test_key test_value')
+
+    PivotalIntegration::Util::Git.set_config 'test_key', 'test_value', :local
+  end
+
+  it 'should raise an error when an unknown scope is specified (set)' do
+    expect { PivotalIntegration::Util::Git.set_config 'test_key', 'test_value', :unknown }.to raise_error
+  end
+
+  it 'should create a branch and set the root_branch and root_remote properties on it' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('master')
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('remote', :branch).and_return('origin')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git pull --quiet --ff-only')
+    PivotalIntegration::Util::Shell.should_receive(:exec).and_return('git checkout --quiet -b dev_branch')
+    PivotalIntegration::Util::Git.should_receive(:set_config).with('root-branch', 'master', :branch)
+    PivotalIntegration::Util::Git.should_receive(:set_config).with('root-remote', 'origin', :branch)
+
+    PivotalIntegration::Util::Git.create_branch 'dev_branch'
+  end
+
+  it 'should not add a hook if it already exists' do
+    Dir.mktmpdir do |root|
+      PivotalIntegration::Util::Git.should_receive(:repository_root).and_return(root)
+      hook = "#{root}/.git/hooks/prepare-commit-msg"
+      File.should_receive(:exist?).with(hook).and_return(true)
+
+      PivotalIntegration::Util::Git.add_hook 'prepare-commit-msg', __FILE__
+
+      File.should_receive(:exist?).and_call_original
+      expect(File.exist?(hook)).to be_false
+    end
+  end
+
+  it 'should add a hook if it does not exist' do
+    Dir.mktmpdir do |root|
+      PivotalIntegration::Util::Git.should_receive(:repository_root).and_return(root)
+      hook = "#{root}/.git/hooks/prepare-commit-msg"
+      File.should_receive(:exist?).with(hook).and_return(false)
+
+      PivotalIntegration::Util::Git.add_hook 'prepare-commit-msg', __FILE__
+
+      File.should_receive(:exist?).and_call_original
+      expect(File.exist?(hook)).to be_true
+    end
+  end
+
+  it 'should add a hook if it already exists and overwrite is true' do
+    Dir.mktmpdir do |root|
+      PivotalIntegration::Util::Git.should_receive(:repository_root).and_return(root)
+      hook = "#{root}/.git/hooks/prepare-commit-msg"
+
+      PivotalIntegration::Util::Git.add_hook 'prepare-commit-msg', __FILE__, true
+
+      File.should_receive(:exist?).and_call_original
+      expect(File.exist?(hook)).to be_true
+    end
+  end
+
+  it 'should fail if root tip and common_ancestor do not match' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('development_branch')
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('root-branch', :branch).and_return('master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git pull --quiet --ff-only')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet development_branch')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git rev-parse master').and_return('root_tip')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git merge-base master development_branch').and_return('common_ancestor')
+
+    lambda { PivotalIntegration::Util::Git.trivial_merge? }.should raise_error(SystemExit)
+
+    expect($stderr.string).to match(/FAIL/)
+  end
+
+  it 'should pass if root tip and common ancestor match' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('development_branch')
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('root-branch', :branch).and_return('master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git pull --quiet --ff-only')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet development_branch')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git rev-parse master').and_return('HEAD')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git merge-base master development_branch').and_return('HEAD')
+
+    PivotalIntegration::Util::Git.trivial_merge?
+
+    expect($stdout.string).to match(/OK/)
+  end
+
+  it 'should merge and delete branches' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('development_branch')
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('root-branch', :branch).and_return('master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with("git merge --quiet --no-ff -m \"Merge development_branch to master\n\n[Completes #12345678]\" development_branch")
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git branch --quiet -D development_branch')
+
+    PivotalIntegration::Util::Git.merge PivotalTracker::Story.new(:id => 12345678), nil, nil
+  end
+
+  it 'should suppress Completes statement' do
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('development_branch')
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('root-branch', :branch).and_return('master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with("git merge --quiet --no-ff -m \"Merge development_branch to master\n\n[#12345678]\" development_branch")
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git branch --quiet -D development_branch')
+
+    PivotalIntegration::Util::Git.merge PivotalTracker::Story.new(:id => 12345678), true, nil
+  end
+
+  it 'should push changes without refs' do
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('remote', :branch).and_return('origin')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git push --quiet origin ')
+
+    PivotalIntegration::Util::Git.push
+  end
+
+  it 'should push changes with refs' do
+    PivotalIntegration::Util::Git.should_receive(:get_config).with('remote', :branch).and_return('origin')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git push --quiet origin foo bar')
+
+    PivotalIntegration::Util::Git.push 'foo', 'bar'
+  end
+
+  it 'should create a commit' do
+    story = PivotalTracker::Story.new(:id => 123456789)
+    PivotalIntegration::Util::Shell.should_receive(:exec).with("git commit --quiet --all --allow-empty --message \"test_message\n\n[#123456789]\"")
+
+    PivotalIntegration::Util::Git.create_commit 'test_message', story
+  end
+
+  it 'should create a release tag' do
+    story = PivotalTracker::Story.new(:id => 123456789)
+    PivotalIntegration::Util::Git.should_receive(:branch_name).and_return('master')
+    PivotalIntegration::Util::Git.should_receive(:create_branch).with('pivotal-tracker-release', false)
+    PivotalIntegration::Util::Git.should_receive(:create_commit).with('1.0.0.RELEASE Release', story)
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git tag v1.0.0.RELEASE')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git checkout --quiet master')
+    PivotalIntegration::Util::Shell.should_receive(:exec).with('git branch --quiet -D pivotal-tracker-release')
+
+    PivotalIntegration::Util::Git.create_release_tag '1.0.0.RELEASE', story
+  end
+end

data/spec/git-pivotal-tracker-integration/util/label_spec.rb

@@ -0,0 +1,193 @@
+# Git Pivotal Tracker Integration
+# Copyright (c) 2013 the original author or authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'spec_helper'
+require 'git-pivotal-tracker-integration/util/label'
+require 'pivotal-tracker'
+
+describe PivotalIntegration::Util::Label do
+
+  before do
+    $stdout = StringIO.new
+    $stderr = StringIO.new
+
+    @project = double('project')
+    @stories = double('stories')
+    @story = double('story')
+    @menu = double('menu')
+  end
+
+  it 'should add labels to story' do
+    old_labels = 'other_label'
+    @story.should_receive(:labels).and_return(old_labels)
+    @story.should_receive(:update).with(:labels => ["other_label", "on_qa"]).and_return(true)
+    @story.should_receive(:name)
+
+    PivotalIntegration::Util::Label.add(@story, 'on_qa')
+  end
+
+  it 'should add unique labels to story' do
+    PivotalTracker::Project.should_receive(:find).and_return(@project)
+    project_id = 123
+    other_story = double(:other_story)
+    @story.should_receive(:project_id).and_return(project_id)
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).and_return([@story, other_story])
+    @story.should_receive(:name).exactly(4).times.and_return('name')
+    other_story.should_receive(:name).exactly(2).times.and_return('other_name')
+    @story.should_receive(:labels).and_return('other_label')
+    other_story.should_receive(:labels).exactly(3).times.and_return('on_qa')
+    other_story.should_receive(:update).and_return(true)
+    @story.should_receive(:update).with(:labels => ["other_label", "on_qa"]).and_return(true)
+    PivotalIntegration::Util::Label.once(@story, 'on_qa')
+  end
+
+  it 'should abort when cannot add labels to story' do
+    old_labels = 'other_label'
+    @story.should_receive(:labels).and_return(old_labels)
+    @story.should_receive(:update).with(:labels => ["other_label", "on_qa"]).and_return(false)
+    lambda { PivotalIntegration::Util::Label.add(@story, 'on_qa') }.should raise_error SystemExit
+  end
+
+  it 'should abort when cannot remove labels from story' do
+    old_labels = 'other_label,on_qa'
+    @story.should_receive(:labels).and_return(old_labels)
+    @story.should_receive(:update).with(:labels => ["other_label"]).and_return(false)
+    lambda { PivotalIntegration::Util::Label.remove(@story, 'on_qa') }.should raise_error SystemExit
+  end
+
+
+
+
+
+
+
+
+
+
+  it 'should pretty print story labels' do
+    @story.should_receive(:labels).and_return('label1,label2')
+    PivotalIntegration::Util::Label.list @story
+  end
+
+  it 'should not pretty print description or notes if there are none (empty)' do
+    story = double('story')
+    story.should_receive(:name)
+    story.should_receive(:description)
+    PivotalTracker::Note.should_receive(:all).and_return([])
+
+    PivotalIntegration::Util::Story.pretty_print story
+
+    expect($stdout.string).to eq(
+      "      Title: \n" +
+      "\n")
+  end
+
+  it 'should not pretty print description or notes if there are none (nil)' do
+    story = double('story')
+    story.should_receive(:name)
+    story.should_receive(:description).and_return('')
+    PivotalTracker::Note.should_receive(:all).and_return([])
+
+    PivotalIntegration::Util::Story.pretty_print story
+
+    expect($stdout.string).to eq(
+      "      Title: \n" +
+      "\n")
+  end
+
+  it 'should assign owner to story and notify about success' do
+    story = double('story')
+    username = 'User Name'
+    story.should_receive(:update).with({ :owned_by => username }).and_return(true)
+
+    PivotalIntegration::Util::Story.assign story, username
+
+    expect($stdout.string).to eq(
+      "Story assigned to #{username}" +
+      "\n")
+  end
+
+  it 'should change story state and notify about success' do
+    story = double('story')
+    state = 'finished'
+    story.should_receive(:update).with({ :current_state => state }).and_return(true)
+
+    PivotalIntegration::Util::Story.mark story, state
+
+    expect($stdout.string).to eq(
+      "Changed state to #{state}" +
+      "\n")
+  end
+
+  it 'should select a story directly if the filter is a number' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:find).with(12345678).and_return(@story)
+
+    story = PivotalIntegration::Util::Story.select_story @project, '12345678'
+
+    expect(story).to be(@story)
+  end
+
+  it 'should select a story if the result of the query is a single story' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).with(
+      :current_state => %w(rejected unstarted unscheduled),
+      :story_type => 'release'
+    ).and_return([@story])
+    @story.should_receive(:owned_by)
+    story = PivotalIntegration::Util::Story.select_story @project, 'release', 1
+
+    expect(story).to be(@story)
+  end
+
+  it 'should prompt the user for a story if the result of the query is more than a single story' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).with(
+      :current_state => %w(rejected unstarted unscheduled),
+      :story_type => 'feature'
+    ).and_return([
+      PivotalTracker::Story.new(:name => 'name-1'),
+      PivotalTracker::Story.new(:name => 'name-2')
+    ])
+    @menu.should_receive(:prompt=)
+    @menu.should_receive(:choice).with('name-1')
+    @menu.should_receive(:choice).with('name-2')
+    PivotalIntegration::Util::Story.should_receive(:choose) { |&arg| arg.call @menu }.and_return(@story)
+
+    story = PivotalIntegration::Util::Story.select_story @project, 'feature'
+
+    expect(story).to be(@story)
+  end
+
+  it 'should prompt the user with the story type if no filter is specified' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).with(
+      :current_state => %w(rejected unstarted unscheduled)
+    ).and_return([
+      PivotalTracker::Story.new(:story_type => 'chore', :name => 'name-1'),
+      PivotalTracker::Story.new(:story_type => 'bug', :name => 'name-2')
+    ])
+    @menu.should_receive(:prompt=)
+    @menu.should_receive(:choice).with('CHORE   name-1')
+    @menu.should_receive(:choice).with('BUG     name-2')
+    PivotalIntegration::Util::Story.should_receive(:choose) { |&arg| arg.call @menu }.and_return(@story)
+
+    story = PivotalIntegration::Util::Story.select_story @project
+
+    expect(story).to be(@story)
+  end
+
+end

data/spec/git-pivotal-tracker-integration/util/shell_spec.rb

@@ -0,0 +1,52 @@
+# Git Pivotal Tracker Integration
+# Copyright (c) 2013 the original author or authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'spec_helper'
+require 'git-pivotal-tracker-integration/util/shell'
+
+describe PivotalIntegration::Util::Shell do
+
+  before do
+    $stdout = StringIO.new
+    $stderr = StringIO.new
+  end
+
+  it 'should return result when exit code is 0' do
+    PivotalIntegration::Util::Shell.should_receive(:`).with('test_command').and_return('test_result')
+    $?.should_receive(:exitstatus).and_return(0)
+
+    result = PivotalIntegration::Util::Shell.exec 'test_command'
+
+    expect(result).to eq('test_result')
+  end
+
+  it "should abort with 'FAIL' when the exit code is not 0" do
+    PivotalIntegration::Util::Shell.should_receive(:`).with('test_command')
+    $?.should_receive(:exitstatus).and_return(-1)
+
+    lambda { PivotalIntegration::Util::Shell.exec 'test_command' }.should raise_error(SystemExit)
+
+    expect($stderr.string).to match(/FAIL/)
+  end
+
+  it 'should return result when the exit code is not 0 and told not to abort on failure' do
+    PivotalIntegration::Util::Shell.should_receive(:`).with('test_command')
+    $?.should_receive(:exitstatus).and_return(-1)
+
+    PivotalIntegration::Util::Shell.exec 'test_command', false
+  end
+
+
+end

data/spec/git-pivotal-tracker-integration/util/story_spec.rb

@@ -0,0 +1,158 @@
+# Git Pivotal Tracker Integration
+# Copyright (c) 2013 the original author or authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'spec_helper'
+require 'git-pivotal-tracker-integration/util/story'
+require 'pivotal-tracker'
+
+describe PivotalIntegration::Util::Story do
+
+  before do
+    $stdout = StringIO.new
+    $stderr = StringIO.new
+
+    @project = double('project')
+    @stories = double('stories')
+    @story = double('story')
+    @menu = double('menu')
+  end
+
+  it 'should pretty print story information' do
+    story = double('story')
+    story.should_receive(:name)
+    story.should_receive(:description).and_return("description-1\ndescription-2")
+    PivotalTracker::Note.should_receive(:all).and_return([
+      PivotalTracker::Note.new(:noted_at => Date.new, :text => 'note-1')
+    ])
+
+    PivotalIntegration::Util::Story.pretty_print story
+
+    expect($stdout.string).to eq(
+      "      Title: \n" +
+      "Description: description-1\n" +
+      "             description-2\n" +
+      "     Note 1: note-1\n" +
+      "\n")
+  end
+
+  it 'should not pretty print description or notes if there are none (empty)' do
+    story = double('story')
+    story.should_receive(:name)
+    story.should_receive(:description)
+    PivotalTracker::Note.should_receive(:all).and_return([])
+
+    PivotalIntegration::Util::Story.pretty_print story
+
+    expect($stdout.string).to eq(
+      "      Title: \n" +
+      "\n")
+  end
+
+  it 'should not pretty print description or notes if there are none (nil)' do
+    story = double('story')
+    story.should_receive(:name)
+    story.should_receive(:description).and_return('')
+    PivotalTracker::Note.should_receive(:all).and_return([])
+
+    PivotalIntegration::Util::Story.pretty_print story
+
+    expect($stdout.string).to eq(
+      "      Title: \n" +
+      "\n")
+  end
+
+  it 'should assign owner to story and notify about success' do
+    story = double('story')
+    username = 'User Name'
+    story.should_receive(:update).with({ :owned_by => username }).and_return(true)
+
+    PivotalIntegration::Util::Story.assign story, username
+
+    expect($stdout.string).to eq(
+      "Story assigned to #{username}" +
+      "\n")
+  end
+
+  it 'should change story state and notify about success' do
+    story = double('story')
+    state = 'finished'
+    story.should_receive(:update).with({ :current_state => state }).and_return(true)
+
+    PivotalIntegration::Util::Story.mark story, state
+
+    expect($stdout.string).to eq(
+      "Changed state to #{state}" +
+      "\n")
+  end
+
+  it 'should select a story directly if the filter is a number' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:find).with(12345678).and_return(@story)
+
+    story = PivotalIntegration::Util::Story.select_story @project, '12345678'
+
+    expect(story).to be(@story)
+  end
+
+  it 'should select a story if the result of the query is a single story' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).with(
+      :current_state => %w(rejected unstarted unscheduled),
+      :story_type => 'release'
+    ).and_return([@story])
+    @story.should_receive(:owned_by)
+    story = PivotalIntegration::Util::Story.select_story @project, 'release', 1
+
+    expect(story).to be(@story)
+  end
+
+  it 'should prompt the user for a story if the result of the query is more than a single story' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).with(
+      :current_state => %w(rejected unstarted unscheduled),
+      :story_type => 'feature'
+    ).and_return([
+      PivotalTracker::Story.new(:name => 'name-1'),
+      PivotalTracker::Story.new(:name => 'name-2')
+    ])
+    @menu.should_receive(:prompt=)
+    @menu.should_receive(:choice).with('name-1')
+    @menu.should_receive(:choice).with('name-2')
+    PivotalIntegration::Util::Story.should_receive(:choose) { |&arg| arg.call @menu }.and_return(@story)
+
+    story = PivotalIntegration::Util::Story.select_story @project, 'feature'
+
+    expect(story).to be(@story)
+  end
+
+  it 'should prompt the user with the story type if no filter is specified' do
+    @project.should_receive(:stories).and_return(@stories)
+    @stories.should_receive(:all).with(
+      :current_state => %w(rejected unstarted unscheduled)
+    ).and_return([
+      PivotalTracker::Story.new(:story_type => 'chore', :name => 'name-1'),
+      PivotalTracker::Story.new(:story_type => 'bug', :name => 'name-2')
+    ])
+    @menu.should_receive(:prompt=)
+    @menu.should_receive(:choice).with('CHORE   name-1')
+    @menu.should_receive(:choice).with('BUG     name-2')
+    PivotalIntegration::Util::Story.should_receive(:choose) { |&arg| arg.call @menu }.and_return(@story)
+
+    story = PivotalIntegration::Util::Story.select_story @project
+
+    expect(story).to be(@story)
+  end
+
+end