From d8a2a3657d13c54e3a31653c188e23ba61b29fdf Mon Sep 17 00:00:00 2001
From: Jacques Distler <distler@golem.ph.utexas.edu>
Date: Tue, 3 Mar 2009 22:27:26 -0600
Subject: [PATCH] Tests

Add tests for x-sendfile, <audio> and <video> support.
---
 test/functional/file_controller_test.rb | 12 ++++++++++++
 test/test_helper.rb                     | 25 +++++++++++++++++++++++++
 test/unit/page_renderer_test.rb         |  6 ++++++
 3 files changed, 43 insertions(+)

diff --git a/test/functional/file_controller_test.rb b/test/functional/file_controller_test.rb
index dbc82f9c..f3696059 100755
--- a/test/functional/file_controller_test.rb
+++ b/test/functional/file_controller_test.rb
@@ -81,6 +81,18 @@ class FileControllerTest < ActionController::TestCase
     assert_equal 'inline; filename="rails.gif"', r.headers['Content-Disposition']
   end
   
+  def test_pic_x_sendfile
+    pic = File.open("#{RAILS_ROOT}/test/fixtures/rails.gif", 'rb') { |f| f.read }
+    @web.wiki_files.create(:file_name => 'rails.gif', :description => 'An image', :content => pic)
+    @request.env.update({ 'HTTP_X_SENDFILE_TYPE' => 'foo' })
+    r = get :file, :web => 'wiki1', :id => 'rails.gif'
+    
+    assert_response(:success, bypass_body_parsing = true)
+    assert_match  '/rails.gif', r.headers['X-Sendfile']
+    assert_equal 'image/gif', r.headers['Content-Type']
+    assert_equal 'inline; filename="rails.gif"', r.headers['Content-Disposition']
+  end
+  
   def test_pic_unknown_pic
     r = get :file, :web => 'wiki1', :id => 'non-existant.gif'
     
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 66a1b1e6..f38eaf6b 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -164,6 +164,31 @@ class StubUrlGenerator < AbstractUrlGenerator
   end
 end
 
+  def media_link(mode, name, text, web_address, known_media, media_type)
+    link = CGI.escape(name)
+    text = CGI.escapeHTML(CGI.unescapeHTML(text || :description))
+    case mode.to_sym
+    when :export
+      if known_media 
+        %{<#{media_type} src="#{CGI.escape(name)}" controls="controls">#{text}</#{media_type}>}
+      else 
+        text
+      end
+    when :publish
+      if known_media
+        %{<#{media_type} src="../file/#{link}" controls="controls">#{text}</#{media_type}>}
+      else 
+        %{<span class="newWikiWord">#{text}</span>} 
+      end
+    else 
+      if known_media 
+        %{<#{media_type} src="../file/#{link}" controls="controls">#{text}</#{media_type}>}
+      else 
+        %{<span class="newWikiWord">#{text}<a href="../file/#{link}">?</a></span>} 
+      end
+    end
+  end
+
 module Test
   module Unit
     module Assertions
diff --git a/test/unit/page_renderer_test.rb b/test/unit/page_renderer_test.rb
index 7fc73e4f..54b2480f 100644
--- a/test/unit/page_renderer_test.rb
+++ b/test/unit/page_renderer_test.rb
@@ -521,6 +521,12 @@ END_THM
     assert_markup_parsed_as(
       "<p><a class='existingWikiWord' href='../file/square.jpg' title='Square'>Blue Square</a></p>",
       '[[square.jpg|Blue Square:file]]')
+    assert_markup_parsed_as(
+      "<p><video controls='controls' src='../file/square.jpg'>Blue Square</video></p>",
+      '[[square.jpg|Blue Square:video]]')
+    assert_markup_parsed_as(
+      "<p><audio controls='controls' src='../file/square.jpg'>Blue Square</audio></p>",
+      '[[square.jpg|Blue Square:audio]]')
     assert_markup_parsed_as( 
       "<p><a class='existingWikiWord' href='../file/square.jpg' title='Square'>Square</a></p>",
       '[[square.jpg:file]]')