lib/drg/tasks/gemfile.rb in drg-0.3.0 vs lib/drg/tasks/gemfile.rb in drg-0.4.0

@@ -5,30 +5,30 @@
 
       def initialize
         @file = ::Bundler.default_gemfile
       end
 
+      # @param [GemfileLine] gem
+      # @param [String] version to update the gem line with
       def update(gem, version)
-        lines[gem] = gem.update(version)
+        lines[gem] = gem.update version
       end
 
       def find_by_name(name)
         lines.each_with_index.each do |line, index|
           next if line =~ /:?path:?\s*(=>)?\s*/
           next if line =~ /:?git(hub)?:?\s*(=>)?\s*/
-          if line =~ /gem\s*['"]#{name}["']/
-            return GemfileLine.new line, index
-          end
+          return GemfileLine.new line, index if line =~ /gem\s*['"]#{name}["']/
         end
         nil
       end
 
       def lines
         @lines ||= File.readlines file
       end
 
       def write
-        File.open(file, 'wb') do |f|
+        File.open file, 'wb' do |f|
           lines.each do |line|
             f << line
           end
         end
       end