Function TestPivotTableDataRange

pivotTable_test.go:368–435

(t *testing.T)

}

func TestPivotTableDataRange(t *testing.T) {
	f := NewFile()
	// Create table in a worksheet
	assert.NoError(t, f.AddTable("Sheet1", &Table{
		Name:  "Table1",
		Range: "A1:D5",
	}))
	for row := 2; row < 6; row++ {
		assert.NoError(t, f.SetCellValue("Sheet1", fmt.Sprintf("A%d", row), rand.Intn(10)))
		assert.NoError(t, f.SetCellValue("Sheet1", fmt.Sprintf("B%d", row), rand.Intn(10)))
		assert.NoError(t, f.SetCellValue("Sheet1", fmt.Sprintf("C%d", row), rand.Intn(10)))
		assert.NoError(t, f.SetCellValue("Sheet1", fmt.Sprintf("D%d", row), rand.Intn(10)))
	}
	// Test add pivot table with table data range
	opts := PivotTableOptions{
		DataRange:           "Table1",
		PivotTableRange:     "Sheet1!G2:K7",
		Rows:                []PivotTableField{{Data: "Column1"}},
		Columns:             []PivotTableField{{Data: "Column2"}},
		RowGrandTotals:      true,
		ColGrandTotals:      true,
		ShowDrill:           true,
		ShowRowHeaders:      true,
		ShowColHeaders:      true,
		ShowLastColumn:      true,
		ShowError:           true,
		PivotTableStyleName: "PivotStyleLight16",
	}
	assert.NoError(t, f.AddPivotTable(&opts))
	assert.NoError(t, f.DeletePivotTable("Sheet1", "PivotTable1"))
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddPivotTable2.xlsx")))
	assert.NoError(t, f.Close())

	assert.NoError(t, f.AddPivotTable(&opts))

	// Test delete pivot table with unsupported table relationships charset
	f.Pkg.Store("xl/tables/table1.xml", MacintoshCyrillicCharset)
	assert.EqualError(t, f.DeletePivotTable("Sheet1", "PivotTable1"), "XML syntax error on line 1: invalid UTF-8")

	// Test delete pivot table with unsupported worksheet relationships charset
	f.Relationships.Delete("xl/worksheets/_rels/sheet1.xml.rels")
	f.Pkg.Store("xl/worksheets/_rels/sheet1.xml.rels", MacintoshCyrillicCharset)
	assert.EqualError(t, f.DeletePivotTable("Sheet1", "PivotTable1"), "XML syntax error on line 1: invalid UTF-8")

	// Test delete pivot table without worksheet relationships
	f.Relationships.Delete("xl/worksheets/_rels/sheet1.xml.rels")
	f.Pkg.Delete("xl/worksheets/_rels/sheet1.xml.rels")
	assert.EqualError(t, f.DeletePivotTable("Sheet1", "PivotTable1"), "table PivotTable1 does not exist")

	t.Run("data_range_with_empty_column", func(t *testing.T) {
		// Test add pivot table with data range doesn't organized as a list with labeled columns
		f := NewFile()
		// Create some data in a sheet
		month := []string{"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"}
		types := []string{"Meat", "Dairy", "Beverages", "Produce"}
		assert.NoError(t, f.SetSheetRow("Sheet1", "A1", &[]string{"Month", "", "Type"}))
		for row := 2; row < 32; row++ {
			assert.NoError(t, f.SetCellValue("Sheet1", fmt.Sprintf("A%d", row), month[rand.Intn(12)]))