axlsx 1.1.4 → 1.3.1

data/test/workbook/worksheet/tc_comment.rb

@@ -0,0 +1,57 @@
+require 'tc_helper.rb'
+
+class TestComment < Test::Unit::TestCase
+  def setup
+    p = Axlsx::Package.new
+    wb = p.workbook
+    @ws = wb.add_worksheet
+    @c1 = @ws.add_comment :ref => 'A1', :text => 'penut machine', :author => 'crank'
+    @c2 = @ws.add_comment :ref => 'C3', :text => 'rust bucket', :author => 'PO'
+  end
+
+  def test_initailize
+    assert_raise(ArgumentError) { Axlsx::Comment.new }
+  end
+
+  def test_author
+    assert(@c1.author == 'crank')
+    assert(@c2.author == 'PO')
+  end
+
+  def test_text
+    assert(@c1.text == 'penut machine')
+    assert(@c2.text == 'rust bucket')
+  end
+
+  def test_author_index
+    assert_equal(@c1.author_index, 1)
+    assert_equal(@c2.author_index, 0)
+  end
+
+  def test_ref
+    assert(@c1.ref == 'A1')
+    assert(@c2.ref == 'C3')
+  end
+
+  def test_vml_shape
+    pos = Axlsx::name_to_indices(@c1.ref)
+    assert(@c1.vml_shape.is_a?(Axlsx::VmlShape))
+    assert(@c1.vml_shape.column == pos[0])
+    assert(@c1.vml_shape.row == pos[1])
+    assert(@c1.vml_shape.row == pos[1])
+    assert_equal(pos[0], @c1.vml_shape.left_column)
+    assert(@c1.vml_shape.top_row == pos[1])
+    assert_equal(pos[0] + 2 , @c1.vml_shape.right_column)
+    assert(@c1.vml_shape.bottom_row == pos[1]+4)
+  end
+
+  def to_xml_string
+    doc = Nokogiri::XML(@c1.to_xml_string)
+    assert_equal(doc.xpath("//comment[@ref='#{@c1.ref}']").size, 1)
+    assert_equal(doc.xpath("//comment[@authorId='#{@c1.author_index.to}']").size, 1)
+    assert_equal(doc.xpath("//t[text()='#{@c1.author}']").size, 1)
+    assert_equal(doc.xpath("//t[text()='#{@c1.text}']").size, 1)
+  end
+
+end
+

data/test/workbook/worksheet/tc_comments.rb

@@ -0,0 +1,57 @@
+require 'tc_helper.rb'
+
+class TestComments < Test::Unit::TestCase
+  def setup
+    p = Axlsx::Package.new
+    wb = p.workbook
+    @ws = wb.add_worksheet
+    @c1 = @ws.add_comment :ref => 'A1', :text => 'penut machine', :author => 'crank'
+    @c2 = @ws.add_comment :ref => 'C3', :text => 'rust bucket', :author => 'PO'
+  end
+
+  def test_initialize
+    assert_raise(ArgumentError) { Axlsx::Comments.new }
+    assert(@ws.comments.vml_drawing.is_a?(Axlsx::VmlDrawing))
+  end
+
+  def test_add_comment
+    assert_equal(@ws.comments.size, 2)
+    assert_raise(ArgumentError) { @ws.comments.add_comment() }
+    assert_raise(ArgumentError) { @ws.comments.add_comment(:text => 'Yes We Can', :ref => 'A1') }
+    assert_raise(ArgumentError) { @ws.comments.add_comment(:author => 'bob', :ref => 'A1') }
+    assert_raise(ArgumentError) { @ws.comments.add_comment(:author => 'bob', :text => 'Yes We Can')}
+    assert_nothing_raised { @ws.comments.add_comment(:author => 'bob', :text => 'Yes We Can', :ref => 'A1') }
+    assert_equal(@ws.comments.size, 3)
+  end
+  def test_authors
+    assert_equal(@ws.comments.authors.size, @ws.comments.size)
+    @ws.add_comment(:text => 'Yes We Can!', :author => :bob, :ref => 'F1')
+    assert_equal(@ws.comments.authors.size, 3)
+    @ws.add_comment(:text => 'Yes We Can!', :author => :bob, :ref => 'F1')
+    assert_equal(@ws.comments.authors.size, 3, 'only unique authors are returned') 
+  end
+  def test_pn
+    assert_equal(@ws.comments.pn, Axlsx::COMMENT_PN % (@ws.index+1).to_s)
+  end
+
+  def test_index
+    assert_equal(@ws.index, @ws.comments.index)
+  end
+
+  def test_to_xml_string
+    doc = Nokogiri::XML(@ws.comments.to_xml_string)
+    schema = Nokogiri::XML::Schema(File.open(Axlsx::SML_XSD))
+    errors = []
+    schema.validate(doc).each do |error|
+      errors << error
+    end
+    assert_equal(0, errors.length)
+
+    # TODO figure out why these xpath expressions dont work!
+    # assert(doc.xpath("//comments"))
+    # assert_equal(doc.xpath("//xmlns:author").size, @ws.comments.authors.size)
+    # assert_equal(doc.xpath("//comment").size, @ws.comments.size)
+  end
+end
+
+

data/test/workbook/worksheet/tc_conditional_formatting.rb

@@ -15,6 +15,50 @@ class TestConditionalFormatting < Test::Unit::TestCase
     assert_equal("AA1:AB100", optioned.sqref)
     assert_equal([1, 2], optioned.rules)
   end
+  def test_add_as_rule
+
+    color_scale = Axlsx::ColorScale.new do |cs|
+      cs.colors.first.rgb = "FFDFDFDF"
+      cs.colors.last.rgb = "FF00FF00"
+      cs.value_objects.first.type = :percentile
+      cs.value_objects.first.val = 5
+    end
+
+    data_bar = Axlsx::DataBar.new :color => "FFFF0000"
+    icon_set = Axlsx::IconSet.new :iconSet => "5Rating"
+    cfr = Axlsx::ConditionalFormattingRule.new( { :type => :containsText, :text => "TRUE",
+                                                      :dxfId => 0, :priority => 1,
+                                                      :formula => 'NOT(ISERROR(SEARCH("FALSE",AB1)))',
+                                                      :color_scale => color_scale,
+                                                      :data_bar => data_bar,
+                                                      :icon_set => icon_set})
+
+    assert(cfr.data_bar.is_a?(Axlsx::DataBar))
+    assert(cfr.icon_set.is_a?(Axlsx::IconSet))
+    assert(cfr.color_scale.is_a?(Axlsx::ColorScale))
+    cfs = @ws.add_conditional_formatting( "B2:B2", [cfr])
+    doc = Nokogiri::XML.parse(cfs.last.to_xml_string)
+    assert_equal(1, doc.xpath(".//conditionalFormatting[@sqref='B2:B2']//cfRule[@type='containsText'][@dxfId=0][@priority=1]").size)
+    assert doc.xpath(".//conditionalFormatting//cfRule[@type='containsText'][@dxfId=0][@priority=1]//formula='NOT(ISERROR(SEARCH(\"FALSE\",AB1)))'")
+
+    cfs.last.rules.last.type = :colorScale
+    doc = Nokogiri::XML.parse(cfs.last.to_xml_string)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//colorScale//cfvo").size, 2)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//colorScale//color").size, 2)
+
+    cfs.last.rules.last.type = :dataBar
+    doc = Nokogiri::XML.parse(cfs.last.to_xml_string)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//dataBar").size, 1)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//dataBar//cfvo").size, 2)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//dataBar//color[@rgb='FFFF0000']").size, 1)
+
+    cfs.last.rules.last.type = :iconSet
+    doc = Nokogiri::XML.parse(cfs.last.to_xml_string)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//iconSet//cfvo").size, 3)
+    assert_equal(doc.xpath(".//conditionalFormatting//cfRule//iconSet[@iconSet='5Rating']").size, 1)
+
+  end
+
 
   def test_add_as_hash
 

data/test/workbook/worksheet/tc_data_bar.rb

@@ -23,7 +23,7 @@ class TestDataBar < Test::Unit::TestCase
     assert_equal(@data_bar.maxLength, 0)
   end
 
-  def test_maxLength
+  def test_showValue
     assert_raise(ArgumentError) { @data_bar.showValue = :invalid_type }
     assert_nothing_raised { @data_bar.showValue =  false}
     assert_equal(@data_bar.showValue, false)

data/test/workbook/worksheet/tc_data_validation.rb

@@ -0,0 +1,265 @@
+# encoding: UTF-8
+require 'tc_helper.rb'
+
+
+class TestDataValidation < Test::Unit::TestCase
+  def setup
+    #inverse defaults
+    @boolean_options = { :allowBlank => false, :showDropDown => true, :showErrorMessage => false, :showInputMessage => true }
+    @nil_options = { :formula1 => 'foo',  :formula2 => 'foo', :errorTitle => 'foo', :operator => :lessThan, :prompt => 'foo', :promptTitle => 'foo', :sqref => 'foo' }
+    @type_option = { :type => :whole }
+    @error_style_option = { :errorStyle => :warning }
+    
+    @string_options = { :formula1 => 'foo', :formula2 => 'foo', :error => 'foo', :errorTitle => 'foo', :prompt => 'foo', :promptTitle => 'foo', :sqref => 'foo' }
+    @symbol_options = { :errorStyle => :warning, :operator => :lessThan, :type => :whole}
+    
+    @options = @boolean_options.merge(@nil_options).merge(@type_option).merge(@error_style_option)
+    
+    @dv = Axlsx::DataValidation.new(@options)
+  end
+  
+  def test_initialize
+    dv = Axlsx::DataValidation.new
+    
+    @boolean_options.each do |key, value|
+      assert_equal(!value, dv.send(key.to_sym), "initialized default #{key} should be #{!value}")
+      assert_equal(value, @dv.send(key.to_sym), "initialized options #{key} should be #{value}")
+    end
+    
+    @nil_options.each do |key, value|
+      assert_equal(nil, dv.send(key.to_sym), "initialized default #{key} should be nil")
+      assert_equal(value, @dv.send(key.to_sym), "initialized options #{key} should be #{value}")
+    end
+    
+    @type_option.each do |key, value|
+      assert_equal(:none, dv.send(key.to_sym), "initialized default #{key} should be :none")
+      assert_equal(value, @dv.send(key.to_sym), "initialized options #{key} should be #{value}")
+    end
+    
+    @error_style_option.each do |key, value|
+      assert_equal(:stop, dv.send(key.to_sym), "initialized default #{key} should be :stop")
+      assert_equal(value, @dv.send(key.to_sym), "initialized options #{key} should be #{value}")
+    end
+  end
+  
+  def test_boolean_attribute_validation
+    @boolean_options.each do |key, value|
+      assert_raise(ArgumentError, "#{key} must be boolean") { @dv.send("#{key}=".to_sym, 'A') }
+      assert_nothing_raised { @dv.send("#{key}=".to_sym, true) }
+    end
+  end
+  
+  def test_string_attribute_validation
+    @string_options.each do |key, value|
+      assert_raise(ArgumentError, "#{key} must be string") { @dv.send("#{key}=".to_sym, :symbol) }
+      assert_nothing_raised { @dv.send("#{key}=".to_sym, "foo") }
+    end
+  end
+  
+  def test_symbol_attribute_validation
+    @symbol_options.each do |key, value|
+      assert_raise(ArgumentError, "#{key} must be symbol") { @dv.send("#{key}=".to_sym, "foo") }
+      assert_nothing_raised { @dv.send("#{key}=".to_sym, value) }
+    end
+  end
+  
+  def test_formula1
+    assert_raise(ArgumentError) { @dv.formula1 = 10 }
+    assert_nothing_raised { @dv.formula1 = "=SUM(A1:A1)" }
+    assert_equal(@dv.formula1, "=SUM(A1:A1)")
+  end
+  
+  def test_formula2
+    assert_raise(ArgumentError) { @dv.formula2 = 10 }
+    assert_nothing_raised { @dv.formula2 = "=SUM(A1:A1)" }
+    assert_equal(@dv.formula2, "=SUM(A1:A1)")
+  end
+  
+  def test_allowBlank
+    assert_raise(ArgumentError) { @dv.allowBlank = "foo´" }
+    assert_nothing_raised { @dv.allowBlank = false }
+    assert_equal(@dv.allowBlank, false)
+  end
+  
+  def test_error
+    assert_raise(ArgumentError) { @dv.error = :symbol }
+    assert_nothing_raised { @dv.error = "This is a error message" }
+    assert_equal(@dv.error, "This is a error message")
+  end
+  
+  def test_errorStyle
+    assert_raise(ArgumentError) { @dv.errorStyle = "foo" }
+    assert_nothing_raised { @dv.errorStyle = :information }
+    assert_equal(@dv.errorStyle, :information)
+  end
+  
+  def test_errorTitle
+    assert_raise(ArgumentError) { @dv.errorTitle = :symbol }
+    assert_nothing_raised { @dv.errorTitle = "This is the error title" }
+    assert_equal(@dv.errorTitle, "This is the error title")
+  end
+  
+  def test_operator
+    assert_raise(ArgumentError) { @dv.operator = "foo" }
+    assert_nothing_raised { @dv.operator = :greaterThan }
+    assert_equal(@dv.operator, :greaterThan)
+  end
+  
+  def test_prompt
+    assert_raise(ArgumentError) { @dv.prompt = :symbol }
+    assert_nothing_raised { @dv.prompt = "This is a prompt message" }
+    assert_equal(@dv.prompt, "This is a prompt message")
+  end
+  
+  def test_promptTitle
+    assert_raise(ArgumentError) { @dv.promptTitle = :symbol }
+    assert_nothing_raised { @dv.promptTitle = "This is the prompt title" }
+    assert_equal(@dv.promptTitle, "This is the prompt title")
+  end
+  
+  def test_showDropDown
+    assert_raise(ArgumentError) { @dv.showDropDown = "foo´" }
+    assert_nothing_raised { @dv.showDropDown = false }
+    assert_equal(@dv.showDropDown, false)
+  end
+  
+  def test_showErrorMessage
+    assert_raise(ArgumentError) { @dv.showErrorMessage = "foo´" }
+    assert_nothing_raised { @dv.showErrorMessage = false }
+    assert_equal(@dv.showErrorMessage, false)
+  end
+  
+  def test_showInputMessage
+    assert_raise(ArgumentError) { @dv.showInputMessage = "foo´" }
+    assert_nothing_raised { @dv.showInputMessage = false }
+    assert_equal(@dv.showInputMessage, false)
+  end
+  
+  def test_sqref
+    assert_raise(ArgumentError) { @dv.sqref = 10 }
+    assert_nothing_raised { @dv.sqref = "A1:A1" }
+    assert_equal(@dv.sqref, "A1:A1")
+  end
+  
+  def test_type
+    assert_raise(ArgumentError) { @dv.type = "foo" }
+    assert_nothing_raised { @dv.type = :list }
+    assert_equal(@dv.type, :list)
+  end
+  
+  def test_whole_decimal_data_time_textLength_to_xml
+    p = Axlsx::Package.new
+    @ws = p.workbook.add_worksheet :name=>"data_validation"
+    @ws.add_data_validation("A1", { :type => :whole, :operator => :between, :formula1 => '5', :formula2 => '10', 
+        :showErrorMessage => true, :errorTitle => 'Wrong input', :error => 'Only values between 5 and 10', 
+        :errorStyle => :information, :showInputMessage => true, :promptTitle => 'Be carful!', 
+        :prompt => 'Only values between 5 and 10'})
+    
+    doc = Nokogiri::XML.parse(@ws.to_xml_string)
+    
+    #test attributes
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='1']/xmlns:dataValidation[@sqref='A1']
+      [@promptTitle='Be carful!'][@prompt='Only values between 5 and 10'][@operator='between'][@errorTitle='Wrong input']
+      [@error='Only values between 5 and 10'][@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@type='whole']
+      [@errorStyle='information']").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='1']/xmlns:dataValidation[@sqref='A1']
+      [@promptTitle='Be carful!'][@prompt='Only values between 5 and 10'][@operator='between'][@errorTitle='Wrong input']
+      [@error='Only values between 5 and 10'][@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true']
+      [@type='whole'][@errorStyle='information']")
+    
+    #test forumula1
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula1").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula1='5'")
+    
+    #test forumula2
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula2").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula2='10'")
+  end
+  
+  def test_list_to_xml
+    p = Axlsx::Package.new
+    @ws = p.workbook.add_worksheet :name=>"data_validation"
+    @ws.add_data_validation("A1", { :type => :list, :formula1 => 'A1:A5',
+        :showErrorMessage => true, :errorTitle => 'Wrong input', :error => 'Only values from list', 
+        :errorStyle => :stop, :showInputMessage => true, :promptTitle => 'Be carful!', 
+        :prompt => 'Only values from list', :showDropDown => true})
+    
+    doc = Nokogiri::XML.parse(@ws.to_xml_string)
+    
+    #test attributes
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='1']/xmlns:dataValidation[@sqref='A1']
+      [@promptTitle='Be carful!'][@prompt='Only values from list'][@errorTitle='Wrong input'][@error='Only values from list']
+      [@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@showDropDown='true'][@type='list']
+      [@errorStyle='stop']").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='1']/xmlns:dataValidation[@sqref='A1']
+      [@promptTitle='Be carful!'][@prompt='Only values from list'][@errorTitle='Wrong input'][@error='Only values from list']
+      [@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@showDropDown='true'][@type='list'][@errorStyle='stop']")
+    
+    #test forumula1
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula1").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula1='A1:A5'")
+  end
+  
+  def test_custom_to_xml
+    p = Axlsx::Package.new
+    @ws = p.workbook.add_worksheet :name=>"data_validation"
+    @ws.add_data_validation("A1", { :type => :custom, :formula1 => '=5/2',
+        :showErrorMessage => true, :errorTitle => 'Wrong input', :error => 'Only values corresponding formula', 
+        :errorStyle => :stop, :showInputMessage => true, :promptTitle => 'Be carful!', 
+        :prompt => 'Only values corresponding formula'})
+    
+    doc = Nokogiri::XML.parse(@ws.to_xml_string)
+    
+    #test attributes
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='1']/xmlns:dataValidation[@sqref='A1'][@promptTitle='Be carful!']
+      [@prompt='Only values corresponding formula'][@errorTitle='Wrong input'][@error='Only values corresponding formula'][@showErrorMessage='true']
+      [@allowBlank='true'][@showInputMessage='true'][@type='custom'][@errorStyle='stop']").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='1']/xmlns:dataValidation[@sqref='A1'][@promptTitle='Be carful!']
+      [@prompt='Only values corresponding formula'][@errorTitle='Wrong input'][@error='Only values corresponding formula']
+      [@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@type='custom'][@errorStyle='stop']")
+    
+    #test forumula1
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula1").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations/xmlns:dataValidation/xmlns:formula1='=5/2'")
+  end
+  
+  def test_multiple_datavalidations_to_xml
+    p = Axlsx::Package.new
+    @ws = p.workbook.add_worksheet :name=>"data_validation"
+    @ws.add_data_validation("A1", { :type => :whole, :operator => :between, :formula1 => '5', :formula2 => '10', 
+        :showErrorMessage => true, :errorTitle => 'Wrong input', :error => 'Only values between 5 and 10', 
+        :errorStyle => :information, :showInputMessage => true, :promptTitle => 'Be carful!', 
+        :prompt => 'Only values between 5 and 10'})
+    @ws.add_data_validation("B1", { :type => :list, :formula1 => 'A1:A5',
+        :showErrorMessage => true, :errorTitle => 'Wrong input', :error => 'Only values from list', 
+        :errorStyle => :stop, :showInputMessage => true, :promptTitle => 'Be carful!', 
+        :prompt => 'Only values from list', :showDropDown => true})
+    
+    doc = Nokogiri::XML.parse(@ws.to_xml_string)
+    
+    #test attributes
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='2']/xmlns:dataValidation[@sqref='A1']
+      [@promptTitle='Be carful!'][@prompt='Only values between 5 and 10'][@operator='between'][@errorTitle='Wrong input']
+      [@error='Only values between 5 and 10'][@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@type='whole']
+      [@errorStyle='information']").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='2']/xmlns:dataValidation[@sqref='A1']
+      [@promptTitle='Be carful!'][@prompt='Only values between 5 and 10'][@operator='between'][@errorTitle='Wrong input']
+      [@error='Only values between 5 and 10'][@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true']
+      [@type='whole'][@errorStyle='information']")
+    
+    #test attributes
+    assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='2']/xmlns:dataValidation[@sqref='B1']
+      [@promptTitle='Be carful!'][@prompt='Only values from list'][@errorTitle='Wrong input'][@error='Only values from list']
+      [@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@showDropDown='true'][@type='list']
+      [@errorStyle='stop']").size)
+    assert doc.xpath("//xmlns:worksheet/xmlns:dataValidations[@count='2']/xmlns:dataValidation[@sqref='B1']
+      [@promptTitle='Be carful!'][@prompt='Only values from list'][@errorTitle='Wrong input'][@error='Only values from list']
+      [@showErrorMessage='true'][@allowBlank='true'][@showInputMessage='true'][@showDropDown='true'][@type='list'][@errorStyle='stop']")
+  end
+  
+  def test_empty_attributes
+     v = Axlsx::DataValidation.new
+     assert_equal(nil, v.send(:get_valid_attributes))
+
+  end
+end

data/test/workbook/worksheet/tc_page_set_up_pr.rb

@@ -0,0 +1,15 @@
+require 'tc_helper.rb'
+
+class TestPageSetUpPr < Test::Unit::TestCase
+ def setup
+   @page_setup_pr = Axlsx::PageSetUpPr.new(:fit_to_page => true, :auto_page_breaks => true)
+ end
+
+ def test_fit_to_page
+   assert_equal true, @page_setup_pr.fit_to_page
+ end
+
+ def test_auto_page_breaks
+   assert_equal true, @page_setup_pr.auto_page_breaks
+ end
+end

data/test/workbook/worksheet/tc_page_setup.rb

@@ -0,0 +1,143 @@
+require 'tc_helper.rb'
+
+class TestPageSetup < Test::Unit::TestCase
+
+  def setup
+    @p = Axlsx::Package.new
+    ws = @p.workbook.add_worksheet :name => "hmmm"
+    @ps = ws.page_setup
+  end
+
+  def test_initialize
+    assert_equal(nil, @ps.fit_to_height)
+    assert_equal(nil, @ps.fit_to_width)
+    assert_equal(nil, @ps.orientation)
+    assert_equal(nil, @ps.paper_height)
+    assert_equal(nil, @ps.paper_width)
+    assert_equal(nil, @ps.scale)
+  end
+
+  def test_initialize_with_options
+    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(:landscape, optioned.orientation)
+    assert_equal("297mm", optioned.paper_height)
+    assert_equal("210mm", optioned.paper_width)
+    assert_equal(50, optioned.scale)
+  end
+
+  def test_set_all_values
+    @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(:landscape, @ps.orientation)
+    assert_equal("297mm", @ps.paper_height)
+    assert_equal("210mm", @ps.paper_width)
+    assert_equal(50, @ps.scale)
+  end
+
+  def test_paper_size
+    assert_raise(ArgumentError) { @ps.paper_size = 119 }
+    assert_nothing_raised {  @ps.paper_size = 10 }
+  end
+
+  def test_set_some_values
+    @ps.set(:fit_to_width => 2, :orientation => :portrait)
+    assert_equal(2, @ps.fit_to_width)
+    assert_equal(:portrait, @ps.orientation)
+    assert_equal(nil, @ps.fit_to_height)
+    assert_equal(nil, @ps.paper_height)
+    assert_equal(nil, @ps.paper_width)
+    assert_equal(nil, @ps.scale)
+  end
+
+  def test_default_fit_to_page?
+    assert(@ps.fit_to_width == nil && @ps.fit_to_height == nil)
+    assert(@ps.fit_to_page? == false)
+  end
+  
+  def test_with_height_fit_to_page?
+    assert(@ps.fit_to_width == nil && @ps.fit_to_height == nil)
+    @ps.set(:fit_to_height => 1)
+    assert(@ps.fit_to_page?)
+  end
+
+  def test_with_width_fit_to_page?
+    assert(@ps.fit_to_width == nil && @ps.fit_to_height == nil)
+    @ps.set(:fit_to_width => 1)
+    assert(@ps.fit_to_page?)
+  end
+
+  def test_to_xml_all_values
+    @ps.set(:fit_to_height => 1, :fit_to_width => 2, :orientation => :landscape, :paper_height => "297mm", :paper_width => "210mm", :scale => 50)
+    doc = Nokogiri::XML.parse(@ps.to_xml_string)
+    assert_equal(1, doc.xpath(".//pageSetup[@fitToHeight='1'][@fitToWidth='2'][@orientation='landscape'][@paperHeight='297mm'][@paperWidth='210mm'][@scale='50']").size)
+  end
+
+  def test_to_xml_some_values
+    @ps.set(:orientation => :portrait)
+    doc = Nokogiri::XML.parse(@ps.to_xml_string)
+    assert_equal(1, doc.xpath(".//pageSetup[@orientation='portrait']").size)
+    assert_equal(0, doc.xpath(".//pageSetup[@fitToHeight]").size)
+    assert_equal(0, doc.xpath(".//pageSetup[@fitToWidth]").size)
+    assert_equal(0, doc.xpath(".//pageSetup[@paperHeight]").size)
+    assert_equal(0, doc.xpath(".//pageSetup[@paperWidth]").size)
+    assert_equal(0, doc.xpath(".//pageSetup[@scale]").size)
+  end
+
+  def test_fit_to_height
+    assert_raise(ArgumentError) { @ps.fit_to_height = 1.5 }
+    assert_nothing_raised { @ps.fit_to_height = 2 }
+    assert_equal(2, @ps.fit_to_height)
+  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)
+  end
+
+  def test_orientation
+    assert_raise(ArgumentError) { @ps.orientation = "" }
+    assert_nothing_raised { @ps.orientation = :default }
+    assert_equal(:default, @ps.orientation)
+  end
+
+  def test_paper_height
+    assert_raise(ArgumentError) { @ps.paper_height = 99 }
+    assert_nothing_raised { @ps.paper_height = "11in" }
+    assert_equal("11in", @ps.paper_height)
+  end
+
+  def test_paper_width
+    assert_raise(ArgumentError) { @ps.paper_width = "22" }
+    assert_nothing_raised { @ps.paper_width = "29.7cm" }
+    assert_equal("29.7cm", @ps.paper_width)
+  end
+
+  def test_scale
+    assert_raise(ArgumentError) { @ps.scale = 50.5 }
+    assert_nothing_raised { @ps.scale = 99 }
+    assert_equal(99, @ps.scale)
+  end
+
+  def test_fit_to
+    fits = @ps.fit_to(:width => 1)
+    assert_equal([1, 9999], fits)
+    fits = @ps.fit_to :height => 1
+    assert_equal(fits, [9999,1])
+    fits = @ps.fit_to :height => 7, :width => 2
+    assert_equal(fits, [2, 7])
+    assert_raise(ArgumentError) { puts @ps.fit_to(:width => true)}
+
+
+  end
+end