doubledrones-annotate 2.4.0.1 → 2.4.0.2

data/lib/annotate/annotate_models.rb

@@ -127,7 +127,7 @@ module AnnotateModels
           false
         else
           # Remove old schema info
-          old_content.sub!(/^# #{COMPAT_PREFIX}.*?\n(#.*\n)*\n/, '')
+          old_content.sub!(/^# #{COMPAT_PREFIX}.*?\n(#.*\n)*/, '')
 
           # Write it back
           new_content = options[:position] == 'before' ?  (info_block + old_content) : (old_content + info_block)

data/spec/annotate/annotate_models_spec.rb

@@ -79,4 +79,56 @@ EOS
     end
   end
 
+  describe "annotating one file already annotated but schema was changed" do
+    def create(file, body="hi")
+      File.open(@dir + '/' + file, "w") do |f|
+        f.puts(body)
+      end
+    end
+
+    before :all do
+      require "tmpdir"
+      @dir = Dir.tmpdir + "/#{Time.now.to_i}" + "/annotate_models"
+      FileUtils.mkdir_p(@dir)
+      AnnotateModels.model_dir = @dir
+      create('foo.rb', <<-EOS)
+class Foo < ActiveRecord::Base
+end
+# == Schema Information
+#
+# Table name: users
+#
+#  id    :integer         primary key
+#  name  :string
+#
+EOS
+    end
+
+    it 'should replace old annotation' do
+      info_block = <<-EOS
+# == Schema Information
+#
+# Table name: users
+#
+#  id    :integer         primary key
+#  name  :string
+#  new   :string
+#
+EOS
+      AnnotateModels.annotate_one_file(@dir + '/foo.rb', info_block).should be_true
+      File.read(@dir + '/foo.rb').should == <<-EOS
+class Foo < ActiveRecord::Base
+end
+# == Schema Information
+#
+# Table name: users
+#
+#  id    :integer         primary key
+#  name  :string
+#  new   :string
+#
+EOS
+    end
+  end
+
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification 
 name: doubledrones-annotate
 version: !ruby/object:Gem::Version 
-  hash: 77
+  hash: 75
   prerelease: false
   segments: 
   - 2
   - 4
   - 0
-  - 1
-  version: 2.4.0.1
+  - 2
+  version: 2.4.0.2
 platform: ruby
 authors: 
 - Cuong Tran