summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--excelize_test.go101
-rw-r--r--rows.go9
-rw-r--r--rows_test.go42
3 files changed, 77 insertions, 75 deletions
diff --git a/excelize_test.go b/excelize_test.go
index 9587f33..b4434b5 100644
--- a/excelize_test.go
+++ b/excelize_test.go
@@ -6,6 +6,7 @@ import (
_ "image/jpeg"
_ "image/png"
"io/ioutil"
+ "reflect"
"strconv"
"strings"
"testing"
@@ -16,7 +17,7 @@ func TestOpenFile(t *testing.T) {
// Test update a XLSX file.
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
// Test get all the rows in a not exists worksheet.
rows := xlsx.GetRows("Sheet4")
@@ -130,17 +131,17 @@ func TestOpenFile(t *testing.T) {
func TestAddPicture(t *testing.T) {
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
// Test add picture to worksheet with offset and location hyperlink.
err = xlsx.AddPicture("Sheet2", "I9", "./test/images/excel.jpg", `{"x_offset": 140, "y_offset": 120, "hyperlink": "#Sheet2!D8", "hyperlink_type": "Location"}`)
if err != nil {
- t.Log(err)
+ t.Error(err)
}
// Test add picture to worksheet with offset, external hyperlink and positioning.
err = xlsx.AddPicture("Sheet1", "F21", "./test/images/excel.png", `{"x_offset": 10, "y_offset": 10, "hyperlink": "https://github.com/360EntSecGroup-Skylar/excelize", "hyperlink_type": "External", "positioning": "oneCell"}`)
if err != nil {
- t.Log(err)
+ t.Error(err)
}
// Test add picture to worksheet with invalid file path.
err = xlsx.AddPicture("Sheet1", "G21", "./test/images/excel.icon", "")
@@ -155,7 +156,7 @@ func TestAddPicture(t *testing.T) {
// Test write file to given path.
err = xlsx.SaveAs("./test/Book2.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
@@ -199,15 +200,15 @@ func TestNewFile(t *testing.T) {
// Test add picture to sheet with scaling and positioning.
err := xlsx.AddPicture("Sheet1", "H2", "./test/images/excel.gif", `{"x_scale": 0.5, "y_scale": 0.5, "positioning": "absolute"}`)
if err != nil {
- t.Log(err)
+ t.Error(err)
}
err = xlsx.AddPicture("Sheet1", "C2", "./test/images/excel.png", "")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
err = xlsx.SaveAs("./test/Book3.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
@@ -219,7 +220,7 @@ func TestColWidth(t *testing.T) {
xlsx.GetColWidth("Sheet1", "C")
err := xlsx.SaveAs("./test/Book4.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
convertRowHeightToPixels(0)
}
@@ -232,7 +233,7 @@ func TestRowHeight(t *testing.T) {
t.Log(xlsx.GetRowHeight("Sheet1", 0))
err := xlsx.SaveAs("./test/Book5.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
convertColWidthToPixels(0)
}
@@ -252,14 +253,14 @@ func TestSetCellHyperLink(t *testing.T) {
xlsx.SetCellHyperLink("Sheet2", "", "Sheet1!D60", "Location")
err = xlsx.Save()
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
func TestGetCellHyperLink(t *testing.T) {
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
link, target := xlsx.GetCellHyperLink("Sheet1", "")
t.Log(link, target)
@@ -274,7 +275,7 @@ func TestGetCellHyperLink(t *testing.T) {
func TestSetCellFormula(t *testing.T) {
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
xlsx.SetCellFormula("Sheet1", "B19", "SUM(Sheet2!D2,Sheet2!D11)")
xlsx.SetCellFormula("Sheet1", "C19", "SUM(Sheet2!D2,Sheet2!D9)")
@@ -282,14 +283,14 @@ func TestSetCellFormula(t *testing.T) {
xlsx.SetCellFormula("Sheet1", "C", "SUM(Sheet2!D2,Sheet2!D9)")
err = xlsx.Save()
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
func TestSetSheetBackground(t *testing.T) {
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
err = xlsx.SetSheetBackground("Sheet2", "./test/images/background.png")
if err != nil {
@@ -301,22 +302,22 @@ func TestSetSheetBackground(t *testing.T) {
}
err = xlsx.SetSheetBackground("Sheet2", "./test/images/background.jpg")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
err = xlsx.SetSheetBackground("Sheet2", "./test/images/background.jpg")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
err = xlsx.Save()
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
func TestMergeCell(t *testing.T) {
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
xlsx.MergeCell("Sheet1", "D9", "D9")
xlsx.MergeCell("Sheet1", "D9", "E9")
@@ -336,19 +337,19 @@ func TestMergeCell(t *testing.T) {
xlsx.GetCellFormula("Sheet1", "G12")
err = xlsx.Save()
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
func TestSetCellStyleAlignment(t *testing.T) {
xlsx, err := OpenFile("./test/Book2.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
var style int
style, err = xlsx.NewStyle(`{"alignment":{"horizontal":"center","ident":1,"justify_last_line":true,"reading_order":0,"relative_indent":1,"shrink_to_fit":true,"text_rotation":45,"vertical":"top","wrap_text":true}}`)
if err != nil {
- t.Log(err)
+ t.Error(err)
}
xlsx.SetCellStyle("Sheet1", "A22", "A22", style)
// Test set cell style with given illegal rows number.
@@ -358,14 +359,14 @@ func TestSetCellStyleAlignment(t *testing.T) {
xlsx.GetCellStyle("Sheet1", "A")
err = xlsx.Save()
if err != nil {
- t.Log(err)
+ t.Error(err)
}
}
func TestSetCellStyleBorder(t *testing.T) {
xlsx, err := OpenFile("./test/Book2.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
var style int
// Test set border with invalid style parameter.
@@ -835,7 +836,7 @@ func TestAutoFilter(t *testing.T) {
func TestAddChart(t *testing.T) {
xlsx, err := OpenFile("./test/Book1.xlsx")
if err != nil {
- t.Log(err)
+ t.Error(err)
}
categories := map[string]string{"A30": "Small", "A31": "Normal", "A32": "Large", "B29": "Apple", "C29": "Orange", "D29": "Pear"}
values := map[string]int{"B30": 2, "C30": 3, "D30": 3, "B31": 5, "C31": 2, "D31": 4, "B32": 6, "C32": 7, "D32": 8}
@@ -1062,3 +1063,53 @@ func TestSetSheetRow(t *testing.T) {
return
}
}
+
+func TestRows(t *testing.T) {
+ xlsx, err := OpenFile("./test/Book1.xlsx")
+ if err != nil {
+ t.Error(err)
+ return
+ }
+ rows, err := xlsx.Rows("Sheet2")
+ if err != nil {
+ t.Error(err)
+ return
+ }
+ rowStrs := make([][]string, 0)
+ var i = 0
+ for rows.Next() {
+ i++
+ columns := rows.Columns()
+ rowStrs = append(rowStrs, columns)
+ }
+ if rows.Error() != nil {
+ t.Error(rows.Error())
+ return
+ }
+ dstRows := xlsx.GetRows("Sheet2")
+ if len(dstRows) != len(rowStrs) {
+ t.Error("values not equal")
+ return
+ }
+ for i := 0; i < len(rowStrs); i++ {
+ if !reflect.DeepEqual(trimSliceSpace(dstRows[i]), trimSliceSpace(rowStrs[i])) {
+ t.Error("values not equal")
+ return
+ }
+ }
+ rows, err = xlsx.Rows("SheetN")
+ if err != nil {
+ t.Log(err)
+ }
+}
+
+func trimSliceSpace(s []string) []string {
+ for {
+ if len(s) > 0 && s[len(s)-1] == "" {
+ s = s[:len(s)-1]
+ } else {
+ break
+ }
+ }
+ return s
+}
diff --git a/rows.go b/rows.go
index 02a6876..03e442a 100644
--- a/rows.go
+++ b/rows.go
@@ -108,11 +108,9 @@ func (rows *Rows) Columns() []string {
if rows.token == nil {
return []string{}
}
-
startElement := rows.token.(xml.StartElement)
r := xlsxRow{}
rows.decoder.DecodeElement(&r, &startElement)
-
d := rows.f.sharedStringsReader()
row := make([]string, len(r.C), len(r.C))
for _, colCell := range r.C {
@@ -135,7 +133,6 @@ func (err ErrSheetNotExist) Error() string {
// Rows return a rows iterator. For example:
//
// rows, err := xlsx.GetRows("Sheet1")
-//
// for rows.Next() {
// for _, colCell := range rows.Columns() {
// fmt.Print(colCell, "\t")
@@ -150,13 +147,9 @@ func (f *File) Rows(sheet string) (*Rows, error) {
return nil, ErrSheetNotExist{sheet}
}
if xlsx != nil {
- output, err := xml.Marshal(f.Sheet[name])
- if err != nil {
- return nil, err
- }
+ output, _ := xml.Marshal(f.Sheet[name])
f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpace(string(output)))
}
-
return &Rows{
f: f,
decoder: xml.NewDecoder(strings.NewReader(f.readXML(name))),
diff --git a/rows_test.go b/rows_test.go
deleted file mode 100644
index eadf077..0000000
--- a/rows_test.go
+++ /dev/null
@@ -1,42 +0,0 @@
-package excelize
-
-import (
- "testing"
-
- "github.com/stretchr/testify/assert"
-)
-
-func trimSliceSpace(s []string) []string {
- for {
- if len(s) > 0 && s[len(s)-1] == "" {
- s = s[:len(s)-1]
- } else {
- break
- }
- }
- return s
-}
-
-func TestRows(t *testing.T) {
- xlsx, err := OpenFile("./test/Book1.xlsx")
- assert.NoError(t, err)
-
- rows, err := xlsx.Rows("Sheet2")
- assert.NoError(t, err)
-
- rowStrs := make([][]string, 0)
- var i = 0
- for rows.Next() {
- i++
- columns := rows.Columns()
- //fmt.Println(i, columns)
- rowStrs = append(rowStrs, columns)
- }
- assert.NoError(t, rows.Error())
-
- dstRows := xlsx.GetRows("Sheet2")
- assert.EqualValues(t, len(dstRows), len(rowStrs))
- for i := 0; i < len(rowStrs); i++ {
- assert.EqualValues(t, trimSliceSpace(dstRows[i]), trimSliceSpace(rowStrs[i]))
- }
-}