If worksheet.fit_to_page must be true, lets set it to true. I am far too lazy and forgetful to remember to set that every time I want to fit_to_width or fit_to_height.

1 files changed, 14 insertions, 3 deletions

diff --git a/test/workbook/worksheet/tc_page_setup.rb b/test/workbook/worksheet/tc_page_setup.rb
index ff323c75..f374b0e5 100644
--- a/test/workbook/worksheet/tc_page_setup.rb
+++ b/test/workbook/worksheet/tc_page_setup.rb
@@ -3,8 +3,8 @@ require 'tc_helper.rb'
 class TestPageSetup < Test::Unit::TestCase
 
   def setup
-    p = Axlsx::Package.new
-    ws = p.workbook.add_worksheet :name => "hmmm"
+    @p = Axlsx::Package.new
+    ws = @p.workbook.add_worksheet :name => "hmmm"
     @ps = ws.page_setup
   end
 
@@ -18,9 +18,17 @@ class TestPageSetup < Test::Unit::TestCase
   end
 
   def test_initialize_with_options
-    optioned = Axlsx::PageSetup.new(:fit_to_height => 1, :fit_to_width => 2, :orientation => :landscape, :paper_height => "297mm", :paper_width => "210mm", :scale => 50)
+    page_setup = { :fit_to_height => 1,
+                   :fit_to_width => 2,
+                   :orientation => :landscape,
+                   :paper_height => "297mm",
+                   :paper_width => "210mm",
+                   :scale => 50 }
+
+    optioned = @p.workbook.add_worksheet(:name => 'optioned', :page_setup => page_setup).page_setup
     assert_equal(1, optioned.fit_to_height)
     assert_equal(2, optioned.fit_to_width)
+    assert_equal(true, optioned.worksheet.fit_to_page)
     assert_equal(:landscape, optioned.orientation)
     assert_equal("297mm", optioned.paper_height)
     assert_equal("210mm", optioned.paper_width)
@@ -31,6 +39,7 @@ class TestPageSetup < Test::Unit::TestCase
     @ps.set(:fit_to_height => 1, :fit_to_width => 2, :orientation => :landscape, :paper_height => "297mm", :paper_width => "210mm", :scale => 50)
     assert_equal(1, @ps.fit_to_height)
     assert_equal(2, @ps.fit_to_width)
+    assert_equal(true, @ps.worksheet.fit_to_page)
     assert_equal(:landscape, @ps.orientation)
     assert_equal("297mm", @ps.paper_height)
     assert_equal("210mm", @ps.paper_width)
@@ -68,12 +77,14 @@ class TestPageSetup < Test::Unit::TestCase
     assert_raise(ArgumentError) { @ps.fit_to_height = 1.5 }
     assert_nothing_raised { @ps.fit_to_height = 2 }
     assert_equal(2, @ps.fit_to_height)
+    assert_equal(true, @ps.worksheet.fit_to_page)
   end
 
   def test_fit_to_width
     assert_raise(ArgumentError) { @ps.fit_to_width = false }
     assert_nothing_raised { @ps.fit_to_width = 1 }
     assert_equal(1, @ps.fit_to_width)
+    assert_equal(true, @ps.worksheet.fit_to_page)
   end
 
   def test_orientation