class Temping

# File lib/temping.rb, line 28
def cleanup
  @model_klasses.each(&:destroy_all)
end

# File lib/temping.rb, line 8
def create(model_name, options = {}, &block)
  factory = ModelFactory.new(model_name.to_s.classify, options, &block)
  klass = factory.klass
  @model_klasses << klass
  klass
end

# File lib/temping.rb, line 15
def teardown
  if @model_klasses.any?
    @model_klasses.each do |klass|
      if Object.const_defined?(klass.name)
        klass.connection.drop_table(klass.table_name)
        Object.send(:remove_const, klass.name)
      end
    end
    @model_klasses.clear
    ActiveSupport::Dependencies::Reference.clear!
  end
end