lib/pessimize/version_mapper.rb in pessimize-0.0.3 vs lib/pessimize/version_mapper.rb in pessimize-0.1.0

@@ -1,10 +1,12 @@
 module Pessimize
   class VersionMapper
-    def call(gems, versions)
+    def call(gems, versions, version_constraint)
       gems.each do |gem|
         if versions.has_key? gem.name
-          gem.version = "~> #{versions[gem.name]}"
+          version_parts = versions[gem.name].split('.')
+          version = version_constraint == 'minor' ? version_parts.first(2).join('.') : version_parts.join('.')
+          gem.version = "~> #{version}"
         end
       end
     end
   end
 end