require File.dirname(__FILE__) + '/../abstract_unit' class RespondToController < ActionController::Base layout :set_layout def html_xml_or_rss respond_to do |type| type.html { render :text => "HTML" } type.xml { render :text => "XML" } type.rss { render :text => "RSS" } type.all { render :text => "Nothing" } end end def js_or_html respond_to do |type| type.html { render :text => "HTML" } type.js { render :text => "JS" } type.all { render :text => "Nothing" } end end def json_or_yaml respond_to do |type| type.json { render :text => "JSON" } type.yaml { render :text => "YAML" } end end def html_or_xml respond_to do |type| type.html { render :text => "HTML" } type.xml { render :text => "XML" } type.all { render :text => "Nothing" } end end def just_xml respond_to do |type| type.xml { render :text => "XML" } end end def using_defaults respond_to do |type| type.html type.js type.xml end end def using_defaults_with_type_list respond_to(:html, :js, :xml) end def made_for_content_type respond_to do |type| type.rss { render :text => "RSS" } type.atom { render :text => "ATOM" } type.all { render :text => "Nothing" } end end def custom_type_handling respond_to do |type| type.html { render :text => "HTML" } type.custom("application/crazy-xml") { render :text => "Crazy XML" } type.all { render :text => "Nothing" } end end def custom_constant_handling Mime::Type.register("text/x-mobile", :mobile) respond_to do |type| type.html { render :text => "HTML" } type.mobile { render :text => "Mobile" } end Mime.send :remove_const, :MOBILE end def custom_constant_handling_without_block Mime::Type.register("text/x-mobile", :mobile) respond_to do |type| type.html { render :text => "HTML" } type.mobile end Mime.send :remove_const, :MOBILE end def handle_any respond_to do |type| type.html { render :text => "HTML" } type.any(:js, :xml) { render :text => "Either JS or XML" } end end def all_types_with_layout respond_to do |type| type.html type.js end end def rescue_action(e) raise end protected def set_layout if action_name == "all_types_with_layout" "standard" end end end RespondToController.view_paths = [ File.dirname(__FILE__) + "/../fixtures/" ] class MimeControllerTest < Test::Unit::TestCase def setup @request = ActionController::TestRequest.new @response = ActionController::TestResponse.new @controller = RespondToController.new @request.host = "www.example.com" end def test_html @request.env["HTTP_ACCEPT"] = "text/html" get :js_or_html assert_equal 'HTML', @response.body get :html_or_xml assert_equal 'HTML', @response.body get :just_xml assert_response 406 end def test_all @request.env["HTTP_ACCEPT"] = "*/*" get :js_or_html assert_equal 'HTML', @response.body # js is not part of all get :html_or_xml assert_equal 'HTML', @response.body get :just_xml assert_equal 'XML', @response.body end def test_xml @request.env["HTTP_ACCEPT"] = "application/xml" get :html_xml_or_rss assert_equal 'XML', @response.body end def test_js_or_html @request.env["HTTP_ACCEPT"] = "text/javascript, text/html" get :js_or_html assert_equal 'JS', @response.body get :html_or_xml assert_equal 'HTML', @response.body get :just_xml assert_response 406 end def test_json_or_yaml get :json_or_yaml assert_equal 'JSON', @response.body get :json_or_yaml, :format => 'json' assert_equal 'JSON', @response.body get :json_or_yaml, :format => 'yaml' assert_equal 'YAML', @response.body { 'YAML' => %w(text/yaml), 'JSON' => %w(application/json text/x-json) }.each do |body, content_types| content_types.each do |content_type| @request.env['HTTP_ACCEPT'] = content_type get :json_or_yaml assert_equal body, @response.body end end end def test_js_or_anything @request.env["HTTP_ACCEPT"] = "text/javascript, */*" get :js_or_html assert_equal 'JS', @response.body get :html_or_xml assert_equal 'HTML', @response.body get :just_xml assert_equal 'XML', @response.body end def test_using_defaults @request.env["HTTP_ACCEPT"] = "*/*" get :using_defaults assert_equal 'Hello world!', @response.body @request.env["HTTP_ACCEPT"] = "text/javascript" get :using_defaults assert_equal '$("body").visualEffect("highlight");', @response.body @request.env["HTTP_ACCEPT"] = "application/xml" get :using_defaults assert_equal "

Hello world!

\n", @response.body end def test_using_defaults_with_type_list @request.env["HTTP_ACCEPT"] = "*/*" get :using_defaults_with_type_list assert_equal 'Hello world!', @response.body @request.env["HTTP_ACCEPT"] = "text/javascript" get :using_defaults_with_type_list assert_equal '$("body").visualEffect("highlight");', @response.body @request.env["HTTP_ACCEPT"] = "application/xml" get :using_defaults_with_type_list assert_equal "

Hello world!

\n", @response.body end def test_with_content_type @request.env["CONTENT_TYPE"] = "application/atom+xml" get :made_for_content_type assert_equal "ATOM", @response.body @request.env["CONTENT_TYPE"] = "application/rss+xml" get :made_for_content_type assert_equal "RSS", @response.body end def test_synonyms @request.env["HTTP_ACCEPT"] = "application/javascript" get :js_or_html assert_equal 'JS', @response.body @request.env["HTTP_ACCEPT"] = "application/x-xml" get :html_xml_or_rss assert_equal "XML", @response.body end def test_custom_types @request.env["HTTP_ACCEPT"] = "application/crazy-xml" get :custom_type_handling assert_equal 'Crazy XML', @response.body @request.env["HTTP_ACCEPT"] = "text/html" get :custom_type_handling assert_equal 'HTML', @response.body end def test_xhtml_alias @request.env["HTTP_ACCEPT"] = "application/xhtml+xml,application/xml" get :html_or_xml assert_equal 'HTML', @response.body end def test_firefox_simulation @request.env["HTTP_ACCEPT"] = "text/xml,application/xml,application/xhtml+xml,text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5" get :html_or_xml assert_equal 'HTML', @response.body end def test_handle_any @request.env["HTTP_ACCEPT"] = "*/*" get :handle_any assert_equal 'HTML', @response.body @request.env["HTTP_ACCEPT"] = "text/javascript" get :handle_any assert_equal 'Either JS or XML', @response.body @request.env["HTTP_ACCEPT"] = "text/xml" get :handle_any assert_equal 'Either JS or XML', @response.body end def test_all_types_with_layout @request.env["HTTP_ACCEPT"] = "text/javascript" get :all_types_with_layout assert_equal 'RJS for all_types_with_layout', @response.body @request.env["HTTP_ACCEPT"] = "text/html" get :all_types_with_layout assert_equal 'HTML for all_types_with_layout', @response.body end def test_xhr xhr :get, :js_or_html assert_equal 'JS', @response.body xhr :get, :using_defaults assert_equal '$("body").visualEffect("highlight");', @response.body end def test_custom_constant get :custom_constant_handling, :format => "mobile" assert_equal "Mobile", @response.body end def custom_constant_handling_without_block assert_raised(ActionController::RenderError) do get :custom_constant_handling, :format => "mobile" end end def test_forced_format get :html_xml_or_rss assert_equal "HTML", @response.body get :html_xml_or_rss, :format => "html" assert_equal "HTML", @response.body get :html_xml_or_rss, :format => "xml" assert_equal "XML", @response.body get :html_xml_or_rss, :format => "rss" assert_equal "RSS", @response.body end def test_extension_synonyms get :html_xml_or_rss, :format => "xhtml" assert_equal "HTML", @response.body end def test_render_action_for_html @controller.instance_eval do def render(*args) unless args.empty? @action = args.first[:action] end response.body = @action end end get :using_defaults assert_equal "using_defaults", @response.body get :using_defaults, :format => "xml" assert_equal "using_defaults.xml.builder", @response.body end end