Add pivot table option to sort headers (#143)

1 files changed, 13 insertions, 0 deletions

diff --git a/test/workbook/worksheet/tc_pivot_table.rb b/test/workbook/worksheet/tc_pivot_table.rb
index 591f0ba9..c7aa53cc 100644
--- a/test/workbook/worksheet/tc_pivot_table.rb
+++ b/test/workbook/worksheet/tc_pivot_table.rb
@@ -87,6 +87,19 @@ class TestPivotTable < Test::Unit::TestCase
     assert_equal(['Year'], pivot_table.no_subtotals_on_headers)
   end
 
+  def test_add_pivot_table_with_months_sorted
+    pivot_table = @ws.add_pivot_table('G5:G6', 'A1:E5', {:sort_on_headers=>['Month']}) do |pt|
+      pt.data = ['Sales']
+      pt.rows = ['Year','Month']
+    end
+    assert_equal({'Month' => :ascending}, pivot_table.sort_on_headers)
+
+    pivot_table.sort_on_headers = {'Month' => :descending}
+    assert_equal({'Month' => :descending}, pivot_table.sort_on_headers)
+
+    shared_test_pivot_table_xml_validity(pivot_table)
+  end
+
   def test_header_indices
     pivot_table = @ws.add_pivot_table('G5:G6', 'A1:E5')
     assert_equal(0,   pivot_table.header_index_of('Year'   ))