require 'generators/generators_test_helper' require 'rails/generators/rails/scaffold/scaffold_generator' class ScaffoldGeneratorTest < Rails::Generators::TestCase include GeneratorsTestHelper arguments %w(product_line title:string product:belongs_to user:references) setup :copy_routes def test_scaffold_on_invoke run_generator # Model assert_file "app/models/product_line.rb", /class ProductLine < ActiveRecord::Base/ assert_file "test/unit/product_line_test.rb", /class ProductLineTest < ActiveSupport::TestCase/ assert_file "test/fixtures/product_lines.yml" assert_migration "db/migrate/create_product_lines.rb", /belongs_to :product/ assert_migration "db/migrate/create_product_lines.rb", /add_index :product_lines, :product_id/ assert_migration "db/migrate/create_product_lines.rb", /references :user/ assert_migration "db/migrate/create_product_lines.rb", /add_index :product_lines, :user_id/ # Route assert_file "config/routes.rb" do |route| assert_match(/resources :product_lines$/, route) end # Controller assert_file "app/controllers/product_lines_controller.rb" do |content| assert_match(/class ProductLinesController < ApplicationController/, content) assert_instance_method :index, content do |m| assert_match(/@product_lines = ProductLine\.all/, m) end assert_instance_method :show, content do |m| assert_match(/@product_line = ProductLine\.find\(params\[:id\]\)/, m) end assert_instance_method :new, content do |m| assert_match(/@product_line = ProductLine\.new/, m) end assert_instance_method :edit, content do |m| assert_match(/@product_line = ProductLine\.find\(params\[:id\]\)/, m) end assert_instance_method :create, content do |m| assert_match(/@product_line = ProductLine\.new\(params\[:product_line\]\)/, m) assert_match(/@product_line\.save/, m) assert_match(/@product_line\.errors/, m) end assert_instance_method :update, content do |m| assert_match(/@product_line = ProductLine\.find\(params\[:id\]\)/, m) assert_match(/@product_line\.update_attributes\(params\[:product_line\]\)/, m) assert_match(/@product_line\.errors/, m) end assert_instance_method :destroy, content do |m| assert_match(/@product_line = ProductLine\.find\(params\[:id\]\)/, m) assert_match(/@product_line\.destroy/, m) end end assert_file "test/functional/product_lines_controller_test.rb", /class ProductLinesControllerTest < ActionController::TestCase/ # Views %w( index edit new show _form ).each { |view| assert_file "app/views/product_lines/#{view}.html.erb" } assert_no_file "app/views/layouts/product_lines.html.erb" # Helpers assert_file "app/helpers/product_lines_helper.rb" assert_file "test/unit/helpers/product_lines_helper_test.rb" # Stylesheets assert_file "public/stylesheets/scaffold.css" end def test_scaffold_on_revoke run_generator run_generator ["product_line"], :behavior => :revoke # Model assert_no_file "app/models/product_line.rb" assert_no_file "test/unit/product_line_test.rb" assert_no_file "test/fixtures/product_lines.yml" assert_no_migration "db/migrate/create_product_lines.rb" # Route assert_file "config/routes.rb" do |route| assert_no_match(/resources :product_lines$/, route) end # Controller assert_no_file "app/controllers/product_lines_controller.rb" assert_no_file "test/functional/product_lines_controller_test.rb" # Views assert_no_file "app/views/product_lines" assert_no_file "app/views/layouts/product_lines.html.erb" # Helpers assert_no_file "app/helpers/product_lines_helper.rb" assert_no_file "test/unit/helpers/product_lines_helper_test.rb" # Stylesheets (should not be removed) assert_file "public/stylesheets/scaffold.css" end def test_scaffold_with_namespace_on_invoke run_generator [ "admin/role", "name:string", "description:string" ] # Model assert_file "app/models/admin.rb", /module Admin/ assert_file "app/models/admin/role.rb", /class Admin::Role < ActiveRecord::Base/ assert_file "test/unit/admin/role_test.rb", /class Admin::RoleTest < ActiveSupport::TestCase/ assert_file "test/fixtures/admin/roles.yml" assert_migration "db/migrate/create_admin_roles.rb" # Route assert_file "config/routes.rb" do |route| assert_match(/namespace :admin do resources :roles end$/, route) end # Controller assert_file "app/controllers/admin/roles_controller.rb" do |content| assert_match(/class Admin::RolesController < ApplicationController/, content) assert_instance_method :index, content do |m| assert_match(/@admin_roles = Admin::Role\.all/, m) end assert_instance_method :show, content do |m| assert_match(/@admin_role = Admin::Role\.find\(params\[:id\]\)/, m) end assert_instance_method :new, content do |m| assert_match(/@admin_role = Admin::Role\.new/, m) end assert_instance_method :edit, content do |m| assert_match(/@admin_role = Admin::Role\.find\(params\[:id\]\)/, m) end assert_instance_method :create, content do |m| assert_match(/@admin_role = Admin::Role\.new\(params\[:admin_role\]\)/, m) assert_match(/@admin_role\.save/, m) assert_match(/@admin_role\.errors/, m) end assert_instance_method :update, content do |m| assert_match(/@admin_role = Admin::Role\.find\(params\[:id\]\)/, m) assert_match(/@admin_role\.update_attributes\(params\[:admin_role\]\)/, m) assert_match(/@admin_role\.errors/, m) end assert_instance_method :destroy, content do |m| assert_match(/@admin_role = Admin::Role\.find\(params\[:id\]\)/, m) assert_match(/@admin_role\.destroy/, m) end end assert_file "test/functional/admin/roles_controller_test.rb", /class Admin::RolesControllerTest < ActionController::TestCase/ # Views %w( index edit new show _form ).each { |view| assert_file "app/views/admin/roles/#{view}.html.erb" } assert_no_file "app/views/layouts/admin/roles.html.erb" # Helpers assert_file "app/helpers/admin/roles_helper.rb" assert_file "test/unit/helpers/admin/roles_helper_test.rb" # Stylesheets assert_file "public/stylesheets/scaffold.css" end def test_scaffold_with_namespace_on_revoke run_generator [ "admin/role", "name:string", "description:string" ] run_generator [ "admin/role" ], :behavior => :revoke # Model assert_file "app/models/admin.rb" # ( should not be remove ) assert_no_file "app/models/admin/role.rb" assert_no_file "test/unit/admin/role_test.rb" assert_no_file "test/fixtures/admin/roles.yml" assert_no_migration "db/migrate/create_admin_roles.rb" # Route assert_file "config/routes.rb" do |route| assert_no_match(/namespace :admin do resources :roles end$/, route) end # Controller assert_no_file "app/controllers/admin/roles_controller.rb" assert_no_file "test/functional/admin/roles_controller_test.rb" # Views assert_no_file "app/views/admin/roles" assert_no_file "app/views/layouts/admin/roles.html.erb" # Helpers assert_no_file "app/helpers/admin/roles_helper.rb" assert_no_file "test/unit/helpers/admin/roles_helper_test.rb" # Stylesheets (should not be removed) assert_file "public/stylesheets/scaffold.css" end def test_scaffold_generator_on_revoke_does_not_mutilate_legacy_map_parameter run_generator # Add a |map| parameter to the routes block manually route_path = File.expand_path("config/routes.rb", destination_root) content = File.read(route_path).gsub(/\.routes\.draw do/) do |match| "#{match} |map|" end File.open(route_path, "wb") { |file| file.write(content) } run_generator ["product_line"], :behavior => :revoke assert_file "config/routes.rb", /\.routes\.draw do\s*\|map\|\s*$/ end def test_scaffold_generator_outputs_error_message_on_missing_attribute_type content = capture(:stderr) { run_generator ["post", "title:string", "body"]} assert_match(/Missing type for attribute 'body'/, content) assert_match(/Example: 'body:string' where string is the type/, content) end end