axlsx 1.2.0 → 2.0.0

data/test/workbook/worksheet/tc_pivot_table.rb

@@ -0,0 +1,120 @@
+require 'tc_helper.rb'
+
+
+def shared_test_pivot_table_xml_validity(pivot_table)
+  schema = Nokogiri::XML::Schema(File.open(Axlsx::SML_XSD))
+  doc = Nokogiri::XML(pivot_table.to_xml_string)
+  errors = []
+  schema.validate(doc).each do |error|
+    errors.push error
+    puts error.message
+  end
+  assert(errors.empty?, "error free validation")
+end
+
+class TestPivotTable < Test::Unit::TestCase
+  def setup
+    p = Axlsx::Package.new
+    @ws = p.workbook.add_worksheet
+
+    @ws << ["Year","Month","Region", "Type", "Sales"]
+    @ws << [2012,  "Nov",  "East",   "Soda", "12345"]
+  end
+
+  def test_initialization
+    assert(@ws.workbook.pivot_tables.empty?)
+    assert(@ws.pivot_tables.empty?)
+  end
+
+  def test_add_pivot_table
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:D5')
+    assert_equal('G5:G6', pivot_table.ref, 'ref assigned from first parameter')
+    assert_equal('A1:D5', pivot_table.range, 'range assigned from second parameter')
+    assert_equal('PivotTable1', pivot_table.name, 'name automatically generated')
+    assert(pivot_table.is_a?(Axlsx::PivotTable), "must create a pivot table")
+    assert_equal(@ws.workbook.pivot_tables.last, pivot_table, "must be added to workbook pivot tables collection")
+    assert_equal(@ws.pivot_tables.last, pivot_table, "must be added to worksheet pivot tables collection")
+  end
+
+  def test_set_pivot_table_data_sheet
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:D5')
+    data_sheet = @ws.clone
+    data_sheet.name = "Pivot Table Data Source"
+
+    assert_equal(pivot_table.data_sheet.name, @ws.name, "must default to the same sheet the pivot table is added to")
+    pivot_table.data_sheet = data_sheet
+    assert_equal(pivot_table.data_sheet.name, data_sheet.name, "data sheet assigned to pivot table")
+  end
+
+  def test_add_pivot_table_with_config
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:E5') do |pt|
+      pt.rows = ['Year', 'Month']
+      pt.columns = ['Type']
+      pt.data = ['Sales']
+      pt.pages = ['Region']
+    end
+    assert_equal(['Year', 'Month'], pivot_table.rows)
+    assert_equal(['Type'], pivot_table.columns)
+    assert_equal([{:ref=>"Sales"}], pivot_table.data)
+    assert_equal(['Region'], pivot_table.pages)
+    shared_test_pivot_table_xml_validity(pivot_table)
+  end
+
+  def test_add_pivot_table_with_options_on_data_field
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:D5') do |pt|
+      pt.data = [{:ref=>"Sales", :subtotal => 'average'}]
+    end
+    assert_equal([{:ref=>"Sales", :subtotal => 'average'}], pivot_table.data)
+  end
+
+  def test_header_indices
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:E5')
+    assert_equal(0,   pivot_table.header_index_of('Year'   ))
+    assert_equal(1,   pivot_table.header_index_of('Month'  ))
+    assert_equal(2,   pivot_table.header_index_of('Region' ))
+    assert_equal(3,   pivot_table.header_index_of('Type'   ))
+    assert_equal(4,   pivot_table.header_index_of('Sales'  ))
+    assert_equal(nil, pivot_table.header_index_of('Missing'))
+    assert_equal(%w(A1 B1 C1 D1 E1), pivot_table.header_cell_refs)
+  end
+
+  def test_pn
+    @ws.add_pivot_table('G5:G6', 'A1:D5')
+    assert_equal(@ws.pivot_tables.first.pn, "pivotTables/pivotTable1.xml")
+  end
+
+  def test_index
+    @ws.add_pivot_table('G5:G6', 'A1:D5')
+    assert_equal(@ws.pivot_tables.first.index, @ws.workbook.pivot_tables.index(@ws.pivot_tables.first))
+  end
+
+  def test_relationships
+    assert(@ws.relationships.empty?)
+    @ws.add_pivot_table('G5:G6', 'A1:D5')
+    assert_equal(@ws.relationships.size, 1, "adding a pivot table adds a relationship")
+    @ws.add_pivot_table('G10:G11', 'A1:D5')
+    assert_equal(@ws.relationships.size, 2, "adding a pivot table adds a relationship")
+  end
+
+  def test_to_xml_string
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:D5')
+    shared_test_pivot_table_xml_validity(pivot_table)
+  end
+
+  def test_to_xml_string_with_configuration
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:E5') do |pt|
+      pt.rows = ['Year', 'Month']
+      pt.columns = ['Type']
+      pt.data = ['Sales']
+      pt.pages = ['Region']
+    end
+    shared_test_pivot_table_xml_validity(pivot_table)
+  end
+
+  def test_to_xml_string_with_options_on_data_field
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:E5') do |pt|
+      pt.data = [{:ref=>"Sales", :subtotal => 'average'}]
+    end
+    shared_test_pivot_table_xml_validity(pivot_table)
+  end
+end

data/test/workbook/worksheet/tc_pivot_table_cache_definition.rb

@@ -0,0 +1,54 @@
+require 'tc_helper.rb'
+
+class TestPivotTableCacheDefinition < Test::Unit::TestCase
+  def setup
+    p = Axlsx::Package.new
+    @ws = p.workbook.add_worksheet
+    5.times do
+      @ws << ["aa","aa","aa","aa"]
+    end
+    @pivot_table = @ws.add_pivot_table('G5:G6', 'A1:D5')
+    @cache_definition = @pivot_table.cache_definition
+  end
+
+  def test_initialization
+    assert(@cache_definition.is_a?(Axlsx::PivotTableCacheDefinition), "must create a pivot table cache definition")
+    assert_equal(@pivot_table, @cache_definition.pivot_table, 'refers back to its pivot table')
+  end
+
+  def test_pn
+    assert_equal('pivotCache/pivotCacheDefinition1.xml', @cache_definition.pn)
+  end
+
+  def test_rId
+    assert_equal @pivot_table.relationships.for(@cache_definition).Id, @cache_definition.rId
+  end
+
+  def test_index
+    assert_equal(0, @cache_definition.index)
+  end
+
+  def test_cache_id
+    assert_equal(1, @cache_definition.cache_id)
+  end
+
+  def test_data_sheet
+    data_sheet = @ws.clone
+    data_sheet.name = "Pivot Table Data Source"
+    @pivot_table.data_sheet = data_sheet
+
+    assert(@cache_definition.to_xml_string.include?(data_sheet.name), "must set the data source correctly")
+  end
+
+  def test_to_xml_string
+    schema = Nokogiri::XML::Schema(File.open(Axlsx::SML_XSD))
+    doc = Nokogiri::XML(@cache_definition.to_xml_string)
+    errors = []
+    schema.validate(doc).each do |error|
+      errors.push error
+      puts error.message
+    end
+    assert(errors.empty?, "error free validation")
+  end
+
+end

data/test/workbook/worksheet/tc_row.rb

@@ -12,13 +12,13 @@ class TestRow < Test::Unit::TestCase
     assert(@row.cells.empty?, "no cells by default")
     assert_equal(@row.worksheet, @ws, "has a reference to the worksheet")
     assert_nil(@row.height, "height defaults to nil")
-    assert(!@row.custom_height?, "no custom height by default")
+    assert(!@row.custom_height, "no custom height by default")
   end
 
   def test_initialize_with_fixed_height
     row = @ws.add_row([1,2,3,4,5], :height=>40)
     assert_equal(40, row.height)
-    assert(row.custom_height?)
+    assert(row.custom_height)
   end
 
   def test_style
@@ -57,7 +57,7 @@ class TestRow < Test::Unit::TestCase
 
   def test_custom_height
     @row.height = 20
-    assert(@row.custom_height?)
+    assert(@row.custom_height)
   end
 
   def test_height
@@ -111,7 +111,7 @@ class TestRow < Test::Unit::TestCase
     @row.add_cell 1
     @row.height = 20
     r_s_xml = Nokogiri::XML(@row.to_xml_string(0, ''))
-    assert_equal(r_s_xml.xpath(".//row[@r=1][@ht=20][@customHeight=1]").size, 1)
+    assert_equal(r_s_xml.xpath(".//row[@r=1][@ht=20][@customHeight='true']").size, 1)
   end
 
 end

data/test/workbook/worksheet/tc_selection.rb

@@ -3,70 +3,34 @@ require 'tc_helper.rb'
 
 class TestSelection < Test::Unit::TestCase
   def setup
-    @nil_options = { :active_cell => 'A2', :active_cell_id => 1, :pane => :top_left, :sqref => 'A2'  }
-    @options = @nil_options
-    
-    @string_options = { :active_cell => 'A2', :sqref => 'A2' }
-    @integer_options = { :active_cell_id => 1 }
-    @symbol_options = { :pane => :top_left }
-    
+    @options = { :active_cell => 'A2', :active_cell_id => 1, :pane => :top_left, :sqref => 'A2'  }
     @selection = Axlsx::Selection.new(@options)
   end
-  
-  def test_initialize
-    selection = Axlsx::Selection.new
-    
-    @nil_options.each do |key, value|
-      assert_equal(nil, selection.send(key.to_sym), "initialized default #{key} should be nil")
-      assert_equal(value, @selection.send(key.to_sym), "initialized options #{key} should be #{value}")
-    end
-  end
-  
-  def test_string_attribute_validation
-    @string_options.each do |key, value|
-      assert_raise(ArgumentError, "#{key} must be string") { @selection.send("#{key}=".to_sym, :symbol) }
-      assert_nothing_raised { @selection.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") { @selection.send("#{key}=".to_sym, "foo") }
-      assert_nothing_raised { @selection.send("#{key}=".to_sym, value) }
-    end
-  end
-  
-  def test_integer_attribute_validation
-    @integer_options.each do |key, value|
-      assert_raise(ArgumentError, "#{key} must be integer") { @selection.send("#{key}=".to_sym, "foo") }
-      assert_nothing_raised { @selection.send("#{key}=".to_sym, value) }
-    end
-  end
-  
+
   def test_active_cell
     assert_raise(ArgumentError) { @selection.active_cell = :active_cell }
     assert_nothing_raised { @selection.active_cell = "F5" }
     assert_equal(@selection.active_cell, "F5")
   end
-  
+
   def test_active_cell_id
     assert_raise(ArgumentError) { @selection.active_cell_id = "foo" }
     assert_nothing_raised { @selection.active_cell_id = 11 }
     assert_equal(@selection.active_cell_id, 11)
   end
-  
+
   def test_pane
     assert_raise(ArgumentError) { @selection.pane = "foo´" }
     assert_nothing_raised { @selection.pane = :bottom_right }
-    assert_equal(@selection.pane, :bottom_right)
+    assert_equal(@selection.pane, "bottomRight")
   end
-  
+
   def test_sqref
     assert_raise(ArgumentError) { @selection.sqref = :sqref }
     assert_nothing_raised { @selection.sqref = "G32" }
     assert_equal(@selection.sqref, "G32")
   end
-  
+
   def test_to_xml
     p = Axlsx::Package.new
     @ws = p.workbook.add_worksheet :name => "sheetview"
@@ -76,18 +40,15 @@ class TestSelection < Test::Unit::TestCase
       vs.add_selection(:bottom_left, { :active_cell => 'E55', :sqref => 'E55' })
       vs.add_selection(:bottom_right, { :active_cell => 'I57', :sqref => 'I57' })
     end
-    
+
     doc = Nokogiri::XML.parse(@ws.to_xml_string)
-    
+
     assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='B2'][@pane='topLeft'][@activeCell='B2']").size)
     assert doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='B2'][@pane='topLeft'][@activeCell='B2']")
-    
     assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='I10'][@pane='topRight'][@activeCell='I10']").size)
     assert doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='I10'][@pane='topRight'][@activeCell='I10']")
-    
     assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='E55'][@pane='bottomLeft'][@activeCell='E55']").size)
     assert doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='E55'][@pane='bottomLeft'][@activeCell='E55']")
-    
     assert_equal(1, doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='I57'][@pane='bottomRight'][@activeCell='I57']").size)
     assert doc.xpath("//xmlns:worksheet/xmlns:sheetViews/xmlns:sheetView/xmlns:selection[@sqref='I57'][@pane='bottomRight'][@activeCell='I57']")
   end

data/test/workbook/worksheet/tc_sheet_calc_pr.rb

@@ -0,0 +1,18 @@
+require 'tc_helper'
+
+class TestSheetCalcPr < Test::Unit::TestCase
+
+  def setup
+    @sheet_calc_pr = Axlsx::SheetCalcPr.new(:full_calc_on_load => false)
+  end
+
+  def test_full_calc_on_load
+    assert_equal false, @sheet_calc_pr.full_calc_on_load
+    assert Axlsx::SheetCalcPr.new.full_calc_on_load
+  end
+
+  def test_to_xml_string
+    doc = Nokogiri::XML(@sheet_calc_pr.to_xml_string)
+    assert_equal 1, doc.xpath('//sheetCalcPr[@fullCalcOnLoad="false"]').size
+  end
+end

data/test/workbook/worksheet/tc_sheet_format_pr.rb

@@ -0,0 +1,88 @@
+require 'tc_helper.rb'
+
+class TestSheetFormatPr < Test::Unit::TestCase
+
+  def setup
+    @options = {
+      :base_col_width => 5,
+      :default_col_width => 7.2,
+      :default_row_height => 5.2,
+      :custom_height => true,
+      :zero_height => false,
+      :thick_top => true,
+      :thick_bottom => true,
+      :outline_level_row => 0,
+      :outline_level_col => 0
+    }
+    @sheet_format_pr = Axlsx::SheetFormatPr.new(@options)
+  end
+
+  def test_default_initialization
+    sheet_format_pr = Axlsx::SheetFormatPr.new
+    assert_equal 8, sheet_format_pr.base_col_width
+    assert_equal 18, sheet_format_pr.default_row_height
+  end
+
+  def test_initialization_with_options
+    @options.each do |key, value|
+      assert_equal value, @sheet_format_pr.instance_variable_get("@#{key}")
+    end
+  end
+
+  def test_base_col_width
+    assert_raise(ArgumentError) { @sheet_format_pr.base_col_width = :foo }
+    assert_nothing_raised { @sheet_format_pr.base_col_width = 1 }
+  end
+
+  def test_outline_level_row
+    assert_raise(ArgumentError) { @sheet_format_pr.outline_level_row = :foo }
+    assert_nothing_raised { @sheet_format_pr.outline_level_row = 1 }
+  end
+
+  def test_outline_level_col
+    assert_raise(ArgumentError) { @sheet_format_pr.outline_level_col = :foo }
+    assert_nothing_raised { @sheet_format_pr.outline_level_col = 1 }
+  end
+
+  def test_default_row_height
+   assert_raise(ArgumentError) { @sheet_format_pr.default_row_height = :foo }
+   assert_nothing_raised { @sheet_format_pr.default_row_height= 1.0 }
+  end
+
+  def test_default_col_width
+   assert_raise(ArgumentError) { @sheet_format_pr.default_col_width= :foo }
+   assert_nothing_raised { @sheet_format_pr.default_col_width = 1.0 }
+  end
+
+  def test_custom_height
+   assert_raise(ArgumentError) { @sheet_format_pr.custom_height= :foo }
+   assert_nothing_raised { @sheet_format_pr.custom_height = true }
+  end
+
+  def test_zero_height
+   assert_raise(ArgumentError) { @sheet_format_pr.zero_height= :foo }
+   assert_nothing_raised { @sheet_format_pr.zero_height = true }
+  end
+  def test_thick_top
+   assert_raise(ArgumentError) { @sheet_format_pr.thick_top= :foo }
+   assert_nothing_raised { @sheet_format_pr.thick_top = true }
+  end
+  def test_thick_bottom
+   assert_raise(ArgumentError) { @sheet_format_pr.thick_bottom= :foo }
+   assert_nothing_raised { @sheet_format_pr.thick_bottom = true }
+  end
+
+  def test_to_xml_string
+    doc = Nokogiri::XML(@sheet_format_pr.to_xml_string)
+    assert doc.xpath("sheetFormatPr[@thickBottom='true']")
+    assert doc.xpath("sheetFormatPr[@baseColWidth=5]")
+    assert doc.xpath("sheetFormatPr[@default_col_width=7.2]")
+    assert doc.xpath("sheetFormatPr[@default_row_height=5.2]")
+    assert doc.xpath("sheetFormatPr[@custom_height='true']")
+    assert doc.xpath("sheetFormatPr[@zero_height='false']")
+    assert doc.xpath("sheetFormatPr[@thick_top='true']")
+    assert doc.xpath("sheetFormatPr[@outline_level_row=0]")
+    assert doc.xpath("sheetFormatPr[@outline_level_col=0]")
+  end
+
+end

data/test/workbook/worksheet/tc_sheet_pr.rb

@@ -0,0 +1,27 @@
+require 'tc_helper.rb'
+
+
+class TestSheetPr < Test::Unit::TestCase
+
+  def setup
+    worksheet = Axlsx::Package.new.workbook.add_worksheet
+    @options = { 
+      :sync_horizontal => false,
+      :sync_vertical => false,
+      :transtion_evaluation => true,
+      :transition_entry => true,
+      :published => false,
+      :filter_mode => true,
+      :enable_format_conditions_calculation => false,
+      :code_name => '007',
+      :sync_ref => 'foo'
+    }
+    @sheet_pr = Axlsx::SheetPr.new(worksheet, @options)
+  end
+
+  def test_initialization
+    @options.each do |key, value|
+      assert_equal value, @sheet_pr.send(key)
+    end
+  end
+end