test.rb

class DslTest < Test::Unit::TestCase
  context "The DSL" do
    before do
      @runner = Foundry::Runner.new
      
      # Record is a simple ActiveRecord::Base class. Save its current public methods
      # so new methods can be removed in +after+.
      @original_record_public_methods = Record.public_methods
    end
    
    # Write tests here that add named scopes to Record.
    it "should create factory scopes using the default name if one is not given" do
      @runner.model Record do
        factory do
          {:title => "AWESOME."}
        end
      end

      assert Record.respond_to?(:valid)
    end

    after do
      # Remove factory methods from Record so the same method name can be added
      # in another test (like 'valid').
      (Record.public_methods - @original_record_public_methods -
      
        # Record gets a few methods added to it that I don't want removed.
        ['with', 'original_primary_key', 'original_locking_column']
      ).each do |method|
        Record.class_eval <<-EOS
          class << self
            remove_method(#{method.to_sym.inspect})
          end
        EOS
      end
    end
  end
end