1 files changed, 24 insertions, 0 deletions
diff --git a/railties/lib/generators/test_unit/model/model_generator.rb b/railties/lib/generators/test_unit/model/model_generator.rb
new file mode 100644
index 0000000000..901ed70468
--- /dev/null
+++ b/railties/lib/generators/test_unit/model/model_generator.rb
@@ -0,0 +1,24 @@
+module TestUnit
+  module Generators
+    class ModelGenerator < Base
+      argument :attributes, :type => :hash, :default => {}, :banner => "field:type, field:type"
+
+      check_class_collision :suffix => "Test"
+
+      # TODO Add DEFAULTS support
+      class_option :skip_fixture, :type => :boolean, :default => false,
+                   :desc => "Don't generate a fixture file"
+
+      def create_test_file
+        template 'unit_test.rb', File.join('test/unit', class_path, "#{file_name}_test.rb")
+      end
+
+      # TODO Add fixture replacement support
+      def create_fixture_file
+        unless options[:skip_fixture]
+          template 'fixtures.yml', File.join('test/fixtures', "#{table_name}.yml")
+        end
+      end
+    end
+  end
+end