added bump generator

Rakefile

@@ -2,8 +2,7 @@
 lib = File.expand_path("../../lib", __FILE__)
 $LOAD_PATH.unshift lib unless $LOAD_PATH.include?(lib)
 require "bundler/gem_tasks"
-require "flavor_gem/gem_tasks"
-
+require "flavor_gem/bump_tasks"
 
 task :default => :test
 

lib/flavor_gem/bump.rb

@@ -6,22 +6,41 @@ class Bump < Thor
 
     desc "generate", "generate rake bump task to update version file easily"
     def generate
-      insert_into_file "Rakefile", gem_tasks, :after => "bundler/gem_tasks\"\n"
+      add_bump_tasks "Rakefile"
     end
 
     desc "delete", "delete generated bump flavor"
     def delete
-      gsub_file "Rakefile", /require "flavor_gem\/gem_tasks"/, ""
-      gsub_file "Rakefile", /require 'flavor_gem\/gem_tasks'/, ""      
+      remove_bump_tasks "Rakefile"
     end
 
     private
-    def rakefile
-      File.read "Rakefile"
+    def bump_tasks
+      "require \"flavor_gem/bump_tasks\"\n"
     end
 
-    def gem_tasks
-      'require "flavor_gem/gem_tasks"'
+    def bump_tasks_regex
+      /require ["']flavor_gem\/bump_tasks["']/
+    end
+
+    def add_bump_tasks file_name
+      rakefile = File.read file_name
+      single_quote_bundler_require = 'require "bundler\/bump_tasks"\n'
+      double_quote_bundler_require = "require 'bundler/bump_tasks'\n"
+      if rakefile =~ bump_tasks_regex
+        return
+      end
+      if rakefile.include? single_quote_bundler_require
+        insert_into_file file_name, bump_tasks, :after => single_quote_bundler_require
+      elsif rakefile.include? double_quote_bundler_require
+        insert_into_file file_name, bump_tasks, :after => double_quote_bundler_require
+      else
+        prepend_to_file file_name, bump_tasks
+      end
+    end
+
+    def remove_bump_tasks file_name
+      gsub_file file_name, /require ["']flavor_gem\/bump_tasks["']\n/, ""      
     end
   end
 end

test/minitest_helper.rb

@@ -5,6 +5,8 @@
 require 'flavor_gem'
 require 'flavor_gem/runner'
 
+
+
 # added by flavor-gem generate minitest
 require 'minitest/autorun'
 

test/test_bump.rb

@@ -1,23 +1,27 @@
 require 'minitest_helper'
 
+class FlavorGem::Bump
+  public :add_bump_tasks, :remove_bump_tasks
+end
+
 class TestBump < Minitest::Test
   def setup
     @bump = FlavorGem::Bump.new
   end
 
-  def gem_tasks
-    'require "flavor_gem/gem_tasks"'    
+  def bump_tasks_regex
+    /require ["']flavor_gem\/bump_tasks["']/    
   end
 
-  # test "should delete bump task from Rakefile" do
-  #   @bump.delete
-  #   rakefile = File.read "Rakefile"
-  #   refute_match /#{gem_tasks}/, rakefile
-  # end
+  test "should add and delete bump task from Rakefile" do
+    rakefile_name = "lib/flavor_gem/template/Rakefile"
 
-  # test "should add bump task into Rakefile" do
-  #   @bump.generate
-  #   rakefile = File.read "Rakefile"
-  #   assert_match /flavor_gem\/gem_tasks/, rakefile
-  # end
+    @bump.add_bump_tasks rakefile_name
+    rakefile = File.read rakefile_name    
+    assert_match bump_tasks_regex, rakefile
+
+    @bump.remove_bump_tasks rakefile_name
+    rakefile = File.read rakefile_name    
+    refute_match bump_tasks_regex, rakefile
+  end
 end