diff options
| -rw-r--r-- | test/workbook/worksheet/tc_header_footer.rb | 151 | ||||
| -rw-r--r-- | test/workbook/worksheet/tc_worksheet.rb | 37 |
2 files changed, 182 insertions, 6 deletions
diff --git a/test/workbook/worksheet/tc_header_footer.rb b/test/workbook/worksheet/tc_header_footer.rb new file mode 100644 index 00000000..4d78aa9d --- /dev/null +++ b/test/workbook/worksheet/tc_header_footer.rb @@ -0,0 +1,151 @@ +require 'tc_helper' + +class TestHeaderFooter < Test::Unit::TestCase + + def setup + @p = Axlsx::Package.new + ws = @p.workbook.add_worksheet :name => 'test' + @hf = ws.header_footer + end + + def test_initialize + assert_equal(nil, @hf.odd_header) + assert_equal(nil, @hf.odd_footer) + + assert_equal(nil, @hf.even_header) + assert_equal(nil, @hf.even_footer) + + assert_equal(nil, @hf.first_header) + assert_equal(nil, @hf.first_footer) + + assert_equal(nil, @hf.different_first) + assert_equal(nil, @hf.different_odd_even) + end + + def test_initialize_with_options + header_footer = { + :odd_header => 'oh', + :odd_footer => 'of', + + :even_header => 'eh', + :even_footer => 'ef', + + :first_header => 'fh', + :first_footer => 'ff', + + :different_first => true, + :different_odd_even => true + } + optioned = @p.workbook.add_worksheet(:name => 'optioned', :header_footer => header_footer).header_footer + + assert_equal('oh', optioned.odd_header) + assert_equal('of', optioned.odd_footer) + + assert_equal('eh', optioned.even_header) + assert_equal('ef', optioned.even_footer) + + assert_equal('fh', optioned.first_header) + assert_equal('ff', optioned.first_footer) + + assert_equal(true, optioned.different_first) + assert_equal(true, optioned.different_odd_even) + end + + def test_string_attributes + %w(odd_header odd_footer even_header even_footer first_header first_footer).each do |attr| + assert_raise(ArgumentError, 'only strings allowed in string attributes') { @hf.send("#{attr}=", 1) } + assert_nothing_raised { @hf.send("#{attr}=", 'test_string') } + end + end + + def test_boolean_attributes + %w(different_first different_odd_even).each do |attr| + assert_raise(ArgumentError, 'only booleanish allowed in string attributes') { @hf.send("#{attr}=", 'foo') } + assert_nothing_raised { @hf.send("#{attr}=", 1) } + end + end + + def test_set_all_values + @hf.set( + :odd_header => 'oh', + :odd_footer => 'of', + + :even_header => 'eh', + :even_footer => 'ef', + + :first_header => 'fh', + :first_footer => 'ff', + + :different_first => true, + :different_odd_even => true + ) + + assert_equal('oh', @hf.odd_header) + assert_equal('of', @hf.odd_footer) + + assert_equal('eh', @hf.even_header) + assert_equal('ef', @hf.even_footer) + + assert_equal('fh', @hf.first_header) + assert_equal('ff', @hf.first_footer) + + assert_equal(true, @hf.different_first) + assert_equal(true, @hf.different_odd_even) + end + + def test_to_xml_all_values + @hf.set( + :odd_header => 'oh', + :odd_footer => 'of', + + :even_header => 'eh', + :even_footer => 'ef', + + :first_header => 'fh', + :first_footer => 'ff', + + :different_first => true, + :different_odd_even => true + ) + + doc = Nokogiri::XML.parse(@hf.to_xml_string) + assert_equal(1, doc.xpath(".//headerFooter[@differentFirst='true'][@differentOddEven='true']").size) + + assert_equal(1, doc.xpath(".//headerFooter/oddHeader").size) + assert_equal('oh', doc.xpath(".//headerFooter/oddHeader").text) + assert_equal(1, doc.xpath(".//headerFooter/oddFooter").size) + assert_equal('of', doc.xpath(".//headerFooter/oddFooter").text) + + assert_equal(1, doc.xpath(".//headerFooter/evenHeader").size) + assert_equal('eh', doc.xpath(".//headerFooter/evenHeader").text) + assert_equal(1, doc.xpath(".//headerFooter/evenFooter").size) + assert_equal('ef', doc.xpath(".//headerFooter/evenFooter").text) + + assert_equal(1, doc.xpath(".//headerFooter/firstHeader").size) + assert_equal('fh', doc.xpath(".//headerFooter/firstHeader").text) + assert_equal(1, doc.xpath(".//headerFooter/firstFooter").size) + assert_equal('ff', doc.xpath(".//headerFooter/firstFooter").text) + end + + def test_to_xml_some_values + @hf.set( + :odd_header => 'oh', + :different_odd_even => false + ) + + doc = Nokogiri::XML.parse(@hf.to_xml_string) + assert_equal(1, doc.xpath(".//headerFooter[@differentOddEven='false']").size) + assert_equal(0, doc.xpath(".//headerFooter[@differentFirst]").size) + + assert_equal(1, doc.xpath(".//headerFooter/oddHeader").size) + assert_equal('oh', doc.xpath(".//headerFooter/oddHeader").text) + assert_equal(0, doc.xpath(".//headerFooter/oddFooter").size) + + assert_equal(0, doc.xpath(".//headerFooter/evenHeader").size) + assert_equal(0, doc.xpath(".//headerFooter/evenFooter").size) + + assert_equal(0, doc.xpath(".//headerFooter/firstHeader").size) + assert_equal(0, doc.xpath(".//headerFooter/firstFooter").size) + end +end + diff --git a/test/workbook/worksheet/tc_worksheet.rb b/test/workbook/worksheet/tc_worksheet.rb index 4c8f2822..da90ffa0 100644 --- a/test/workbook/worksheet/tc_worksheet.rb +++ b/test/workbook/worksheet/tc_worksheet.rb @@ -55,6 +55,17 @@ class TestWorksheet < Test::Unit::TestCase end end + def test_header_footer + assert(@ws.header_footer.is_a? Axlsx::HeaderFooter) + end + + def test_header_footer_yield + @ws.header_footer do |hf| + assert(hf.is_a? Axlsx::HeaderFooter) + assert(@ws.header_footer == hf) + end + end + def test_no_autowidth @ws.workbook.use_autowidth = false @ws.add_row [1,2,3,4] @@ -65,7 +76,8 @@ class TestWorksheet < Test::Unit::TestCase page_margins = {:left => 2, :right => 2, :bottom => 2, :top => 2, :header => 2, :footer => 2} page_setup = {:fit_to_height => 1, :fit_to_width => 1, :orientation => :landscape, :paper_width => "210mm", :paper_height => "297mm", :scale => 80} print_options = {:grid_lines => true, :headings => true, :horizontal_centered => true, :vertical_centered => true} - optioned = @ws.workbook.add_worksheet(:name => 'bob', :page_margins => page_margins, :page_setup => page_setup, :print_options => print_options, :selected => true, :show_gridlines => false) + header_footer = {:different_first => false, :different_odd_even => false, :odd_header => 'Header'} + optioned = @ws.workbook.add_worksheet(:name => 'bob', :page_margins => page_margins, :page_setup => page_setup, :print_options => print_options, :header_footer => header_footer, :selected => true, :show_gridlines => false) page_margins.keys.each do |key| assert_equal(page_margins[key], optioned.page_margins.send(key)) end @@ -75,6 +87,9 @@ class TestWorksheet < Test::Unit::TestCase print_options.keys.each do |key| assert_equal(print_options[key], optioned.print_options.send(key)) end + header_footer.keys.each do |key| + assert_equal(header_footer[key], optioned.header_footer.send(key)) + end assert_equal(optioned.name, 'bob') assert_equal(optioned.selected, true) assert_equal(optioned.show_gridlines, false) @@ -293,6 +308,16 @@ class TestWorksheet < Test::Unit::TestCase assert_equal(doc.xpath('//xmlns:worksheet/xmlns:printOptions[@gridLines="true"][@horizontalCentered="true"]').size, 1) end + def test_to_xml_string_header_footer + @ws.header_footer do |hf| + hf.different_first = false + hf.different_odd_even = false + hf.odd_header = 'Test Header' + end + doc = Nokogiri::XML(@ws.to_xml_string) + assert_equal(doc.xpath('//xmlns:worksheet/xmlns:headerFooter[@differentFirst="false"][@differentOddEven="false"]').size, 1) + end + def test_to_xml_string_drawing @ws.add_chart Axlsx::Pie3DChart doc = Nokogiri::XML(@ws.to_xml_string) @@ -317,7 +342,7 @@ class TestWorksheet < Test::Unit::TestCase def test_styles assert(@ws.styles.is_a?(Axlsx::Styles), 'worksheet provides access to styles') end - + def test_to_xml_string_with_illegal_chars nasties = "\v\u2028\u0001\u0002\u0003\u0004\u0005\u0006\u0007\u0008\u001f" @ws.add_row [nasties] @@ -400,7 +425,7 @@ class TestWorksheet < Test::Unit::TestCase @ws.add_row [1, 2, 3] assert_nothing_raised {@ws.protect_range(@ws.rows.first.cells) } assert_equal('A1:C1', @ws.send(:protected_ranges).last.sqref) - + end def test_merge_cells @ws.add_row [1,2,3] @@ -412,7 +437,7 @@ class TestWorksheet < Test::Unit::TestCase assert_equal(@ws.send(:merged_cells).size, 3) assert_equal(@ws.send(:merged_cells).last, "A3:B3") end - + def test_merge_cells_sorts_correctly_by_row_when_given_array 10.times do |i| @ws.add_row [i] @@ -420,7 +445,7 @@ class TestWorksheet < Test::Unit::TestCase @ws.merge_cells [@ws.rows[8].cells.first, @ws.rows[9].cells.first] assert_equal "A9:A10", @ws.send(:merged_cells).first end - + def test_auto_filter assert(@ws.auto_filter.range.nil?) assert_raise(ArgumentError) { @ws.auto_filter = 123 } @@ -432,6 +457,6 @@ class TestWorksheet < Test::Unit::TestCase @ws.auto_filter.range = 'A1:D9' @ws.auto_filter.add_column 0, :filters, :filter_items => [1] doc = Nokogiri::XML(@ws.to_xml_string) - assert(doc.xpath('//sheetPr[@filterMode="true"]')) + assert(doc.xpath('//sheetPr[@filterMode="true"]')) end end |