diff options
Diffstat (limited to 'railties/lib/generators/test_unit/model/model_generator.rb')
-rw-r--r-- | railties/lib/generators/test_unit/model/model_generator.rb | 24 |
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 |