merge

.gitignore

@@ -4,3 +4,4 @@ rdoc/*
 coverage/*
 spec/debug.log
 pkg/*
+.rvmrc
\ No newline at end of file

History.txt

+== 2.4.2 2009-11-21
+
+* Annotates (spec|test)/factories/<model>_factory.rb files
+
+== 2.4.1 2009-11-20
+
+* Annotates thoughtbot's factory_girl factories (test/factories/<model>_factory.rb)
+* Move default annotation position back to top
+
+== 2.4.0 2009-12-13
+
+* Incorporated lots of patches from the Github community, including support for Blueprints fixtures
+* Several bug fixes
+
 == 2.1 2009-10-18
 
 * New options

README.rdoc

@@ -7,6 +7,7 @@ Add a comment summarizing the current schema to the top or bottom of each of you
   * Tests and Specs
   * Object Daddy exemplars
   * Machinist blueprints
+  * Thoughtbot's factory_girl factories, i.e. the (spec|test)/factories/<model>_factory.rb files
 
 The schema comment looks like this:
 
@@ -46,8 +47,8 @@ From github:
 
   git clone git://github.com/ctran/annotate_models.git annotate
   cd annotate
-  rake gem
-  sudo gem install pkg/annotate-*.gem
+  gem build annotate.gemspec
+  sudo gem install annotate-<version>.gem
 
 == USAGE
 
@@ -77,6 +78,9 @@ adjust your <tt>rake db:migrate</tt> tasks so that they update the
 annotations in your model files for you once the migration is
 completed.
 
+Warning: ImageMagick installs a tool called `annotate` too (if you're using MacPorts it's in `/opt/local/bin/annotate`. So if you see Usage: annotate imagein.jpg imageout.jpg then put `/usr/bin` ahead on the path and you'll get ours instead. 
+
+
 == OPTIONS
 
     Usage: annotate [options] [model_file]*
@@ -89,7 +93,8 @@ completed.
         -s, --simple-indexes             Concat the column's related indexes in the annotation
             --model-dir dir              Annotate model files stored in dir rather than app/models
         -R, --require path               Additional files to require before loading models
-        -e, --exclude [tests,fixtures]   Do not annotate fixtures, test files, or both
+        -e [tests,fixtures,factories]    Do not annotate fixtures, test files, and/or factories
+            --exclude
 
 
 == WARNING
@@ -103,7 +108,7 @@ to an automatically created comment block.
 
 == LINKS
 
-* Factory Girl => http://github.com/thoughtbot/factory_girl (NOT IMPLEMENTED)
+* Factory Girl => http://github.com/thoughtbot/factory_girl
 * Object Daddy => http://github.com/flogic/object_daddy
 * SpatialAdapter => http://github.com/pdeffendol/spatial_adapter
 * PostgisAdapter => http://github.com/nofxx/postgis_adapter
@@ -118,6 +123,7 @@ Original code by: Dave Thomas -- Pragmatic Programmers, LLC
 Overhauled by: Alex Chaffee
 Gemmed by: Cuong Tran
 Maintained by: Alex Chaffee and Cuong Tran
+Homepage: http://github.com/ctran/annotate_models
 
 Modifications by:
 
@@ -134,5 +140,6 @@ Modifications by:
  - Bob Potter - http://github.com/bpot
  - Gavin Montague - http://github.com/govan/
  - Alexander Semyonov - http://github.com/rotuka/
+ - Nathan Brazil - http://github.com/bitaxis/
  
 and many others that I may have missed to add.

Rakefile

-require 'rubygems'
+# require 'rubygems'
 require 'rake'
-require 'lib/annotate'
+require File.dirname(__FILE__) + '/lib/annotate'
 
 # want other tests/tasks run by default? Add them to the list
 task :default => [:spec]
@@ -8,39 +8,43 @@ task :default => [:spec]
 begin
   require 'jeweler'
   Jeweler::Tasks.new do |gem|
-    gem.name = "annotate"
+    gem.name = "netsign-annotate"
     gem.executables = "annotate"
     gem.summary = "Annotates Rails Models, routes, fixtures, and others based on the database schema."
-    gem.description = gem.summary
+    gem.description = "Packaged for netSIGN, gem experimentation"
     gem.email = ["alex@stinky.com", 'ctran@pragmaquest.com', "x@nofxx.com"]
-    gem.homepage = "http://github.com/ctran/annotate"
+    gem.homepage = "https://github.com/miyucy/annotate_models"
     gem.authors = ['Cuong Tran', "Alex Chaffee", "Marcos Piccinini"]
     gem.files =  FileList["[A-Z]*.*", "{bin,lib,tasks,spec}/**/*"]
-    gem.rubyforge_project = "annotate"
-    
+    gem.rubyforge_project = "netsign-annotate"
+
     # note that Jeweler automatically reads the version from VERSION.yml
     # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
   end
+
+  # Jeweler::RubyforgeTasks.new do |rubyforge|
+  #   rubyforge.doc_task = "rdoc"
+  # end
   
-  Jeweler::RubyforgeTasks.new do |rubyforge|
-    rubyforge.doc_task = "rdoc"
-  end
+  Jeweler::GemcutterTasks.new
   
 rescue LoadError
   puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
 end
 
-require 'spec/rake/spectask'
-Spec::Rake::SpecTask.new(:spec) do |spec|
-  spec.libs << 'lib' << 'spec'
-  spec.spec_files = FileList['spec/**/*_spec.rb']
-end
+# Dir["#{File.dirname(__FILE)}/tasks/*.rake"].sort.each { |ext| load ext }
 
-Spec::Rake::SpecTask.new(:rcov) do |spec|
-  spec.libs << 'lib' << 'spec'
-  spec.pattern = 'spec/**/*_spec.rb'
-  spec.rcov = true
-end
+# require 'spec/rake/spectask'
+# Spec::Rake::SpecTask.new(:spec) do |spec|
+#   spec.libs << 'lib' << 'spec'
+#   spec.spec_files = FileList['spec/**/*_spec.rb']
+# end
+# 
+# Spec::Rake::SpecTask.new(:rcov) do |spec|
+#   spec.libs << 'lib' << 'spec'
+#   spec.pattern = 'spec/**/*_spec.rb'
+#   spec.rcov = true
+# end
 
 require 'rake/rdoctask'
 Rake::RDocTask.new do |rdoc|

VERSION.yml

 --- 
   :major: 2
   :minor: 4
-  :patch: 0
+  :patch: 3

annotate.gemspec

 # Generated by jeweler
-# DO NOT EDIT THIS FILE
-# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
 # -*- encoding: utf-8 -*-
 
 Gem::Specification.new do |s|
   s.name = %q{annotate}
-  s.version = "2.4.0"
+  s.version = "2.4.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Cuong Tran", "Alex Chaffee", "Marcos Piccinini"]
-  s.date = %q{2009-10-23}
+  s.date = %q{2011-01-29}
   s.default_executable = %q{annotate}
   s.description = %q{Annotates Rails Models, routes, fixtures, and others based on the database schema.}
   s.email = ["alex@stinky.com", "ctran@pragmaquest.com", "x@nofxx.com"]
@@ -27,17 +27,18 @@ Gem::Specification.new do |s|
      "lib/annotate/annotate_routes.rb",
      "lib/tasks/annotate_models.rake",
      "lib/tasks/annotate_routes.rake",
+     "lib/tasks/migrate.rake",
      "spec/annotate/annotate_models_spec.rb",
      "spec/annotate/annotate_routes_spec.rb",
      "spec/annotate_spec.rb",
      "spec/spec.opts",
-     "spec/spec_helper.rb",
-     "tasks/migrate.rake"
+     "spec/spec_helper.rb"
   ]
   s.homepage = %q{http://github.com/ctran/annotate}
   s.rdoc_options = ["--charset=UTF-8"]
   s.require_paths = ["lib"]
-  s.rubygems_version = %q{1.3.5}
+  s.rubyforge_project = %q{annotate}
+  s.rubygems_version = %q{1.3.7}
   s.summary = %q{Annotates Rails Models, routes, fixtures, and others based on the database schema.}
   s.test_files = [
     "spec/annotate/annotate_models_spec.rb",
@@ -50,9 +51,10 @@ Gem::Specification.new do |s|
     current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
     s.specification_version = 3
 
-    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
     else
     end
   else
   end
 end
+

bin/annotate

@@ -8,48 +8,48 @@ task = :annotate_models
 OptionParser.new do |opts|
   opts.banner = "Usage: annotate [options] [model_file]*"
 
-  opts.on('-d', '--delete', 
-  "Remove annotations from all model files") do
+  opts.on('-d', '--delete',
+          "Remove annotations from all model files") do
     task = :remove_annotation
   end
 
-  opts.on('-p', '--position [before|after]', ['before', 'after'], 
-  "Place the annotations at the top (before) or the bottom (after) of the model file") do |p| 
+  opts.on('-p', '--position [before|after]', ['before', 'after'],
+          "Place the annotations at the top (before) or the bottom (after) of the model file") do |p|
     ENV['position'] = p
   end
 
-  opts.on('-r', '--routes', 
-  "Annotate routes.rb with the output of 'rake routes'") do
+  opts.on('-r', '--routes',
+          "Annotate routes.rb with the output of 'rake routes'") do
     task = :annotate_routes
   end
 
-  opts.on('-v', '--version', 
-  "Show the current version of this gem") do 
+  opts.on('-v', '--version',
+          "Show the current version of this gem") do
     puts "annotate v#{Annotate.version}"; exit
   end
 
-  opts.on('-m', '--show-migration', 
-  "Include the migration version number in the annotation") do
+  opts.on('-m', '--show-migration',
+          "Include the migration version number in the annotation") do
     ENV['include_version'] = "yes"
   end
 
-  opts.on('-i', '--show-indexes', 
-  "List the table's database indexes in the annotation") do   
+  opts.on('-i', '--show-indexes',
+          "List the table's database indexes in the annotation") do
     ENV['show_indexes'] = "yes"
   end
 
   opts.on('-s', '--simple-indexes',
-  "Concat the column's related indexes in the annotation") do
+          "Concat the column's related indexes in the annotation") do
     ENV['simple_indexes'] = "yes"
   end
 
-  opts.on('--model-dir dir', 
-  "Annotate model files stored in dir rather than app/models") do |dir|
+  opts.on('--model-dir dir',
+          "Annotate model files stored in dir rather than app/models") do |dir|
     ENV['model_dir'] = dir
   end
 
   opts.on('-R', '--require path',
-  "Additional files to require before loading models") do |path|
+          "Additional files to require before loading models") do |path|
     if ENV['require']
       ENV['require'] = ENV['require'] + ",#{path}"
     else
@@ -61,6 +61,10 @@ OptionParser.new do |opts|
     exclusions.each { |exclusion| ENV["exclude_#{exclusion}"] = "yes" }
   end
 
+  opts.on('-f', '--format [bare|rdoc]', ['bare', 'rdoc'], 'rdoc: render Schema Infomation as RDoc') do |fmt|
+    ENV['format_#{fmt}'] = 'yes'
+  end
+
 end.parse!
 
 if Annotate.load_tasks

lib/annotate.rb

-$:.unshift(File.dirname(__FILE__)) unless
-  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+require 'yaml'
 
 module Annotate
   def self.version
@@ -11,7 +10,7 @@ module Annotate
       version = "0.0.0"
     end
   end
-  
+
   def self.load_tasks
     if File.exists?('Rakefile')
       load 'Rakefile'

lib/tasks/annotate_models.rake

 desc "Add schema information (as comments) to model and fixture files"
 task :annotate_models => :environment do
-  require 'annotate/annotate_models'
+  require File.join(File.dirname(__FILE__), '..', 'annotate/annotate_models')
   options={}
-  options[:position_in_class] = ENV['position_in_class'] || ENV['position'] || :before
-  options[:position_in_fixture] = ENV['position_in_fixture'] || ENV['position']  || :before
-  options[:show_indexes] = ENV['show_indexes']
-  options[:simple_indexes] = ENV['simple_indexes']
-  options[:model_dir] = ENV['model_dir']
-  options[:include_version] = ENV['include_version']
-  options[:require] = ENV['require'] ? ENV['require'].split(',') : []
-  options[:exclude_tests] = ENV['exclude_tests']
-  options[:exclude_fixtures] = ENV['exclude_fixtures']
+  options[:position_in_class]   = ENV['position_in_class']   || ENV['position'] || :before
+  options[:position_in_factory] = ENV['position_in_factory'] || ENV['position'] || :before
+  options[:position_in_fixture] = ENV['position_in_fixture'] || ENV['position'] || :before
+  options[:show_indexes]        = ENV['show_indexes']
+  options[:simple_indexes]      = ENV['simple_indexes']
+  options[:model_dir]           = ENV['model_dir']
+  options[:include_version]     = ENV['include_version']
+  options[:require]             = ENV['require'] ? ENV['require'].split(', ') : []
+  options[:exclude_tests]       = ENV['exclude_tests']
+  options[:exclude_fixtures]    = ENV['exclude_fixtures']
   AnnotateModels.do_annotations(options)
 end
 
 desc "Remove schema information from model and fixture files"
 task :remove_annotation => :environment do
-  require 'annotate/annotate_models'
+  require File.join(File.dirname(__FILE__), '..', 'annotate/annotate_models')
   options={}
   options[:model_dir] = ENV['model_dir']
   AnnotateModels.remove_annotations(options)

lib/tasks/annotate_routes.rake

 desc "Prepends the route map to the top of routes.rb"
-task :annotate_routes do
+task :annotate_routes => :environment do
   require 'annotate/annotate_routes'
   AnnotateRoutes.do_annotate
 end

tasks/migrate.rake → lib/tasks/migrate.rake

@@ -5,15 +5,32 @@
 # run after doing db:migrate. 
 # Unfortunately it relies on ENV for options; it'd be nice to be able to set options
 # in a per-project config file so this task can read them.
+
+def run_annotate_models?
+  update_on_migrate = true
+  (defined? ANNOTATE_MODELS_PREFS::UPDATE_ON_MIGRATE) && 
+    (!ANNOTATE_MODELS_PREFS::UPDATE_ON_MIGRATE.nil?) ? 
+      ANNOTATE_MODELS_PREFS::UPDATE_ON_MIGRATE : true
+end
+
+
 namespace :db do
   task :migrate do
+    if run_annotate_models?
+      Annotate::Migration.update_annotations
+    end
+  end
+
+  task :update => [:migrate] do
     Annotate::Migration.update_annotations
   end
 
   namespace :migrate do
     [:up, :down, :reset, :redo].each do |t|
       task t do
-        Annotate::Migration.update_annotations
+        if run_annotate_models?
+          Annotate::Migration.update_annotations
+        end
       end
     end
   end

netsign-annotate.gemspec

+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{netsign-annotate}
+  s.version = "2.4.3"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Cuong Tran", "Alex Chaffee", "Marcos Piccinini"]
+  s.date = %q{2011-01-30}
+  s.default_executable = %q{annotate}
+  s.description = %q{Packaged for netSIGN, gem experimentation}
+  s.email = ["alex@stinky.com", "ctran@pragmaquest.com", "x@nofxx.com"]
+  s.executables = ["annotate"]
+  s.extra_rdoc_files = [
+    "README.rdoc"
+  ]
+  s.files = [
+    "History.txt",
+    "README.rdoc",
+    "VERSION.yml",
+    "bin/annotate",
+    "lib/annotate.rb",
+    "lib/annotate/annotate_models.rb",
+    "lib/annotate/annotate_routes.rb",
+    "lib/tasks/annotate_models.rake",
+    "lib/tasks/annotate_routes.rake",
+    "lib/tasks/migrate.rake",
+    "spec/annotate/annotate_models_spec.rb",
+    "spec/annotate/annotate_routes_spec.rb",
+    "spec/annotate_spec.rb",
+    "spec/spec.opts",
+    "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{https://github.com/miyucy/annotate_models}
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{netsign-annotate}
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Annotates Rails Models, routes, fixtures, and others based on the database schema.}
+  s.test_files = [
+    "spec/annotate/annotate_models_spec.rb",
+    "spec/annotate/annotate_routes_spec.rb",
+    "spec/annotate_spec.rb",
+    "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end
+

spec/annotate/annotate_models_spec.rb

+#encoding: utf-8
 require File.dirname(__FILE__) + '/../spec_helper.rb'
 require 'annotate/annotate_models'
-require 'rubygems'
-require 'activesupport'
+require 'active_support'
+require 'fakefs/spec_helpers'
+require 'tmpdir'
 
 describe AnnotateModels do
-
-  def mock_klass(stubs={})
-    @mock_file ||= mock("Klass", stubs)
+  include FakeFS::SpecHelpers
+
+  def mock_class(table_name, primary_key, columns)
+    options = {
+      :connection   => mock("Conn", :indexes => []),
+      :table_name   => table_name,
+      :primary_key  => primary_key.to_s,
+      :column_names => columns.map { |col| col.name.to_s },
+      :columns      => columns
+    }
+
+    mock("An ActiveRecord class", options)
   end
 
-  def mock_column(stubs={})
-    @mock_column ||= mock("Column", stubs)
+  def mock_column(name, type, options={})
+    default_options = {
+      :limit   => nil,
+      :null    => false,
+      :default => nil
+    }
+
+    stubs = default_options.dup
+    stubs.merge!(options)
+    stubs.merge!(:name => name, :type => type)
+
+    mock("Column", stubs)
   end
 
   it { AnnotateModels.quote(nil).should eql("NULL") }
@@ -21,45 +42,58 @@ describe AnnotateModels do
   it { AnnotateModels.quote(1e-20).should eql("1.0e-20") }
 
   it "should get schema info" do
+    klass = mock_class(:users, :id, [
+                                     mock_column(:id, :integer),
+                                     mock_column(:name, :string, :limit => 50)
+                                    ])
 
-    AnnotateModels.get_schema_info(mock_klass(
-      :connection => mock("Conn", :indexes => []),
-      :table_name => "users",
-      :primary_key => "id",
-      :column_names => ["id","login"],
-      :columns => [
-        mock_column(:type => "integer", :default => nil, :null => false, :name => "id", :limit => nil),
-        mock_column(:type => "string", :default => nil, :null => false, :name => "name", :limit => 50)
-      ]), "Schema Info").should eql(<<-EOS)
+    AnnotateModels.get_schema_info(klass, "Schema Info").should eql(<<-EOS)
 # Schema Info
 #
 # Table name: users
 #
-#  id    :integer         not null, primary key
-#  id    :integer         not null, primary key
+#  id   :integer         not null, primary key
+#  name :string(50)      not null
 #
 
 EOS
+  end
+
+  it "should get schema info as RDoc" do
+    klass = mock_class(:users, :id, [
+                                     mock_column(:id, :integer),
+                                     mock_column(:name, :string, :limit => 50)
+                                    ])
+    ENV.stub!(:[]).with('format_rdoc').and_return(true)
+    AnnotateModels.get_schema_info(klass, AnnotateModels::PREFIX).should eql(<<-EOS)
+# #{AnnotateModels::PREFIX}
+#
+# Table name: users
+#
+# *id*::   <tt>integer, not null, primary key</tt>
+# *name*:: <tt>string(50), not null</tt>
+#--
+# #{AnnotateModels::END_MARK}
+#++
 
+EOS
   end
 
   describe "#get_model_class" do
-    module ::ActiveRecord
-      class Base
-      end
-    end
 
     def create(file, body="hi")
-      File.open(@dir + '/' + file, "w") do |f|
+      path = @dir + '/' + file
+      File.open(path, "w") do |f|
         f.puts(body)
       end
+      path
     end
 
     before :all do
-      require "tmpdir"
-      @dir = Dir.tmpdir + "/#{Time.now.to_i}" + "/annotate_models"
+      @dir = File.join Dir.tmpdir, "annotate_models"
       FileUtils.mkdir_p(@dir)
       AnnotateModels.model_dir = @dir
+
       create('foo.rb', <<-EOS)
         class Foo < ActiveRecord::Base
         end
@@ -69,31 +103,42 @@ EOS
           acts_as_awesome :yah
         end
       EOS
+      create('foo_with_utf8.rb', <<-EOS)
+        #encoding: utf-8
+        class FooWithUtf8 < ActiveRecord::Base
+          UTF8STRINGS = %w[résumé façon âge]
+        end
+      EOS
     end
+
     it "should work" do
       klass = AnnotateModels.get_model_class("foo.rb")
       klass.name.should == "Foo"
     end
+
     it "should not care about unknown macros" do
       klass = AnnotateModels.get_model_class("foo_with_macro.rb")
       klass.name.should == "FooWithMacro"
     end
+
+    it "should not complain of invalid multibyte char (USASCII)" do
+      klass = AnnotateModels.get_model_class("foo_with_utf8.rb")
+      klass.name.should == "FooWithUtf8"
+    end
   end
 
   describe "#remove_annotation_of_file" do
     def create(file, body="hi")
-      File.open(@dir + '/' + file, "w") do |f|
+      File.open(file, "w") do |f|
         f.puts(body)
       end
     end
+
     def content(file)
-      File.read(@dir + '/' + file)
+      File.read(file)
     end
 
-    before :all do
-      require "tmpdir"
-      @dir = Dir.tmpdir + "/#{Time.now.to_i}" + "/annotate_models"
-      FileUtils.mkdir_p(@dir)
+    it "should remove before annotate" do
       create("before.rb", <<-EOS)
 # == Schema Information
 #
@@ -107,6 +152,16 @@ EOS
 class Foo < ActiveRecord::Base
 end
       EOS
+
+      AnnotateModels.remove_annotation_of_file("before.rb")
+
+      content("before.rb").should == <<-EOS
+class Foo < ActiveRecord::Base
+end
+      EOS
+    end
+
+    it "should remove after annotate" do
       create("after.rb", <<-EOS)
 class Foo < ActiveRecord::Base
 end
@@ -121,21 +176,55 @@ end
 #
 
       EOS
-    end
-    it "should remove before annotate" do
-      AnnotateModels.remove_annotation_of_file(@dir + '/' + "before.rb")
-      content("before.rb").should == <<-EOS
+
+      AnnotateModels.remove_annotation_of_file("after.rb")
+
+      content("after.rb").should == <<-EOS
 class Foo < ActiveRecord::Base
 end
       EOS
     end
-    it "should remove after annotate" do
-      AnnotateModels.remove_annotation_of_file(@dir + '/' + "after.rb")
-      content("after.rb").should == <<-EOS
-class Foo < ActiveRecord::Base
+  end
+
+  describe "annotating a file" do
+    before do
+      @file_name    = "user.rb"
+      @file_content = <<-EOS
+class User < ActiveRecord::Base
 end
       EOS
+      File.open(@file_name, "wb") { |f| f.write @file_content }
+      @klass = mock_class(:users, :id, [
+                                        mock_column(:id, :integer),
+                                        mock_column(:name, :string, :limit => 50)
+                                       ])
+      @schema_info = AnnotateModels.get_schema_info(@klass, "== Schema Info")
+    end
+
+    it "should annotate the file before the model if position == 'before'" do
+      AnnotateModels.annotate_one_file(@file_name, @schema_info, :position => "before")
+      File.read(@file_name).should == "#{@schema_info}#{@file_content}"
     end
-  end
 
+    it "should annotate before if given :position => :before" do
+      AnnotateModels.annotate_one_file(@file_name, @schema_info, :position => :before)
+      File.read(@file_name).should == "#{@schema_info}#{@file_content}"
+    end
+
+    it "should annotate before if given :position => :after" do
+      AnnotateModels.annotate_one_file(@file_name, @schema_info, :position => :after)
+      File.read(@file_name).should == "#{@file_content}\n#{@schema_info}"
+    end
+
+    it "should update annotate position" do
+      AnnotateModels.annotate_one_file(@file_name, @schema_info, :position => :before)
+
+      another_schema_info = AnnotateModels.get_schema_info(mock_class(:users, :id, [mock_column(:id, :integer),]),
+                                                           "== Schema Info")
+
+      AnnotateModels.annotate_one_file(@file_name, another_schema_info, :position => :after)
+
+      File.read(@file_name).should == "#{@file_content}\n#{another_schema_info}"
+    end
+  end
 end

spec/spec.opts

+--format=specdoc
 --colour
\ No newline at end of file

spec/spec_helper.rb

-begin
-  require 'spec'
-rescue LoadError
-  require 'rubygems'
-  gem 'rspec'
-  require 'spec'
-end
-
+require 'rubygems'
+require 'spec'
 $:.unshift(File.dirname(__FILE__) + '/../lib')
 require 'annotate'

todo.txt

 TODO
 -----
-change default position back to "top" for all
 add "top" and "bottom" as synonyms for "before" and "after"
 change 'exclude' to 'only' (double negatives are not unconfusing)