crayfish 0.2.0 → 0.2.1

data/README.md

@@ -93,6 +93,41 @@ You can also use Prawn directly:
 %>
 ```
 
+## Testing
+
+You can intercept the template compiler to perform view tests on the generated HTML with the :html option.
+
+``` Ruby
+class MyPdfController < ApplicationController
+  attr_reader :options
+
+  def show
+    @options = params[:options]
+    @pdf = MyPdfModel.find(params[:id])
+  end
+
+end
+```
+
+``` Ruby
+class MyPdfControllerTest < ActionController::TestCase
+
+  setup do
+    @pdf = Factory.create :my_pdf
+  end
+
+  test "should show PDF" do
+    get :show, id: @pdf.to_param, :format => :pdf, :options => {:html => true}
+    assert_response :success
+
+    assert_tag  :tag => 'div', 
+                :attributes => {
+                  :class => 'my_div'
+                }
+  end
+end
+```
+
 ## License
 
 (The MIT License)

data/lib/crayfish/action_view.rb

@@ -26,10 +26,14 @@ module Crayfish
     def render *args
       if @branch_level
         @branch_level += 1
-        stack = @output_buffer
-        @output_buffer = ::ActionView::OutputBuffer.new
-        super
-        @output_buffer = stack
+        if @options[:html]
+          super
+        else
+          stack = @output_buffer
+          @output_buffer = ::ActionView::OutputBuffer.new
+          super
+          @output_buffer = stack
+        end
         @branch_level -= 1
       else
         super
@@ -68,7 +72,7 @@ module Crayfish
       def flush paint=true
         buf = @output_buffer.to_s || ''
         paint(buf,true) if paint
-        @output_buffer = ::ActionView::OutputBuffer.new
+        @output_buffer = ::ActionView::OutputBuffer.new unless @options[:html]
         buf
       end
 

data/lib/crayfish/html.rb

@@ -188,16 +188,24 @@ module Crayfish
       elsif node.kind_of?(Prawn::Table)
         if node.post_resize
           if /^(?<size>.*)%$/ =~ node.post_resize
+            new_width = parent_width * size.to_f / 100.0
+            column_widths = node.column_widths.dup
+            scale = new_width / column_widths.sum
             node.instance_variable_set '@column_widths', nil
+
             # reset constraints
-            node.cells.each do |cell|
-              cell.instance_variable_set '@max_width', parent_width
+            (0..node.row_length).map do |n|
+              row = node.row(n).columns(0..-1)
+              row.each_with_index do |cell,i|
+                 cell.instance_variable_set '@max_width', column_widths[i]*scale
+              end
             end
-            node.width = parent_width
+            node.width = new_width
             node.send(:set_column_widths)
             node.send(:position_cells)
           end
         end
+
         column_widths = node.column_widths
         (0..node.row_length).map do |n|
 

data/lib/crayfish/version.rb

@@ -21,5 +21,5 @@
 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 module Crayfish
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

data/test/action_view_test.rb

@@ -0,0 +1,89 @@
+# Copyright (c) 2012 Bingoentreprenøren AS
+# Copyright (c) 2012 Patrick Hanevold
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+require 'test_helper'
+
+class CrayfishActionViewTest < ActiveSupport::TestCase
+
+  class Response
+    def content_type
+      'pdf'
+    end
+
+    def content_type= val
+    end
+
+    def headers
+      @headers ||= {}
+    end
+  end
+
+  class ActionController
+    include Crayfish::ActionController
+
+    def response
+      @response ||= Response.new
+    end
+
+    def headers
+      response.headers
+    end
+
+    def options
+      { :html => true }
+    end
+  end
+
+  class ActionView
+    include Crayfish::ActionView
+
+    def controller
+      @controller ||= ActionController.new
+    end
+  end
+
+  def setup
+    @view = ActionView.new
+    @view.send(:setup)
+  end
+
+  test "forced html paint" do
+    assert_equal @view.send(:paint,'plain',true), 'plain'
+  end
+
+  test "should not flush output bufer in HTML mode" do
+    @view.send(:output_buffer) << "A"
+    @view.send(:flush)
+    assert_equal @view.send(:output_buffer), 'A'
+  end
+
+  test "should flush output bufer in PDF mode" do
+    ActionController.any_instance.stubs(:options).returns({})
+    view = ActionView.new
+    view.send(:setup)
+
+    view.send(:output_buffer) << "A"
+    view.send(:flush)
+    assert_equal view.send(:output_buffer), ''
+  end
+
+end