package xlsx import ( // "bytes" // "encoding/xml" // "strconv" "strings" . "gopkg.in/check.v1" ) type LibSuite struct {} var _ = Suite(&LibSuite{}) // Test we can correctly open a XSLX file and return a xlsx.File // struct. func (l *LibSuite) TestOpenFile(c *C) { var xlsxFile *File var error error xlsxFile, error = OpenFile("testfile.xlsx") c.Assert(error, IsNil) c.Assert(xlsxFile, NotNil) } // Test that when we open a real XLSX file we create xlsx.Sheet // objects for the sheets inside the file and that these sheets are // themselves correct. func (l *LibSuite) TestCreateSheet(c *C) { var xlsxFile *File var err error var sheet *Sheet var row *Row xlsxFile, err = OpenFile("testfile.xlsx") c.Assert(err, IsNil) c.Assert(xlsxFile, NotNil) sheetLen := len(xlsxFile.Sheets) c.Assert(sheetLen, Equals, 3) sheet = xlsxFile.Sheets[0] rowLen := len(sheet.Rows) c.Assert(rowLen, Equals, 2) row = sheet.Rows[0] c.Assert(len(row.Cells), Equals, 2) cell := row.Cells[0] cellstring := cell.String() c.Assert(cellstring, Equals, "Foo") } // Test that GetStyle correctly converts the xlsxStyle.Fonts. func (l *LibSuite) TestGetStyleWithFonts(c *C) { var cell *Cell var style *Style var xStyles *xlsxStyles var fonts []xlsxFont var cellXfs []xlsxXf fonts = make([]xlsxFont, 1) fonts[0] = xlsxFont{ Sz: xlsxVal{Val: "10"}, Name: xlsxVal{Val: "Calibra"}} cellXfs = make([]xlsxXf, 1) cellXfs[0] = xlsxXf{ApplyFont: true, FontId: 0} xStyles = &xlsxStyles{Fonts: fonts, CellXfs: cellXfs} cell = &Cell{Value: "123", styleIndex: 1, styles: xStyles} style = cell.GetStyle() c.Assert(style, NotNil) c.Assert(style.Font.Size, Equals, 10) c.Assert(style.Font.Name, Equals, "Calibra") } // Test that GetStyle correctly converts the xlsxStyle.Fills. func (l *LibSuite) TestGetStyleWithFills(c *C) { var cell *Cell var style *Style var xStyles *xlsxStyles var fills []xlsxFill var cellXfs []xlsxXf fills = make([]xlsxFill, 1) fills[0] = xlsxFill{ PatternFill: xlsxPatternFill{ PatternType: "solid", FgColor: xlsxColor{RGB: "FF000000"}, BgColor: xlsxColor{RGB: "00FF0000"}}} cellXfs = make([]xlsxXf, 1) cellXfs[0] = xlsxXf{ApplyFill: true, FillId: 0} xStyles = &xlsxStyles{Fills: fills, CellXfs: cellXfs} cell = &Cell{Value: "123", styleIndex: 1, styles: xStyles} style = cell.GetStyle() fill := style.Fill c.Assert(fill.PatternType, Equals, "solid") c.Assert(fill.BgColor, Equals, "00FF0000") c.Assert(fill.FgColor, Equals, "FF000000") } // Test that GetStyle correctly converts the xlsxStyle.Borders. func (l *LibSuite) TestGetStyleWithBorders(c *C) { var cell *Cell var style *Style var xStyles *xlsxStyles var borders []xlsxBorder var cellXfs []xlsxXf borders = make([]xlsxBorder, 1) borders[0] = xlsxBorder{ Left: xlsxLine{Style: "thin"}, Right: xlsxLine{Style: "thin"}, Top: xlsxLine{Style: "thin"}, Bottom: xlsxLine{Style: "thin"}} cellXfs = make([]xlsxXf, 1) cellXfs[0] = xlsxXf{ApplyBorder: true, BorderId: 0} xStyles = &xlsxStyles{Borders: borders, CellXfs: cellXfs} cell = &Cell{Value: "123", styleIndex: 1, styles: xStyles} style = cell.GetStyle() border := style.Border c.Assert(border.Left, Equals, "thin") c.Assert(border.Right, Equals, "thin") c.Assert(border.Top, Equals, "thin") c.Assert(border.Bottom, Equals, "thin") } // Test that we can correctly extract a reference table from the // sharedStrings.xml file embedded in the XLSX file and return a // reference table of string values from it. func (l *LibSuite) TestReadSharedStringsFromZipFile(c *C) { var xlsxFile *File var err error xlsxFile, err = OpenFile("testfile.xlsx") c.Assert(err, IsNil) c.Assert(xlsxFile.referenceTable, NotNil) } // Helper function used to test contents of a given xlsxXf against // expectations. func testXf(c *C, result, expected *xlsxXf) { c.Assert(result.ApplyAlignment, Equals, expected.ApplyAlignment) c.Assert(result.ApplyBorder, Equals, expected.ApplyBorder) c.Assert(result.ApplyFont, Equals, expected.ApplyFont) c.Assert(result.ApplyFill, Equals, expected.ApplyFill) c.Assert(result.ApplyProtection, Equals, expected.ApplyProtection) c.Assert(result.BorderId, Equals, expected.BorderId) c.Assert(result.FillId, Equals, expected.FillId) c.Assert(result.FontId, Equals, expected.FontId) c.Assert(result.NumFmtId, Equals, expected.NumFmtId) } // We can correctly extract a style table from the style.xml file // embedded in the XLSX file and return a styles struct from it. func (l *LibSuite) TestReadStylesFromZipFile(c *C) { var xlsxFile *File var err error var fontCount, fillCount, borderCount, cellStyleXfCount, cellXfCount int var font xlsxFont var fill xlsxFill var border xlsxBorder var xf xlsxXf xlsxFile, err = OpenFile("testfile.xlsx") c.Assert(err, IsNil) c.Assert(xlsxFile.styles, NotNil) fontCount = len(xlsxFile.styles.Fonts) c.Assert(fontCount, Equals, 4) font = xlsxFile.styles.Fonts[0] c.Assert(font.Sz.Val, Equals, "11") c.Assert(font.Name.Val, Equals, "Calibri") fillCount = len(xlsxFile.styles.Fills) c.Assert(fillCount, Equals, 3) fill = xlsxFile.styles.Fills[2] c.Assert(fill.PatternFill.PatternType, Equals, "solid") borderCount = len(xlsxFile.styles.Borders) c.Assert(borderCount, Equals, 2) border = xlsxFile.styles.Borders[1] c.Assert(border.Left.Style, Equals, "thin") c.Assert(border.Right.Style, Equals, "thin") c.Assert(border.Top.Style, Equals, "thin") c.Assert(border.Bottom.Style, Equals, "thin") cellStyleXfCount = len(xlsxFile.styles.CellStyleXfs) c.Assert(cellStyleXfCount, Equals, 20) xf = xlsxFile.styles.CellStyleXfs[0] expectedXf := &xlsxXf{ ApplyAlignment: true, ApplyBorder: true, ApplyFont: true, ApplyFill: false, ApplyProtection: true, BorderId: 0, FillId: 0, FontId: 0, NumFmtId: 164} testXf(c, &xf, expectedXf) cellXfCount = len(xlsxFile.styles.CellXfs) c.Assert(cellXfCount, Equals, 3) xf = xlsxFile.styles.CellXfs[0] expectedXf = &xlsxXf{ ApplyAlignment: false, ApplyBorder: false, ApplyFont: false, ApplyFill: false, ApplyProtection: false, BorderId: 0, FillId: 0, FontId: 0, NumFmtId: 164} testXf(c, &xf, expectedXf) } // We can correctly extract a map of relationship Ids to the worksheet files in // which they are contained from the XLSX file. func (l *LibSuite) TestReadWorkbookRelationsFromZipFile(c *C) { var xlsxFile *File var err error xlsxFile, err = OpenFile("testfile.xlsx") c.Assert(err, IsNil) sheetCount := len(xlsxFile.Sheet) c.Assert(sheetCount, Equals, 3) } // which they are contained from the XLSX file, even when the // worksheet files have arbitrary, non-numeric names. func (l *LibSuite) TestReadWorkbookRelationsFromZipFileWithFunnyNames(c *C) { var xlsxFile *File var err error xlsxFile, err = OpenFile("testrels.xlsx") c.Assert(err, IsNil) sheetCount := len(xlsxFile.Sheet) c.Assert(sheetCount, Equals, 2) bob := xlsxFile.Sheet["Bob"] row1 := bob.Rows[0] cell1 := row1.Cells[0] c.Assert(cell1.String(), Equals, "I am Bob") } func (l *LibSuite) TestLettersToNumeric(c *C) { cases := map[string]int{"A": 0, "G": 6, "z": 25, "AA": 26, "Az": 51, "BA": 52, "Bz": 77, "ZA": 26*26 + 0, "ZZ": 26*26 + 25, "AAA": 26*26 + 26 + 0, "AMI": 1022} for input, ans := range cases { output := lettersToNumeric(input) c.Assert(output, Equals, ans) } } func (l *LibSuite) TestLetterOnlyMapFunction(c *C) { var input string = "ABC123" var output string = strings.Map(letterOnlyMapF, input) c.Assert(output, Equals, "ABC") input = "abc123" output = strings.Map(letterOnlyMapF, input) c.Assert(output, Equals, "ABC") } func (l *LibSuite) TestIntOnlyMapFunction(c *C) { var input string = "ABC123" var output string = strings.Map(intOnlyMapF, input) c.Assert(output, Equals, "123") } // func (l *LibSuite) TestGetCoordsFromCellIDString(c *C) { // var cellIDString string = "A3" // var x, y int // var err error // x, y, err = getCoordsFromCellIDString(cellIDString) // if err != nil { // t.Error(err) // } // if x != 0 { // t.Error("Expected x == 0, but got ", strconv.Itoa(x)) // } // if y != 2 { // t.Error("Expected y == 2, but got ", strconv.Itoa(y)) // } // } // func (l *LibSuite) TestGetMaxMinFromDimensionRef(c *C) { // var dimensionRef string = "A1:B2" // var minx, miny, maxx, maxy int // var err error // minx, miny, maxx, maxy, err = getMaxMinFromDimensionRef(dimensionRef) // if err != nil { // t.Error(err) // } // if minx != 0 { // t.Error("Expected minx == 0, but got ", strconv.Itoa(minx)) // } // if miny != 0 { // t.Error("Expected miny == 0, but got ", strconv.Itoa(miny)) // } // if maxx != 1 { // t.Error("Expected maxx == 0, but got ", strconv.Itoa(maxx)) // } // if maxy != 1 { // t.Error("Expected maxy == 0, but got ", strconv.Itoa(maxy)) // } // } // func (l *LibSuite) TestGetRangeFromString(c *C) { // var rangeString string // var lower, upper int // var err error // rangeString = "1:3" // lower, upper, err = getRangeFromString(rangeString) // if err != nil { // t.Error(err) // } // if lower != 1 { // t.Error("Expected lower bound == 1, but got ", strconv.Itoa(lower)) // } // if upper != 3 { // t.Error("Expected upper bound == 3, but got ", strconv.Itoa(upper)) // } // } // func (l *LibSuite) TestMakeRowFromSpan(c *C) { // var rangeString string // var row *Row // var length int // rangeString = "1:3" // row = makeRowFromSpan(rangeString) // length = len(row.Cells) // if length != 3 { // t.Error("Expected a row with 3 cells, but got ", strconv.Itoa(length)) // } // rangeString = "5:7" // Note - we ignore lower bound! // row = makeRowFromSpan(rangeString) // length = len(row.Cells) // if length != 7 { // t.Error("Expected a row with 7 cells, but got ", strconv.Itoa(length)) // } // rangeString = "1:1" // row = makeRowFromSpan(rangeString) // length = len(row.Cells) // if length != 1 { // t.Error("Expected a row with 1 cells, but got ", strconv.Itoa(length)) // } // } // func (l *LibSuite) TestReadRowsFromSheet(c *C) { // var sharedstringsXML = bytes.NewBufferString(` // // // // Foo // // // Bar // // // Baz // // // Quuk // // `) // var sheetxml = bytes.NewBufferString(` // // // // // // // // // // // // // 0 // // // 1 // // // // // 2 // // // 3 // // // // // `) // worksheet := new(xlsxWorksheet) // err := xml.NewDecoder(sheetxml).Decode(worksheet) // if err != nil { // t.Error(err.Error()) // return // } // sst := new(xlsxSST) // err = xml.NewDecoder(sharedstringsXML).Decode(sst) // if err != nil { // t.Error(err.Error()) // return // } // file := new(File) // file.referenceTable = MakeSharedStringRefTable(sst) // rows, maxCols, maxRows := readRowsFromSheet(worksheet, file) // if maxRows != 2 { // t.Error("Expected maxRows == 2") // } // if maxCols != 2 { // t.Error("Expected maxCols == 2") // } // row := rows[0] // if len(row.Cells) != 2 { // t.Error("Expected len(row.Cells) == 2, got ", strconv.Itoa(len(row.Cells))) // } // cell1 := row.Cells[0] // if cell1.String() != "Foo" { // t.Error("Expected cell1.String() == 'Foo', got ", cell1.String()) // } // cell2 := row.Cells[1] // if cell2.String() != "Bar" { // t.Error("Expected cell2.String() == 'Bar', got ", cell2.String()) // } // } // func (l *LibSuite) TestReadRowsFromSheetWithLeadingEmptyRows(c *C) { // var sharedstringsXML = bytes.NewBufferString(` // ABCDEF`) // var sheetxml = bytes.NewBufferString(` // // // // // // // // // // // // 0 // // // // // 1 // // // // // // // // // // // // `) // worksheet := new(xlsxWorksheet) // err := xml.NewDecoder(sheetxml).Decode(worksheet) // if err != nil { // t.Error(err.Error()) // return // } // sst := new(xlsxSST) // err = xml.NewDecoder(sharedstringsXML).Decode(sst) // if err != nil { // t.Error(err.Error()) // return // } // file := new(File) // file.referenceTable = MakeSharedStringRefTable(sst) // rows, maxCols, maxRows := readRowsFromSheet(worksheet, file) // if maxRows != 2 { // t.Error("Expected maxRows == 2, got ", strconv.Itoa(len(rows))) // } // if maxCols != 1 { // t.Error("Expected maxCols == 1, got ", strconv.Itoa(maxCols)) // } // } // func (l *LibSuite) TestReadRowsFromSheetWithEmptyCells(c *C) { // var sharedstringsXML = bytes.NewBufferString(` // // // // Bob // // // Alice // // // Sue // // // Yes // // // No // // // `) // var sheetxml = bytes.NewBufferString(` // // // // // // // 0 // // // // // 1 // // // // // 2 // // // // // // // 3 // // // // // 4 // // // // // 3 // // // // // // // 4 // // // // // 3 // // // // // // // `) // worksheet := new(xlsxWorksheet) // err := xml.NewDecoder(sheetxml).Decode(worksheet) // if err != nil { // t.Error(err.Error()) // return // } // sst := new(xlsxSST) // err = xml.NewDecoder(sharedstringsXML).Decode(sst) // if err != nil { // t.Error(err.Error()) // return // } // file := new(File) // file.referenceTable = MakeSharedStringRefTable(sst) // rows, maxCols, maxRows := readRowsFromSheet(worksheet, file) // if maxRows != 3 { // t.Error("Expected maxRows == 3, got ", strconv.Itoa(len(rows))) // } // if maxCols != 3 { // t.Error("Expected maxCols == 3, got ", strconv.Itoa(maxCols)) // } // row := rows[2] // if len(row.Cells) != 3 { // t.Error("Expected len(row.Cells) == 3, got ", strconv.Itoa(len(row.Cells))) // } // cell1 := row.Cells[0] // if cell1.String() != "No" { // t.Error("Expected cell1.String() == 'No', got ", cell1.String()) // } // cell2 := row.Cells[1] // if cell2.String() != "" { // t.Error("Expected cell2.String() == '', got ", cell2.String()) // } // cell3 := row.Cells[2] // if cell3.String() != "Yes" { // t.Error("Expected cell3.String() == 'Yes', got ", cell3.String()) // } // } // func (l *LibSuite) TestReadRowsFromSheetWithTrailingEmptyCells(c *C) { // var row *Row // var cell1, cell2, cell3, cell4 *Cell // var sharedstringsXML = bytes.NewBufferString(` // // ABCD`) // var sheetxml = bytes.NewBufferString(` // // 01231111111 // `) // worksheet := new(xlsxWorksheet) // err := xml.NewDecoder(sheetxml).Decode(worksheet) // if err != nil { // t.Error(err.Error()) // return // } // sst := new(xlsxSST) // err = xml.NewDecoder(sharedstringsXML).Decode(sst) // if err != nil { // t.Error(err.Error()) // return // } // file := new(File) // file.referenceTable = MakeSharedStringRefTable(sst) // rows, maxCol, maxRow := readRowsFromSheet(worksheet, file) // if maxCol != 4 { // t.Error("Expected maxCol == 4, got ", strconv.Itoa(maxCol)) // } // if maxRow != 8 { // t.Error("Expected maxRow == 8, got ", strconv.Itoa(maxRow)) // } // row = rows[0] // if len(row.Cells) != 4 { // t.Error("Expected len(row.Cells) == 4, got ", strconv.Itoa(len(row.Cells))) // } // cell1 = row.Cells[0] // if cell1.String() != "A" { // t.Error("Expected cell1.String() == 'A', got ", cell1.String()) // } // cell2 = row.Cells[1] // if cell2.String() != "B" { // t.Error("Expected cell2.String() == 'B', got ", cell2.String()) // } // cell3 = row.Cells[2] // if cell3.String() != "C" { // t.Error("Expected cell3.String() == 'C', got ", cell3.String()) // } // cell4 = row.Cells[3] // if cell4.String() != "D" { // t.Error("Expected cell4.String() == 'D', got ", cell4.String()) // } // row = rows[1] // if len(row.Cells) != 4 { // t.Error("Expected len(row.Cells) == 4, got ", strconv.Itoa(len(row.Cells))) // } // cell1 = row.Cells[0] // if cell1.String() != "1" { // t.Error("Expected cell1.String() == '1', got ", cell1.String()) // } // cell2 = row.Cells[1] // if cell2.String() != "" { // t.Error("Expected cell2.String() == '', got ", cell2.String()) // } // cell3 = row.Cells[2] // if cell3.String() != "" { // t.Error("Expected cell3.String() == '', got ", cell3.String()) // } // cell4 = row.Cells[3] // if cell4.String() != "" { // t.Error("Expected cell4.String() == '', got ", cell4.String()) // } // }