1 files changed, 48 insertions, 0 deletions
diff --git a/railties/test/rails_generator/rails_model_generator_test.rb b/railties/test/rails_generator/rails_model_generator_test.rb
new file mode 100644
index 0000000000..aea2abafba
--- /dev/null
+++ b/railties/test/rails_generator/rails_model_generator_test.rb
@@ -0,0 +1,48 @@
+require 'generators/generator_test_helper'
+
+class RailsModelGeneratorTest < GeneratorTestCase
+
+  def test_model_generates_resources
+    run_generator('model', %w(Product name:string))
+
+    assert_generated_model_for :product
+    assert_generated_fixtures_for :products
+    assert_generated_migration :create_products
+  end
+
+  def test_model_skip_migration_skips_migration
+    run_generator('model', %w(Product name:string --skip-migration))
+
+    assert_generated_model_for :product
+    assert_generated_fixtures_for :products
+    assert_skipped_migration :create_products
+  end
+
+  def test_model_with_attributes_generates_resources_with_attributes
+    run_generator('model', %w(Product name:string supplier_id:integer created_at:timestamp))
+
+    assert_generated_model_for :product
+    assert_generated_fixtures_for :products
+    assert_generated_migration :create_products do |t|
+      assert_generated_column t, :name, :string
+      assert_generated_column t, :supplier_id, :integer
+      assert_generated_column t, :created_at, :timestamp
+    end
+  end
+
+  def test_model_with_reference_attributes_generates_belongs_to_associations
+    run_generator('model', %w(Product name:string supplier:references))
+
+    assert_generated_model_for :product do |body|
+      assert body =~ /^\s+belongs_to :supplier/, "#{body.inspect} should contain 'belongs_to :supplier'"
+    end
+  end
+
+  def test_model_with_belongs_to_attributes_generates_belongs_to_associations
+    run_generator('model', %w(Product name:string supplier:belongs_to))
+
+    assert_generated_model_for :product do |body|
+      assert body =~ /^\s+belongs_to :supplier/, "#{body.inspect} should contain 'belongs_to :supplier'"
+    end
+  end
+end