Modify prerelease checking

Add additional specs for `--pre` flag

1 files changed, 11 insertions, 2 deletions

diff --git a/spec/commands/add_spec.rb b/spec/commands/add_spec.rb
index 815a3a6c..d6b9b46f 100644
--- a/spec/commands/add_spec.rb
+++ b/spec/commands/add_spec.rb
@@ -32,15 +32,24 @@ describe "bundle add" do
       expect(bundled_app("Gemfile").read).to include("gem 'activesupport', '~> 2.3.5'")
     end
 
-    it "adds the gem with the last prerelease version" do
+    it "`--pre` flag adds the gem with the latest prerelease version" do
       update_repo2 do
-        build_gem "activesupport", "3.0.0"
         build_gem "activesupport", "3.0.0.beta"
       end
 
       bundle "add activesupport --pre"
       expect(bundled_app("Gemfile").read).to include("gem 'activesupport', '~> 3.0.0.beta'")
     end
+
+    it "`--pre` flag adds the gem with the latest non-prerelease version if it is available" do
+      update_repo2 do
+        build_gem "activesupport", "3.0.0.beta"
+        build_gem "activesupport", "3.0.0"
+      end
+
+      bundle "add activesupport --pre"
+      expect(bundled_app("Gemfile").read).to include("gem 'activesupport', '~> 3.0.0'")
+    end
   end
 
   context "when group is set" do