Add 'top' and 'bottom' as synonyms for 'before' and 'after' models positions

Author: kamilbielawski

lib/annotate/annotate_models.rb

@@ -224,7 +224,7 @@ def get_index_info(klass, options={})
     # === Options (opts)
     #  :force<Symbol>:: whether to update the file even if it doesn't seem to need it.
     #  :position_in_*<Symbol>:: where to place the annotated section in fixture or model file,
-    #                           :before or :after. Default is :before.
+    #                           :before, :top, :after or :bottom. Default is :before.
     #
     def annotate_one_file(file_name, info_block, position, options={})
       if File.exist?(file_name)
@@ -259,7 +259,7 @@ def annotate_one_file(file_name, info_block, position, options={})
             old_content.sub!(encoding, '')
             old_content.sub!(PATTERN, '')
 
-            new_content = options[position].to_s == 'after' ?
+            new_content = %w(after bottom).include?(options[position].to_s) ?
               (encoding_header + (old_content.rstrip + "\n\n" + info_block)) :
               (encoding_header + info_block + "\n" + old_content)
           end

spec/annotate/annotate_models_spec.rb

@@ -425,6 +425,16 @@ def encoding_comments_list_each
       expect(File.read(@model_file_name)).to eq("#{@schema_info}\n#{@file_content}")
     end
 
+    it "should put annotation before class if :position == 'top'" do
+      annotate_one_file :position => "top"
+      expect(File.read(@model_file_name)).to eq("#{@schema_info}\n#{@file_content}")
+    end
+
+    it "should put annotation before class if :position => :top" do
+      annotate_one_file :position => :top
+      expect(File.read(@model_file_name)).to eq("#{@schema_info}\n#{@file_content}")
+    end
+
     it "should put annotation after class if :position => 'after'" do
       annotate_one_file :position => 'after'
       expect(File.read(@model_file_name)).to eq("#{@file_content}\n#{@schema_info}")
@@ -435,6 +445,16 @@ def encoding_comments_list_each
       expect(File.read(@model_file_name)).to eq("#{@file_content}\n#{@schema_info}")
     end
 
+    it "should put annotation after class if :position => 'bottom'" do
+      annotate_one_file :position => 'bottom'
+      expect(File.read(@model_file_name)).to eq("#{@file_content}\n#{@schema_info}")
+    end
+
+    it "should put annotation after class if :position => :bottom" do
+      annotate_one_file :position => :bottom
+      expect(File.read(@model_file_name)).to eq("#{@file_content}\n#{@schema_info}")
+    end
+
     describe "with existing annotation => :before" do
       before do
         annotate_one_file :position => :before