Tests for model annotating with encoding comments

Author: Artūrs Mekšs

spec/annotate/annotate_models_spec.rb

@@ -345,6 +345,16 @@ def annotate_one_file options = {}
       Annotate::PATH_OPTIONS.each { |key| ENV[key.to_s] = '' }
     end
 
+    def encoding_comments_list_each
+      [
+        '# encoding: UTF-8',
+        '# coding: UTF-8',
+        '# -*- coding: UTF-8 -*-',
+        '#encoding: utf-8',
+        '# -*- encoding : utf-8 -*-'
+      ].each{|encoding_comment| yield encoding_comment }
+    end
+
     it "should annotate the file before the model if position == 'before'" do
       annotate_one_file :position => "before"
       File.read(@model_file_name).should == "#{@schema_info}#{@file_content}"
@@ -387,6 +397,20 @@ class Foo::User < ActiveRecord::Base
       File.read(model_file_name).should == "#{schema_info}#{file_content}"
     end
 
+    it "should not touch encoding comments" do
+      encoding_comments_list_each do |encoding_comment|
+        write_model "user.rb", <<-EOS
+#{encoding_comment}
+class User < ActiveRecord::Base
+end
+        EOS
+
+        annotate_one_file :position => :before
+
+        File.read(@model_file_name).should == "#{encoding_comment}\n#{@schema_info}#{@file_content}"
+      end
+    end
+
     describe "if a file can't be annotated" do
        before do
          write_model('user.rb', <<-EOS)