mongo_mapper_generators 1.0.0

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Steve Agalloco
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,61 @@
+= mongo_mapper_generators
+
+MMG is intended for use with the MongoMapper rubygem.  Use it to generate Document and EmbeddedDocument models.  Also included is a generator for a rails initializer.
+
+== Install
+
+MongoMapperGenerators is only released on gemcutter. To install, you can setup gemcutter as your default gem source.
+
+ $ gem install gemcutter
+ $ gem tumble
+ 
+Then you can install MMG:
+ 
+ $ gem install mongo_mapper_generators
+
+You can also just declare the source:
+
+ $ gem install mongo_mapper_generators -s http://gemcutter.org
+
+== Dependencies
+
+* MongoMapper
+* Rails
+
+== Usage
+
+Once you install the gem, the generators will be available to all Rails
+applications on your system. If you run script/generate without any
+additional arguments you should see the available generators listed.
+
+To run the generator, go to your rails project directory and call it
+using the script/generate or script/destroy command.
+
+  script/generate mongo_model Person name:string email:string age:integer
+  
+It will generate the model, a unit test and a factory.
+
+Generators also exist for generating EmbeddedDocument models as well as a rails initializer.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Credits
+
+I was inspired by a number of different sources, including:
+
+* Blitz - http://github.com/dancroak/blitz
+* NiftyGenerators - http://github.com/ryanb/nifty-generators
+* ShouldaGenerator - http://github.com/technicalpickles/shoulda_generator
+* John Nunemaker's rails initializer - http://gist.github.com/232953
+
+== Copyright
+
+Copyright (c) 2009 Steve Agalloco. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,67 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "mongo_mapper_generators"
+    gem.summary = %Q{Generators for use with the MongoMapper rubygem}
+    gem.description = %Q{Generators for use with the MongoMapper rubygem.  Use to generate Documents, EmbeddedDocuments, and Rails initializer.}
+    gem.email = "steve.agalloco@gmail.com"
+    gem.homepage = "http://github.com/spagalloco/mongo_mapper_generators"
+    gem.authors = ["Steve Agalloco"]
+    gem.files = FileList["[A-Z]*", "{lib,rails_generators,test}/**/*"]
+    
+    gem.add_development_dependency "shoulda", ">= 2.10.2"
+    gem.add_development_dependency "cucumber", ">= 0.4.0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+begin
+  require 'cucumber/rake/task'
+  Cucumber::Rake::Task.new(:features)
+
+  task :features => :check_dependencies
+rescue LoadError
+  task :features do
+    abort "Cucumber is not available. In order to run features, you must: sudo gem install cucumber"
+  end
+end
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "mongo_mapper_generators #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+1.0.0

data/lib/mongo_mapper_generators.rb

@@ -0,0 +1,5 @@
+module MongomapperGenerators
+  VERSION = '0.1.0'
+  
+  # nothing to see here, the real action is under rails_generators
+end

data/rails_generators/embedded_model/USAGE

@@ -0,0 +1,24 @@
+Description:
+    Stubs out a new model. Pass the model name, either CamelCased or
+    under_scored, and an optional list of attribute pairs as arguments.
+
+    Attribute pairs are key_name:mongo_type arguments specifying the
+    model's attributes. Timestamps and Userstamps are added by default.
+
+Examples:
+    `./script/generate embedded_model account`
+
+        creates an Account model:
+            Model:      app/models/account.rb
+
+    `./script/generate embedded_model post title:string body:string published:boolean`
+
+        creates a Post model with a string title, string body, published flag, timestamps! and userstamps!.
+
+    `./script/generate embedded_model post --skip-timestamps`
+
+        creates a Post model with no timestamps.
+
+    `./script/generate embedded_model post --skip-userstamps`
+
+        creates a Post model with no userstamps.

data/rails_generators/embedded_model/embedded_model_generator.rb

@@ -0,0 +1,64 @@
+require File.join(File.dirname(__FILE__), "..", "support", "generator_helper")
+
+class EmbeddedModelGenerator < Rails::Generator::NamedBase
+  attr_accessor :belongs, :many, :timestamps, :userstamps
+  
+  default_options :skip_factories  => false, :skip_timestamps => false
+  
+  def initialize(runtime_args, runtime_options = {})
+    super
+    
+    parsed_attributes = []
+    
+    @args.each do |arg|
+      if arg.include? ':'
+        parsed_attributes << MongoAttribute.new(*arg.split(":"))
+      end
+    end
+    
+    @attributes = parsed_attributes.reject { |each| %w(many belongs_to).include? each.name }
+    @many = parsed_attributes.select { |each| each.name == 'many' }
+    @belongs = parsed_attributes.select { |each| each.name == 'belongs_to' }
+    @timestamps = !options[:skip_timestamps]
+    @userstamps = !options[:skip_userstamps]
+  end
+  
+  def manifest
+    record do |m|
+      m.directory 'app/models'
+      m.template 'embedded_model.rb', File.join('app/models', class_path, "#{file_name}.rb")
+      
+      m.directory 'test/unit'
+      m.template 'unit_test.rb', File.join('test/unit', class_path, "#{file_name}_test.rb")
+
+      unless options[:skip_factories]
+        m.directory 'test/factories'
+        m.template 'factory.rb', File.join('test/factories', "#{plural_name}.rb")
+      end
+    end
+  end
+
+  def factory_line(attribute, file_name)
+    "#{file_name}.#{attribute.name} #{attribute.default_for_factory}"
+  end
+  
+  protected
+
+    def banner
+      "Usage: #{$0} #{spec.name} ModelName [field:type, field:type] --skip-factories --skip-timestamps --skip-userstamps"
+    end
+
+    def add_options!(opt)
+      opt.separator ''
+      opt.separator 'Options:'
+      opt.on("--skip-factories", "Don't generate a factory for this model") { |v| 
+        options[:skip_factories] = v 
+      }
+      opt.on("--skip-timestamps", "Don't add timestamps to this model") { |v| 
+        options[:skip_timestamps] = v 
+      }
+      opt.on("--skip-userstamps", "Don't add userstamps to this model") { |v| 
+        options[:skip_userstamps] = v 
+      }
+    end
+end

data/rails_generators/embedded_model/templates/embedded_model.rb

@@ -0,0 +1,29 @@
+class <%= class_name %>
+  include MongoMapper::EmbeddedDocument
+
+<% if attributes.any? -%>
+<% attributes.each do |attribute| -%>
+  key :<%= attribute.name %>, <%= attribute.type %>
+<% end -%>
+<% end -%>
+
+<% if many.any? -%>
+<% many.each do |each| -%>
+  many :<%= each.type %>
+<% end -%>
+<% end -%>
+
+<% if belongs.any? -%>
+<% belongs.each do |each| -%>
+  belongs_to :<%= each.type %>
+<% end -%>
+<% end -%>
+
+<% if timestamps -%>
+  timestamps!
+<% end -%>
+
+<% if userstamps -%>
+  userstamps!
+<% end -%>
+end

data/rails_generators/embedded_model/templates/factory.rb

@@ -0,0 +1,9 @@
+Factory.define :<%= file_name %> do |<%= file_name %>|
+<% attributes.each do |attribute| -%>
+  <%= factory_line(attribute, file_name) %>
+<% end -%>
+<% if timestamps %>
+  <%= factory_line(MongoAttribute.new('created_at', 'timestamp'), file_name) %>
+  <%= factory_line(MongoAttribute.new('updated_at', 'timestamp'), file_name) %>
+<% end -%>
+end

data/rails_generators/embedded_model/templates/unit_test.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class <%= class_name %>Test < ActiveSupport::TestCase
+  should "be valid with factory" do
+    assert_valid Factory.build(:<%= file_name -%>)
+  end
+end

data/rails_generators/mongo_initializer/USAGE

@@ -0,0 +1,14 @@
+Description:
+    Creates a database.yml file and a rails initializer for use with MongoMapper.
+
+Examples:
+    `./script/generate mongo_initializer`
+
+        creates an initializer:
+            Initializer:      config/initializers/mongomapper.rb
+			Config:			  config/database.yml
+			
+
+    `./script/generate mongo_initializer databasename`
+
+		generates the same files as above, and uses the databasename in the config file.

data/rails_generators/mongo_initializer/mongo_initializer_generator.rb

@@ -0,0 +1,18 @@
+class MongoInitializerGenerator < Rails::Generator::Base
+  attr :file_name 
+  
+  def initialize(runtime_args, runtime_options = {})
+    super
+    @file_name = @args.first || 'default'
+  end
+
+  def manifest
+    record do |m|
+      m.directory 'config'
+      m.directory 'config/initializers'
+      
+      m.file 'database.yml', "config/database.yml"      
+      m.template 'initializer.rb', "config/initializers/mongomapper.rb"
+    end
+  end
+end

data/rails_generators/mongo_initializer/templates/database.yml

@@ -0,0 +1,15 @@
+development: &global_settings
+  database: <%= file_name %>_development
+  host: 127.0.0.1
+  port: 27017
+
+test:
+  database: <%= file_name %>_test
+  <<: *global_settings
+
+production:
+  host: hostname
+  database: <%= file_name %>_production
+  username: username
+  password: password
+  <<: *global_settings

data/rails_generators/mongo_initializer/templates/initializer.rb

@@ -0,0 +1,25 @@
+config = YAML.load_file(File.join(Rails.root,'config','database.yml'))[Rails.env]
+
+MongoMapper.connection = Mongo::Connection.new(config['host'], config['port'], {
+  :logger         => Rails.logger
+})
+
+MongoMapper.database = config['database']
+if config['username'].present?
+  MongoMapper.database.authenticate(config['username'], config['password'])
+end
+
+Dir[Rails.root + 'app/models/**/*.rb'].each do |model_path|
+  File.basename(model_path, '.rb').classify.constantize
+end
+MongoMapper.ensure_indexes!
+
+ActionController::Base.rescue_responses['MongoMapper::DocumentNotFound'] = :not_found
+
+if defined?(PhusionPassenger)
+  PhusionPassenger.on_event(:starting_worker_process) do |forked|
+    # if using older than 0.6.5 of MM then you want database instead of connection
+    # MongoMapper.database.connect_to_master if forked
+    MongoMapper.connection.connect_to_master if forked
+  end
+end

data/rails_generators/mongo_model/USAGE

@@ -0,0 +1,24 @@
+Description:
+    Stubs out a new model. Pass the model name, either CamelCased or
+    under_scored, and an optional list of attribute pairs as arguments.
+
+    Attribute pairs are key_name:mongo_type arguments specifying the
+    model's attributes. Timestamps and Userstamps are added by default.
+
+Examples:
+    `./script/generate mongo_model account`
+
+        creates an Account model:
+            Model:      app/models/account.rb
+
+    `./script/generate mongo_model post title:string body:string published:boolean`
+
+        creates a Post model with a string title, string body, published flag, timestamps! and userstamps!.
+
+    `./script/generate mongo_model post --skip-timestamps`
+
+        creates a Post model with no timestamps.
+
+    `./script/generate mongo_model post --skip-userstamps`
+
+        creates a Post model with no userstamps.

data/rails_generators/mongo_model/mongo_model_generator.rb

@@ -0,0 +1,64 @@
+require File.join(File.dirname(__FILE__), "..", "support", "generator_helper")
+
+class MongoModelGenerator < Rails::Generator::NamedBase
+  attr_accessor :belongs, :many, :timestamps, :userstamps
+  
+  default_options :skip_factories  => false, :skip_timestamps => false
+  
+  def initialize(runtime_args, runtime_options = {})
+    super
+    
+    parsed_attributes = []
+    
+    @args.each do |arg|
+      if arg.include? ':'
+        parsed_attributes << MongoAttribute.new(*arg.split(":"))
+      end
+    end
+    
+    @attributes = parsed_attributes.reject { |each| %w(many belongs_to).include? each.name }
+    @many = parsed_attributes.select { |each| each.name == 'many' }
+    @belongs = parsed_attributes.select { |each| each.name == 'belongs_to' }
+    @timestamps = !options[:skip_timestamps]
+    @userstamps = !options[:skip_userstamps]
+  end
+  
+  def manifest
+    record do |m|
+      m.directory 'app/models'
+      m.template 'mongo_model.rb', "app/models/#{singular_name}.rb"
+      
+      m.directory 'test/unit'
+      m.template 'unit_test.rb',  File.join('test/unit', class_path, "#{singular_name}_test.rb")
+
+      unless options[:skip_factories]
+        m.directory 'test/factories'
+        m.template 'factory.rb', File.join('test/factories', "#{plural_name}.rb")
+      end
+    end
+  end
+  
+  def factory_line(attribute, file_name)
+    "#{file_name}.#{attribute.name} #{attribute.default_for_factory}"
+  end  
+
+  protected
+
+    def banner
+      "Usage: #{$0} #{spec.name} ModelName [field:type, field:type] --skip-factories --skip-timestamps --skip-userstamps"
+    end
+
+    def add_options!(opt)
+      opt.separator ''
+      opt.separator 'Options:'
+      opt.on("--skip-factories", "Don't generate a factory for this model") { |v| 
+        options[:skip_factories] = v 
+      }
+      opt.on("--skip-timestamps", "Don't add timestamps to this model") { |v| 
+        options[:skip_timestamps] = v 
+      }
+      opt.on("--skip-userstamps", "Don't add userstamps to this model") { |v| 
+        options[:skip_userstamps] = v 
+      }
+    end
+end

data/rails_generators/mongo_model/templates/factory.rb

@@ -0,0 +1,9 @@
+Factory.define :<%= file_name %> do |<%= file_name %>|
+<% attributes.each do |attribute| -%>
+  <%= factory_line(attribute, file_name) %>
+<% end -%>
+<% if timestamps %>
+  <%= factory_line(MongoAttribute.new('created_at', 'timestamp'), file_name) %>
+  <%= factory_line(MongoAttribute.new('updated_at', 'timestamp'), file_name) %>
+<% end -%>
+end

data/rails_generators/mongo_model/templates/mongo_model.rb

@@ -0,0 +1,29 @@
+class <%= class_name %>
+  include MongoMapper::Document
+
+<% if attributes.any? -%>
+<% attributes.each do |attribute| -%>
+  key :<%= attribute.name %>, <%= attribute.type %>
+<% end -%>
+<% end -%>
+
+<% if many.any? -%>
+<% many.each do |each| -%>
+  many :<%= each.type %>
+<% end -%>
+<% end -%>
+
+<% if belongs.any? -%>
+<% belongs.each do |each| -%>
+  belongs_to :<%= each.type %>
+<% end -%>
+<% end -%>
+
+<% if timestamps -%>
+  timestamps!
+<% end -%>
+
+<% if userstamps -%>
+  userstamps!
+<% end -%>
+end

data/rails_generators/mongo_model/templates/unit_test.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class <%= class_name %>Test < ActiveSupport::TestCase
+  should "be valid with factory" do
+    assert_valid Factory.build(:<%= file_name -%>)
+  end
+end

data/rails_generators/support/generator_helper.rb

@@ -0,0 +1,33 @@
+class MongoAttribute
+  attr_accessor :name, :type
+  
+  def initialize(name, type)
+    @name, @type = name, type
+  end
+  
+  def type
+    case @name
+    when 'many'
+      @type.tableize
+    when 'index', 'belongs_to'
+      @type.downcase
+    else
+      @type.capitalize
+    end
+  end
+  
+  def default_for_factory
+    @default ||= case @type.downcase.to_sym
+      when :integer                     then "{ 1 }"
+      when :float                       then "{ 1.5 }"
+      when :decimal                     then "{ 9.99 }"
+      when :datetime, :timestamp, :time then "{ Time.now.to_s(:db) }"
+      when :date                        then "{ Date.today.to_s(:db) }"
+      when :string                      then "{ 'string' }"
+      when :text                        then "{ 'text' }"
+      when :boolean                     then "{ false }"
+      else
+        ""
+    end
+  end
+end

data/test/fixtures/tmp/test.log

@@ -0,0 +1 @@
+# Logfile created on Tue Dec 22 00:34:12 -0500 2009 by logger.rb/22283

data/test/helper.rb

@@ -0,0 +1,25 @@
+require 'rubygems'
+require 'ruby-debug'
+require 'active_support'
+require 'active_record'
+require 'action_controller'
+require 'action_view'
+require 'shoulda'
+
+require File.join(File.dirname(__FILE__), 'shoulda_macros', 'generator_macros')
+require File.join(File.dirname(__FILE__), 'stolen_from_railties')
+
+unless defined?(RAILS_DEFAULT_LOGGER)
+  @test_log = File.join(RAILS_ROOT, 'test.log')
+  RAILS_DEFAULT_LOGGER = Logger.new(@test_log)
+end
+
+Rails::Generator::Base.prepend_sources Rails::Generator::PathSource.new(:mongo_mapper_generators, File.join(File.dirname(__FILE__), "..", "rails_generators"))
+
+class GeneratorTestCase
+  def run_generator(name, params=[])
+    silence_generator do
+      build_generator(name, params).command(:create).invoke!
+    end
+  end
+end

data/test/shoulda_macros/generator_macros.rb

@@ -0,0 +1,18 @@
+class Test::Unit::TestCase
+  class << self
+    
+    def should_generate_file(file, &block)
+      should "generate file #{file}" do
+        yield("foo") if block_given?
+        assert_generated_file(file)
+      end
+    end
+    
+    def should_not_generate_file(file)
+      should "not generate file #{file}" do
+        assert !File.exists?(file),"The file '#{file}' should not exist"
+      end
+    end
+
+  end
+end

data/test/stolen_from_railties.rb

@@ -0,0 +1,287 @@
+require 'test/unit'
+require 'fileutils'
+
+# Mock out what we need from AR::Base
+module ActiveRecord
+  class Base
+    class << self
+      attr_accessor :pluralize_table_names, :timestamped_migrations
+    end
+    self.pluralize_table_names = true
+    self.timestamped_migrations = true
+  end
+
+  module ConnectionAdapters
+    class Column
+      attr_reader :name, :default, :type, :limit, :null, :sql_type, :precision, :scale
+
+      def initialize(name, default, sql_type = nil)
+        @name = name
+        @default = default
+        @type = @sql_type = sql_type
+      end
+
+      def human_name
+        @name.humanize
+      end
+    end
+  end
+end
+
+# Mock up necessities from ActionView
+module ActionView
+  module Helpers
+    module ActionRecordHelper; end
+    class InstanceTag; end
+  end
+end
+
+# Set RAILS_ROOT appropriately fixture generation
+tmp_dir = "#{File.dirname(__FILE__)}/fixtures/tmp"
+
+if defined? RAILS_ROOT
+  RAILS_ROOT.replace tmp_dir
+else
+  RAILS_ROOT = tmp_dir
+end
+FileUtils.mkdir_p RAILS_ROOT
+
+require 'initializer'
+
+# Mocks out the configuration
+module Rails
+  def self.configuration
+    Rails::Configuration.new
+  end
+end
+
+require 'rails_generator'
+
+class GeneratorTestCase < Test::Unit::TestCase
+  include FileUtils
+
+  def setup
+    ActiveRecord::Base.pluralize_table_names = true
+
+    mkdir_p "#{RAILS_ROOT}/app/views/layouts"
+    mkdir_p "#{RAILS_ROOT}/config"
+    mkdir_p "#{RAILS_ROOT}/db"
+    mkdir_p "#{RAILS_ROOT}/test/fixtures"
+    mkdir_p "#{RAILS_ROOT}/public/stylesheets"
+
+    File.open("#{RAILS_ROOT}/config/routes.rb", 'w') do |f|
+      f << "ActionController::Routing::Routes.draw do |map|\n\nend"
+    end
+  end
+
+  def teardown
+    rm_rf "#{RAILS_ROOT}/app"
+    rm_rf "#{RAILS_ROOT}/test"
+    rm_rf "#{RAILS_ROOT}/config"
+    rm_rf "#{RAILS_ROOT}/db"
+    rm_rf "#{RAILS_ROOT}/public"
+  end
+
+  def test_truth
+    # don't complain, test/unit
+  end
+
+  # Instantiates the Generator.
+  def build_generator(name, params)
+    Rails::Generator::Base.instance(name, params)
+  end
+
+  # Runs the +create+ command (like the command line does).
+  def run_generator(name, params)
+    silence_generator do
+      build_generator(name, params).command(:create).invoke!
+    end
+  end
+
+  # Silences the logger temporarily and returns the output as a String.
+  def silence_generator
+    logger_original = Rails::Generator::Base.logger
+    myout = StringIO.new
+    Rails::Generator::Base.logger = Rails::Generator::SimpleLogger.new(myout)
+    yield if block_given?
+    Rails::Generator::Base.logger = logger_original
+    myout.string
+  end
+
+  # Asserts that the given controller was generated.
+  # It takes a name or symbol without the <tt>_controller</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_controller_for(name, parent = "ApplicationController")
+    assert_generated_class "app/controllers/#{name.to_s.underscore}_controller", parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given model was generated.
+  # It takes a name or symbol and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_model_for(name, parent = "ActiveRecord::Base")
+    assert_generated_class "app/models/#{name.to_s.underscore}", parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given helper was generated.
+  # It takes a name or symbol without the <tt>_helper</tt> part.
+  # The contents of the module source file is passed to a block.
+  def assert_generated_helper_for(name)
+    assert_generated_module "app/helpers/#{name.to_s.underscore}_helper" do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given functional test was generated.
+  # It takes a name or symbol without the <tt>_controller_test</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_functional_test_for(name, parent = "ActionController::TestCase")
+    assert_generated_class "test/functional/#{name.to_s.underscore}_controller_test",parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given unit test was generated.
+  # It takes a name or symbol without the <tt>_test</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_unit_test_for(name, parent = "ActiveSupport::TestCase")
+    assert_generated_class "test/unit/#{name.to_s.underscore}_test", parent do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given file was generated.
+  # The contents of the file is passed to a block.
+  def assert_generated_file(path)
+    assert_file_exists(path)
+    File.open("#{RAILS_ROOT}/#{path}") do |f|
+      yield f.read if block_given?
+    end
+  end
+
+  # asserts that the given file exists
+  def assert_file_exists(path)
+    assert File.exist?("#{RAILS_ROOT}/#{path}"),
+      "The file '#{RAILS_ROOT}/#{path}' should exist"
+  end
+
+  # Asserts that the given class source file was generated.
+  # It takes a path without the <tt>.rb</tt> part and an optional super class.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_class(path, parent = nil)
+    # FIXME: Sucky way to detect namespaced classes
+    if path.split('/').size > 3
+      path =~ /\/?(\d+_)?(\w+)\/(\w+)$/
+      class_name = "#{$2.camelize}::#{$3.camelize}"
+    else
+      path =~ /\/?(\d+_)?(\w+)$/
+      class_name = $2.camelize
+    end
+
+    assert_generated_file("#{path}.rb") do |body|
+      assert_match /class #{class_name}#{parent.nil? ? '':" < #{parent}"}/, body, "the file '#{path}.rb' should be a class"
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given module source file was generated.
+  # It takes a path without the <tt>.rb</tt> part.
+  # The contents of the class source file is passed to a block.
+  def assert_generated_module(path)
+    # FIXME: Sucky way to detect namespaced modules
+    if path.split('/').size > 3
+      path =~ /\/?(\w+)\/(\w+)$/
+      module_name = "#{$1.camelize}::#{$2.camelize}"
+    else
+      path =~ /\/?(\w+)$/
+      module_name = $1.camelize
+    end
+
+    assert_generated_file("#{path}.rb") do |body|
+      assert_match /module #{module_name}/, body, "the file '#{path}.rb' should be a module"
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given CSS stylesheet file was generated.
+  # It takes a path without the <tt>.css</tt> part.
+  # The contents of the stylesheet source file is passed to a block.
+  def assert_generated_stylesheet(path)
+    assert_generated_file("public/stylesheets/#{path}.css") do |body|
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given YAML file was generated.
+  # It takes a path without the <tt>.yml</tt> part.
+  # The parsed YAML tree is passed to a block.
+  def assert_generated_yaml(path)
+    assert_generated_file("#{path}.yml") do |body|
+      yaml = YAML.load(body)
+      assert yaml, 'YAML data missing'
+      yield yaml if block_given?
+    end
+  end
+
+  # Asserts that the given fixtures YAML file was generated.
+  # It takes a fixture name without the <tt>.yml</tt> part.
+  # The parsed YAML tree is passed to a block.
+  def assert_generated_fixtures_for(name)
+    assert_generated_yaml "test/fixtures/#{name.to_s.underscore}" do |yaml|
+      yield yaml if block_given?
+    end
+  end
+
+  # Asserts that the given views were generated.
+  # It takes a controller name and a list of views (including extensions).
+  # The body of each view is passed to a block.
+  def assert_generated_views_for(name, *actions)
+    actions.each do |action|
+      assert_generated_file("app/views/#{name.to_s.underscore}/#{action}") do |body|
+        yield body if block_given?
+      end
+    end
+  end
+
+  def assert_generated_migration(name, parent = "ActiveRecord::Migration")
+    file = Dir.glob("#{RAILS_ROOT}/db/migrate/*_#{name.to_s.underscore}.rb").first
+    file = file.match(/db\/migrate\/[0-9]+_\w+/).to_s
+    assert_generated_class file, parent do |body|
+      assert_match /timestamps/, body, "should have timestamps defined"
+      yield body if block_given?
+    end
+  end
+
+  # Asserts that the given migration file was not generated.
+  # It takes the name of the migration as a parameter.
+  def assert_skipped_migration(name)
+    migration_file = "#{RAILS_ROOT}/db/migrate/001_#{name.to_s.underscore}.rb"
+    assert !File.exist?(migration_file), "should not create migration #{migration_file}"
+  end
+
+  # Asserts that the given resource was added to the routes.
+  def assert_added_route_for(name)
+    assert_generated_file("config/routes.rb") do |body|
+      assert_match /map.resources :#{name.to_s.underscore}/, body,
+        "should add route for :#{name.to_s.underscore}"
+    end
+  end
+
+  # Asserts that the given methods are defined in the body.
+  # This does assume standard rails code conventions with regards to the source code.
+  # The body of each individual method is passed to a block.
+  def assert_has_method(body, *methods)
+    methods.each do |name|
+      assert body =~ /^  def #{name}(\(.+\))?\n((\n|   .*\n)*)  end/, "should have method #{name}"
+      yield(name, $2) if block_given?
+    end
+  end
+
+  # Asserts that the given column is defined in the migration.
+  def assert_generated_column(body, name, type)
+    assert_match /t\.#{type.to_s} :#{name.to_s}/, body, "should have column #{name.to_s} defined"
+  end
+end

data/test/test_mongo_mapper_generators.rb

@@ -0,0 +1,46 @@
+require 'helper'
+
+class TestMongoMapperGenerators < GeneratorTestCase
+  
+  context "running the mongo_initializer generator" do
+    setup do
+      run_generator('mongo_initializer')
+    end
+    
+    should_generate_file 'config/initializers/mongomapper.rb'
+    should_generate_file 'config/database.yml'
+  end
+  
+  context "running the mongo_model generator" do
+    
+    context "with factories" do    
+      setup do
+        run_generator('mongo_model', %w(Person name:string email:string many:addresses index:email belongs_to:corporation))
+      end
+    
+      should_generate_file 'app/models/person.rb'
+      should_generate_file 'test/unit/person_test.rb'
+      should_generate_file 'test/factories/people.rb'
+    end
+    
+    context "without factories" do
+      setup do
+        run_generator('mongo_model', %w(Person name:string email:string many:addresses index:email belongs_to:corporation --skip-factories))
+      end
+    
+      should_generate_file 'app/models/person.rb'
+      should_generate_file 'test/unit/person_test.rb'
+      should_not_generate_file 'test/factories/people.rb'
+    end
+  end
+
+  context "running the embedded_model generator" do
+    setup do
+      run_generator('embedded_model', %w(Address street:string state:integer city:string zip:string))
+    end
+    
+    should_generate_file 'app/models/address.rb'
+    should_generate_file 'test/unit/address_test.rb'
+    should_generate_file 'test/factories/addresses.rb'
+  end
+end

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification 
+name: mongo_mapper_generators
+version: !ruby/object:Gem::Version 
+  version: 1.0.0
+platform: ruby
+authors: 
+- Steve Agalloco
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-01-15 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: shoulda
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.10.2
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: cucumber
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.4.0
+    version: 
+description: Generators for use with the MongoMapper rubygem.  Use to generate Documents, EmbeddedDocuments, and Rails initializer.
+email: steve.agalloco@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/mongo_mapper_generators.rb
+- rails_generators/embedded_model/USAGE
+- rails_generators/embedded_model/embedded_model_generator.rb
+- rails_generators/embedded_model/templates/embedded_model.rb
+- rails_generators/embedded_model/templates/factory.rb
+- rails_generators/embedded_model/templates/unit_test.rb
+- rails_generators/mongo_initializer/USAGE
+- rails_generators/mongo_initializer/mongo_initializer_generator.rb
+- rails_generators/mongo_initializer/templates/database.yml
+- rails_generators/mongo_initializer/templates/initializer.rb
+- rails_generators/mongo_model/USAGE
+- rails_generators/mongo_model/mongo_model_generator.rb
+- rails_generators/mongo_model/templates/factory.rb
+- rails_generators/mongo_model/templates/mongo_model.rb
+- rails_generators/mongo_model/templates/unit_test.rb
+- rails_generators/support/generator_helper.rb
+- test/fixtures/tmp/test.log
+- test/helper.rb
+- test/shoulda_macros/generator_macros.rb
+- test/stolen_from_railties.rb
+- test/test_mongo_mapper_generators.rb
+has_rdoc: true
+homepage: http://github.com/spagalloco/mongo_mapper_generators
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Generators for use with the MongoMapper rubygem
+test_files: 
+- test/helper.rb
+- test/shoulda_macros/generator_macros.rb
+- test/stolen_from_railties.rb
+- test/test_mongo_mapper_generators.rb