rows_test.go 38.1 KB
Newer Older
1 2 3
package excelize

import (
xurime's avatar
xurime 已提交
4
	"bytes"
5
	"encoding/xml"
6 7 8 9 10
	"fmt"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
D
ducquangkstn 已提交
11
	"github.com/stretchr/testify/require"
12 13
)

14 15 16 17 18 19 20 21 22 23
func TestGetRows(t *testing.T) {
	f := NewFile()
	assert.NoError(t, f.SetCellValue("Sheet1", "A1", "A1"))
	// Test get rows with unsupported charset shared strings table
	f.SharedStrings = nil
	f.Pkg.Store(defaultXMLPathSharedStrings, MacintoshCyrillicCharset)
	_, err := f.GetRows("Sheet1")
	assert.NoError(t, err)
}

24
func TestRows(t *testing.T) {
25
	const sheet2 = "Sheet2"
xurime's avatar
xurime 已提交
26
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
27
	assert.NoError(t, err)
28

29 30 31
	// Test get rows with invalid sheet name
	_, err = f.Rows("Sheet:1")
	assert.EqualError(t, err, ErrSheetNameInvalid.Error())
32

33 34
	rows, err := f.Rows(sheet2)
	assert.NoError(t, err)
35
	var collectedRows [][]string
36
	for rows.Next() {
37 38 39
		columns, err := rows.Columns()
		assert.NoError(t, err)
		collectedRows = append(collectedRows, trimSliceSpace(columns))
40 41 42 43
	}
	if !assert.NoError(t, rows.Error()) {
		t.FailNow()
	}
44
	assert.NoError(t, rows.Close())
45

xurime's avatar
xurime 已提交
46
	returnedRows, err := f.GetRows(sheet2)
47
	assert.NoError(t, err)
48 49
	for i := range returnedRows {
		returnedRows[i] = trimSliceSpace(returnedRows[i])
50
	}
51 52
	if !assert.Equal(t, collectedRows, returnedRows) {
		t.FailNow()
53
	}
54
	assert.NoError(t, f.Close())
xurime's avatar
xurime 已提交
55

56
	f.Pkg.Store("xl/worksheets/sheet1.xml", nil)
57 58
	_, err = f.Rows("Sheet1")
	assert.NoError(t, err)
59

60
	// Test reload the file to memory from system temporary directory
61
	f, err = OpenFile(filepath.Join("test", "Book1.xlsx"), Options{UnzipXMLSizeLimit: 128})
62 63 64 65
	assert.NoError(t, err)
	value, err := f.GetCellValue("Sheet1", "A19")
	assert.NoError(t, err)
	assert.Equal(t, "Total:", value)
66
	// Test load shared string table to memory
67 68 69 70 71 72
	err = f.SetCellValue("Sheet1", "A19", "A19")
	assert.NoError(t, err)
	value, err = f.GetCellValue("Sheet1", "A19")
	assert.NoError(t, err)
	assert.Equal(t, "A19", value)
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetRow.xlsx")))
73
	assert.NoError(t, f.Close())
74

75
	// Test rows iterator with unsupported charset shared strings table
76 77 78 79 80 81
	f.SharedStrings = nil
	f.Pkg.Store(defaultXMLPathSharedStrings, MacintoshCyrillicCharset)
	rows, err = f.Rows(sheet2)
	assert.NoError(t, err)
	_, err = rows.Columns()
	assert.EqualError(t, err, "XML syntax error on line 1: invalid UTF-8")
82 83
}

D
ducquangkstn 已提交
84
func TestRowsIterator(t *testing.T) {
85
	sheetName, rowCount, expectedNumRow := "Sheet2", 0, 11
86
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
D
ducquangkstn 已提交
87 88
	require.NoError(t, err)

89
	rows, err := f.Rows(sheetName)
D
ducquangkstn 已提交
90
	require.NoError(t, err)
91

D
ducquangkstn 已提交
92 93 94 95 96
	for rows.Next() {
		rowCount++
		require.True(t, rowCount <= expectedNumRow, "rowCount is greater than expected")
	}
	assert.Equal(t, expectedNumRow, rowCount)
97 98
	assert.NoError(t, rows.Close())
	assert.NoError(t, f.Close())
99

100
	// Valued cell sparse distribution test
101
	f, sheetName, rowCount, expectedNumRow = NewFile(), "Sheet1", 0, 3
102 103
	cells := []string{"C1", "E1", "A3", "B3", "C3", "D3", "E3"}
	for _, cell := range cells {
104
		assert.NoError(t, f.SetCellValue(sheetName, cell, 1))
105
	}
106
	rows, err = f.Rows(sheetName)
107 108 109
	require.NoError(t, err)
	for rows.Next() {
		rowCount++
110
		require.True(t, rowCount <= expectedNumRow, "rowCount is greater than expected")
111
	}
112
	assert.Equal(t, expectedNumRow, rowCount)
D
ducquangkstn 已提交
113 114
}

115 116 117 118 119 120 121 122 123 124 125
func TestRowsGetRowOpts(t *testing.T) {
	sheetName := "Sheet2"
	expectedRowStyleID1 := RowOpts{Height: 17.0, Hidden: false, StyleID: 1}
	expectedRowStyleID2 := RowOpts{Height: 17.0, Hidden: false, StyleID: 0}
	expectedRowStyleID3 := RowOpts{Height: 17.0, Hidden: false, StyleID: 2}
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
	require.NoError(t, err)

	rows, err := f.Rows(sheetName)
	require.NoError(t, err)

126 127 128 129 130 131 132 133 134 135 136 137 138
	assert.Equal(t, true, rows.Next())
	_, err = rows.Columns()
	require.NoError(t, err)
	rowOpts := rows.GetRowOpts()
	assert.Equal(t, expectedRowStyleID1, rowOpts)
	assert.Equal(t, true, rows.Next())
	rowOpts = rows.GetRowOpts()
	assert.Equal(t, expectedRowStyleID2, rowOpts)
	assert.Equal(t, true, rows.Next())
	_, err = rows.Columns()
	require.NoError(t, err)
	rowOpts = rows.GetRowOpts()
	assert.Equal(t, expectedRowStyleID3, rowOpts)
139 140
}

141
func TestRowsError(t *testing.T) {
142
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
143 144 145
	if !assert.NoError(t, err) {
		t.FailNow()
	}
146
	_, err = f.Rows("SheetN")
147
	assert.EqualError(t, err, "sheet SheetN does not exist")
148
	assert.NoError(t, f.Close())
149 150 151
}

func TestRowHeight(t *testing.T) {
152 153
	f := NewFile()
	sheet1 := f.GetSheetName(0)
154

155
	assert.EqualError(t, f.SetRowHeight(sheet1, 0, defaultRowHeightPixels+1.0), newInvalidRowNumberError(0).Error())
156

157
	_, err := f.GetRowHeight("Sheet1", 0)
158
	assert.EqualError(t, err, newInvalidRowNumberError(0).Error())
159

160 161
	assert.NoError(t, f.SetRowHeight(sheet1, 1, 111.0))
	height, err := f.GetRowHeight(sheet1, 1)
162 163
	assert.NoError(t, err)
	assert.Equal(t, 111.0, height)
164

165
	// Test set row height overflow max row height limit
166
	assert.EqualError(t, f.SetRowHeight(sheet1, 4, MaxRowHeight+1), ErrMaxRowHeight.Error())
167

168
	// Test get row height that rows index over exists rows
169
	height, err = f.GetRowHeight(sheet1, 5)
170
	assert.NoError(t, err)
171
	assert.Equal(t, defaultRowHeight, height)
172

173
	// Test get row height that rows heights haven't changed
174
	height, err = f.GetRowHeight(sheet1, 3)
175
	assert.NoError(t, err)
176
	assert.Equal(t, defaultRowHeight, height)
177

178
	// Test set and get row height on not exists worksheet
179
	assert.EqualError(t, f.SetRowHeight("SheetN", 1, 111.0), "sheet SheetN does not exist")
180
	_, err = f.GetRowHeight("SheetN", 3)
181
	assert.EqualError(t, err, "sheet SheetN does not exist")
xurime's avatar
xurime 已提交
182

183 184 185 186 187 188 189 190
	// Test set row height with invalid sheet name
	assert.EqualError(t, f.SetRowHeight("Sheet:1", 1, 10.0), ErrSheetNameInvalid.Error())

	// Test get row height with invalid sheet name
	_, err = f.GetRowHeight("Sheet:1", 3)
	assert.EqualError(t, err, ErrSheetNameInvalid.Error())

	// Test get row height with custom default row height
191 192 193 194
	assert.NoError(t, f.SetSheetProps(sheet1, &SheetPropsOptions{
		DefaultRowHeight: float64Ptr(30.0),
		CustomHeight:     boolPtr(true),
	}))
xurime's avatar
xurime 已提交
195 196 197 198
	height, err = f.GetRowHeight(sheet1, 100)
	assert.NoError(t, err)
	assert.Equal(t, 30.0, height)

199
	// Test set row height with custom default row height with prepare XML
xurime's avatar
xurime 已提交
200 201
	assert.NoError(t, f.SetCellValue(sheet1, "A10", "A10"))

202 203
	_, err = f.NewSheet("Sheet2")
	assert.NoError(t, err)
204 205 206 207 208
	assert.NoError(t, f.SetCellValue("Sheet2", "A2", true))
	height, err = f.GetRowHeight("Sheet2", 1)
	assert.NoError(t, err)
	assert.Equal(t, 15.0, height)

209
	err = f.SaveAs(filepath.Join("test", "TestRowHeight.xlsx"))
210 211 212 213
	if !assert.NoError(t, err) {
		t.FailNow()
	}

214
	assert.Equal(t, 0.0, convertColWidthToPixels(0))
215 216
}

xurime's avatar
xurime 已提交
217 218 219 220
func TestColumns(t *testing.T) {
	f := NewFile()
	rows, err := f.Rows("Sheet1")
	assert.NoError(t, err)
221 222 223 224 225 226 227

	rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="2"><c r="A1" t="s"><v>1</v></c></row></sheetData></worksheet>`)))
	_, err = rows.Columns()
	assert.NoError(t, err)
	rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="2"><c r="A1" t="s"><v>1</v></c></row></sheetData></worksheet>`)))
	rows.curRow = 1
	_, err = rows.Columns()
228
	assert.NoError(t, err)
229

230
	rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="A"><c r="A1" t="s"><v>1</v></c></row><row r="A"><c r="2" t="inlineStr"><is><t>B</t></is></c></row></sheetData></worksheet>`)))
231
	assert.True(t, rows.Next())
xurime's avatar
xurime 已提交
232 233 234
	_, err = rows.Columns()
	assert.EqualError(t, err, `strconv.Atoi: parsing "A": invalid syntax`)

235
	rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="1"><c r="A1" t="s"><v>1</v></c></row><row r="A"><c r="2" t="inlineStr"><is><t>B</t></is></c></row></sheetData></worksheet>`)))
xurime's avatar
xurime 已提交
236 237 238 239
	_, err = rows.Columns()
	assert.NoError(t, err)

	rows.decoder = f.xmlNewDecoder(bytes.NewReader([]byte(`<worksheet><sheetData><row r="1"><c r="A" t="s"><v>1</v></c></row></sheetData></worksheet>`)))
240
	assert.True(t, rows.Next())
xurime's avatar
xurime 已提交
241
	_, err = rows.Columns()
242
	assert.Equal(t, newCellNameToCoordinatesError("A", newInvalidCellNameError("A")), err)
xurime's avatar
xurime 已提交
243 244 245 246 247 248 249 250 251

	// Test token is nil
	rows.decoder = f.xmlNewDecoder(bytes.NewReader(nil))
	_, err = rows.Columns()
	assert.NoError(t, err)
}

func TestSharedStringsReader(t *testing.T) {
	f := NewFile()
252
	// Test read shared string with unsupported charset
253
	f.Pkg.Store(defaultXMLPathSharedStrings, MacintoshCyrillicCharset)
254 255
	_, err := f.sharedStringsReader()
	assert.EqualError(t, err, "XML syntax error on line 1: invalid UTF-8")
256
	// Test read shared strings with unsupported charset content types
257 258 259 260 261
	f = NewFile()
	f.ContentTypes = nil
	f.Pkg.Store(defaultXMLPathContentTypes, MacintoshCyrillicCharset)
	_, err = f.sharedStringsReader()
	assert.EqualError(t, err, "XML syntax error on line 1: invalid UTF-8")
262
	// Test read shared strings with unsupported charset workbook relationships
263 264 265 266 267
	f = NewFile()
	f.Relationships.Delete(defaultXMLPathWorkbookRels)
	f.Pkg.Store(defaultXMLPathWorkbookRels, MacintoshCyrillicCharset)
	_, err = f.sharedStringsReader()
	assert.EqualError(t, err, "XML syntax error on line 1: invalid UTF-8")
xurime's avatar
xurime 已提交
268 269
}

270
func TestRowVisibility(t *testing.T) {
271
	f, err := prepareTestBook1()
272 273 274
	assert.NoError(t, err)
	_, err = f.NewSheet("Sheet3")
	assert.NoError(t, err)
275 276
	assert.NoError(t, f.SetRowVisible("Sheet3", 2, false))
	assert.NoError(t, f.SetRowVisible("Sheet3", 2, true))
277 278
	visible, err := f.GetRowVisible("Sheet3", 2)
	assert.Equal(t, true, visible)
279
	assert.NoError(t, err)
280 281
	visible, err = f.GetRowVisible("Sheet3", 25)
	assert.Equal(t, false, visible)
282
	assert.NoError(t, err)
283
	assert.EqualError(t, f.SetRowVisible("Sheet3", 0, true), newInvalidRowNumberError(0).Error())
284
	assert.EqualError(t, f.SetRowVisible("SheetN", 2, false), "sheet SheetN does not exist")
285 286
	// Test set row visibility with invalid sheet name
	assert.EqualError(t, f.SetRowVisible("Sheet:1", 1, false), ErrSheetNameInvalid.Error())
287

288
	visible, err = f.GetRowVisible("Sheet3", 0)
289
	assert.Equal(t, false, visible)
290
	assert.EqualError(t, err, newInvalidRowNumberError(0).Error())
291
	_, err = f.GetRowVisible("SheetN", 1)
292
	assert.EqualError(t, err, "sheet SheetN does not exist")
293 294 295
	// Test get row visibility with invalid sheet name
	_, err = f.GetRowVisible("Sheet:1", 1)
	assert.EqualError(t, err, ErrSheetNameInvalid.Error())
296
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestRowVisibility.xlsx")))
297 298
}

299
func TestRemoveRow(t *testing.T) {
xurime's avatar
xurime 已提交
300
	f := NewFile()
301
	sheet1 := f.GetSheetName(0)
xurime's avatar
xurime 已提交
302
	r, err := f.workSheetReader(sheet1)
xurime's avatar
xurime 已提交
303
	assert.NoError(t, err)
304
	const (
305 306
		colCount = 10
		rowCount = 10
307
	)
308
	assert.NoError(t, fillCells(f, sheet1, colCount, rowCount))
309

310
	assert.NoError(t, f.SetCellHyperLink(sheet1, "A5", "https://github.com/xuri/excelize", "External"))
311

312
	assert.EqualError(t, f.RemoveRow(sheet1, -1), newInvalidRowNumberError(-1).Error())
313

314
	assert.EqualError(t, f.RemoveRow(sheet1, 0), newInvalidRowNumberError(0).Error())
315

xurime's avatar
xurime 已提交
316
	assert.NoError(t, f.RemoveRow(sheet1, 4))
317 318 319 320
	if !assert.Len(t, r.SheetData.Row, rowCount-1) {
		t.FailNow()
	}

321
	assert.NoError(t, f.MergeCell(sheet1, "B3", "B5"))
322

xurime's avatar
xurime 已提交
323
	assert.NoError(t, f.RemoveRow(sheet1, 2))
324 325 326 327
	if !assert.Len(t, r.SheetData.Row, rowCount-2) {
		t.FailNow()
	}

xurime's avatar
xurime 已提交
328
	assert.NoError(t, f.RemoveRow(sheet1, 4))
329 330 331 332
	if !assert.Len(t, r.SheetData.Row, rowCount-3) {
		t.FailNow()
	}

333
	err = f.AutoFilter(sheet1, "A2:A2", []AutoFilterOptions{{Column: "A", Expression: "x != blanks"}})
334 335 336 337
	if !assert.NoError(t, err) {
		t.FailNow()
	}

xurime's avatar
xurime 已提交
338
	assert.NoError(t, f.RemoveRow(sheet1, 1))
339 340 341 342
	if !assert.Len(t, r.SheetData.Row, rowCount-4) {
		t.FailNow()
	}

xurime's avatar
xurime 已提交
343
	assert.NoError(t, f.RemoveRow(sheet1, 2))
344 345 346 347
	if !assert.Len(t, r.SheetData.Row, rowCount-5) {
		t.FailNow()
	}

xurime's avatar
xurime 已提交
348
	assert.NoError(t, f.RemoveRow(sheet1, 1))
349 350 351 352
	if !assert.Len(t, r.SheetData.Row, rowCount-6) {
		t.FailNow()
	}

xurime's avatar
xurime 已提交
353 354 355
	assert.NoError(t, f.RemoveRow(sheet1, 10))
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestRemoveRow.xlsx")))

356 357 358 359 360 361 362 363 364
	f = NewFile()
	assert.NoError(t, f.MergeCell("Sheet1", "A1", "C1"))
	assert.NoError(t, f.MergeCell("Sheet1", "A2", "C2"))
	assert.NoError(t, f.RemoveRow("Sheet1", 1))
	mergedCells, err := f.GetMergeCells("Sheet1")
	assert.NoError(t, err)
	assert.Equal(t, "A1", mergedCells[0].GetStartAxis())
	assert.Equal(t, "C1", mergedCells[0].GetEndAxis())

xurime's avatar
xurime 已提交
365
	// Test remove row on not exist worksheet
366 367 368
	assert.EqualError(t, f.RemoveRow("SheetN", 1), "sheet SheetN does not exist")
	// Test remove row with invalid sheet name
	assert.EqualError(t, f.RemoveRow("Sheet:1", 1), ErrSheetNameInvalid.Error())
369 370
}

371
func TestInsertRows(t *testing.T) {
372 373 374
	f := NewFile()
	sheet1 := f.GetSheetName(0)
	r, err := f.workSheetReader(sheet1)
xurime's avatar
xurime 已提交
375
	assert.NoError(t, err)
376
	const (
377 378
		colCount = 10
		rowCount = 10
379
	)
380
	assert.NoError(t, fillCells(f, sheet1, colCount, rowCount))
381

382
	assert.NoError(t, f.SetCellHyperLink(sheet1, "A5", "https://github.com/xuri/excelize", "External"))
383

384
	assert.NoError(t, f.InsertRows(sheet1, 1, 1))
385 386 387 388
	if !assert.Len(t, r.SheetData.Row, rowCount+1) {
		t.FailNow()
	}

389
	assert.NoError(t, f.InsertRows(sheet1, 4, 1))
390 391 392 393
	if !assert.Len(t, r.SheetData.Row, rowCount+2) {
		t.FailNow()
	}

394 395 396 397
	assert.NoError(t, f.InsertRows(sheet1, 4, 2))
	if !assert.Len(t, r.SheetData.Row, rowCount+4) {
		t.FailNow()
	}
398 399
	// Test insert rows with invalid sheet name
	assert.EqualError(t, f.InsertRows("Sheet:1", 1, 1), ErrSheetNameInvalid.Error())
400 401 402 403 404 405 406 407 408

	assert.EqualError(t, f.InsertRows(sheet1, -1, 1), newInvalidRowNumberError(-1).Error())
	assert.EqualError(t, f.InsertRows(sheet1, 0, 1), newInvalidRowNumberError(0).Error())
	assert.EqualError(t, f.InsertRows(sheet1, 4, 0), ErrParameterInvalid.Error())
	assert.EqualError(t, f.InsertRows(sheet1, 4, TotalRows), ErrMaxRows.Error())
	assert.EqualError(t, f.InsertRows(sheet1, 4, TotalRows-5), ErrMaxRows.Error())
	assert.EqualError(t, f.InsertRows(sheet1, TotalRows, 1), ErrMaxRows.Error())

	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestInsertRows.xlsx")))
409 410
}

411
// Test internal structure state after insert operations. It is important
412 413
// for insert workflow to be constant to avoid side effect with functions
// related to internal structure.
414
func TestInsertRowsInEmptyFile(t *testing.T) {
415 416 417
	f := NewFile()
	sheet1 := f.GetSheetName(0)
	r, err := f.workSheetReader(sheet1)
xurime's avatar
xurime 已提交
418
	assert.NoError(t, err)
419
	assert.NoError(t, f.InsertRows(sheet1, 1, 1))
420
	assert.Len(t, r.SheetData.Row, 0)
421
	assert.NoError(t, f.InsertRows(sheet1, 2, 1))
422
	assert.Len(t, r.SheetData.Row, 0)
423
	assert.NoError(t, f.InsertRows(sheet1, 99, 1))
424
	assert.Len(t, r.SheetData.Row, 0)
425
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestInsertRowInEmptyFile.xlsx")))
426 427
}

428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
func prepareTestBook2() (*File, error) {
	f := NewFile()
	for cell, val := range map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	} {
		if err := f.SetCellStr("Sheet1", cell, val); err != nil {
			return f, err
		}
	}
	return f, nil
}

xurime's avatar
xurime 已提交
445
func TestDuplicateRowFromSingleRow(t *testing.T) {
446 447 448 449 450 451 452 453 454 455 456 457 458
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")

	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}

	t.Run("FromSingleRow", func(t *testing.T) {
459 460 461
		f := NewFile()
		assert.NoError(t, f.SetCellStr(sheet, "A1", cells["A1"]))
		assert.NoError(t, f.SetCellStr(sheet, "B1", cells["B1"]))
462

463
		assert.NoError(t, f.DuplicateRow(sheet, 1))
464
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "FromSingleRow_1"))) {
465 466 467 468 469 470 471
			t.FailNow()
		}
		expect := map[string]string{
			"A1": cells["A1"], "B1": cells["B1"],
			"A2": cells["A1"], "B2": cells["B1"],
		}
		for cell, val := range expect {
472
			v, err := f.GetCellValue(sheet, cell)
473 474
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
475 476 477 478
				t.FailNow()
			}
		}

479
		assert.NoError(t, f.DuplicateRow(sheet, 2))
480
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "FromSingleRow_2"))) {
481 482 483 484 485 486 487 488
			t.FailNow()
		}
		expect = map[string]string{
			"A1": cells["A1"], "B1": cells["B1"],
			"A2": cells["A1"], "B2": cells["B1"],
			"A3": cells["A1"], "B3": cells["B1"],
		}
		for cell, val := range expect {
489
			v, err := f.GetCellValue(sheet, cell)
490 491
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
492 493 494 495
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
496 497 498 499 500 501 502 503 504 505 506 507 508 509
}

func TestDuplicateRowUpdateDuplicatedRows(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")

	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}
510 511

	t.Run("UpdateDuplicatedRows", func(t *testing.T) {
512 513 514
		f := NewFile()
		assert.NoError(t, f.SetCellStr(sheet, "A1", cells["A1"]))
		assert.NoError(t, f.SetCellStr(sheet, "B1", cells["B1"]))
515

516
		assert.NoError(t, f.DuplicateRow(sheet, 1))
517

518 519
		assert.NoError(t, f.SetCellStr(sheet, "A2", cells["A2"]))
		assert.NoError(t, f.SetCellStr(sheet, "B2", cells["B2"]))
520

521
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "UpdateDuplicatedRows"))) {
522 523 524 525 526 527 528
			t.FailNow()
		}
		expect := map[string]string{
			"A1": cells["A1"], "B1": cells["B1"],
			"A2": cells["A2"], "B2": cells["B2"],
		}
		for cell, val := range expect {
529
			v, err := f.GetCellValue(sheet, cell)
530 531
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
532 533 534 535
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
536 537 538 539 540 541 542 543 544 545 546 547 548
}

func TestDuplicateRowFirstOfMultipleRows(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}
549
	t.Run("FirstOfMultipleRows", func(t *testing.T) {
550 551
		f, err := prepareTestBook2()
		assert.NoError(t, err)
552
		assert.NoError(t, f.DuplicateRow(sheet, 1))
553

554
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "FirstOfMultipleRows"))) {
555 556 557 558 559 560 561 562 563
			t.FailNow()
		}
		expect := map[string]string{
			"A1": cells["A1"], "B1": cells["B1"],
			"A2": cells["A1"], "B2": cells["B1"],
			"A3": cells["A2"], "B3": cells["B2"],
			"A4": cells["A3"], "B4": cells["B3"],
		}
		for cell, val := range expect {
564
			v, err := f.GetCellValue(sheet, cell)
565 566
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
567 568 569 570
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
571 572 573 574 575
}

func TestDuplicateRowZeroWithNoRows(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
576 577

	t.Run("ZeroWithNoRows", func(t *testing.T) {
578
		f := NewFile()
579

580
		assert.EqualError(t, f.DuplicateRow(sheet, 0), newInvalidRowNumberError(0).Error())
581

582
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "ZeroWithNoRows"))) {
583 584
			t.FailNow()
		}
585

586
		val, err := f.GetCellValue(sheet, "A1")
587 588
		assert.NoError(t, err)
		assert.Equal(t, "", val)
589
		val, err = f.GetCellValue(sheet, "B1")
590 591
		assert.NoError(t, err)
		assert.Equal(t, "", val)
592
		val, err = f.GetCellValue(sheet, "A2")
593 594
		assert.NoError(t, err)
		assert.Equal(t, "", val)
595
		val, err = f.GetCellValue(sheet, "B2")
596 597 598 599
		assert.NoError(t, err)
		assert.Equal(t, "", val)

		assert.NoError(t, err)
600 601 602 603
		expect := map[string]string{
			"A1": "", "B1": "",
			"A2": "", "B2": "",
		}
604

605
		for cell, val := range expect {
606
			v, err := f.GetCellValue(sheet, cell)
607 608
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
609 610 611 612
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
613 614 615 616 617
}

func TestDuplicateRowMiddleRowOfEmptyFile(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
618 619

	t.Run("MiddleRowOfEmptyFile", func(t *testing.T) {
620
		f := NewFile()
621

622
		assert.NoError(t, f.DuplicateRow(sheet, 99))
623

624
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "MiddleRowOfEmptyFile"))) {
625 626 627 628 629 630 631 632
			t.FailNow()
		}
		expect := map[string]string{
			"A98":  "",
			"A99":  "",
			"A100": "",
		}
		for cell, val := range expect {
633
			v, err := f.GetCellValue(sheet, cell)
634 635
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
636 637 638 639
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
640 641 642 643 644 645 646 647 648 649 650 651 652 653
}

func TestDuplicateRowWithLargeOffsetToMiddleOfData(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")

	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}
654
	t.Run("WithLargeOffsetToMiddleOfData", func(t *testing.T) {
655 656
		f, err := prepareTestBook2()
		assert.NoError(t, err)
657
		assert.NoError(t, f.DuplicateRowTo(sheet, 1, 3))
658

659
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "WithLargeOffsetToMiddleOfData"))) {
660 661 662 663 664 665 666 667 668
			t.FailNow()
		}
		expect := map[string]string{
			"A1": cells["A1"], "B1": cells["B1"],
			"A2": cells["A2"], "B2": cells["B2"],
			"A3": cells["A1"], "B3": cells["B1"],
			"A4": cells["A3"], "B4": cells["B3"],
		}
		for cell, val := range expect {
669
			v, err := f.GetCellValue(sheet, cell)
670 671
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
672 673 674 675
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688
}

func TestDuplicateRowWithLargeOffsetToEmptyRows(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}
689
	t.Run("WithLargeOffsetToEmptyRows", func(t *testing.T) {
690 691
		f, err := prepareTestBook2()
		assert.NoError(t, err)
692
		assert.NoError(t, f.DuplicateRowTo(sheet, 1, 7))
693

694
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "WithLargeOffsetToEmptyRows"))) {
695 696 697 698 699 700 701 702 703
			t.FailNow()
		}
		expect := map[string]string{
			"A1": cells["A1"], "B1": cells["B1"],
			"A2": cells["A2"], "B2": cells["B2"],
			"A3": cells["A3"], "B3": cells["B3"],
			"A7": cells["A1"], "B7": cells["B1"],
		}
		for cell, val := range expect {
704
			v, err := f.GetCellValue(sheet, cell)
705 706
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
707 708 709 710
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
711 712 713 714 715 716 717 718 719 720 721 722 723
}

func TestDuplicateRowInsertBefore(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}
724
	t.Run("InsertBefore", func(t *testing.T) {
725 726
		f, err := prepareTestBook2()
		assert.NoError(t, err)
727
		assert.NoError(t, f.DuplicateRowTo(sheet, 2, 1))
728
		assert.NoError(t, f.DuplicateRowTo(sheet, 10, 4))
729

730
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "InsertBefore"))) {
731 732 733 734 735 736 737
			t.FailNow()
		}

		expect := map[string]string{
			"A1": cells["A2"], "B1": cells["B2"],
			"A2": cells["A1"], "B2": cells["B1"],
			"A3": cells["A2"], "B3": cells["B2"],
738
			"A5": cells["A3"], "B5": cells["B3"],
739 740
		}
		for cell, val := range expect {
741
			v, err := f.GetCellValue(sheet, cell)
742 743
			assert.NoError(t, err)
			if !assert.Equal(t, val, v, cell) {
744 745 746 747
				t.FailNow()
			}
		}
	})
xurime's avatar
xurime 已提交
748 749 750 751 752 753 754 755 756 757 758 759 760
}

func TestDuplicateRowInsertBeforeWithLargeOffset(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}
761
	t.Run("InsertBeforeWithLargeOffset", func(t *testing.T) {
762 763
		f, err := prepareTestBook2()
		assert.NoError(t, err)
764
		assert.NoError(t, f.DuplicateRowTo(sheet, 3, 1))
765

766
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "InsertBeforeWithLargeOffset"))) {
767 768 769 770 771 772 773 774 775 776
			t.FailNow()
		}

		expect := map[string]string{
			"A1": cells["A3"], "B1": cells["B3"],
			"A2": cells["A1"], "B2": cells["B1"],
			"A3": cells["A2"], "B3": cells["B2"],
			"A4": cells["A3"], "B4": cells["B3"],
		}
		for cell, val := range expect {
777
			v, err := f.GetCellValue(sheet, cell)
778 779
			assert.NoError(t, err)
			if !assert.Equal(t, val, v) {
780 781 782 783 784 785
				t.FailNow()
			}
		}
	})
}

786 787 788
func TestDuplicateRowInsertBeforeWithMergeCells(t *testing.T) {
	const sheet = "Sheet1"
	outFile := filepath.Join("test", "TestDuplicateRow.%s.xlsx")
789 790 791
	t.Run("InsertBeforeWithLargeOffset", func(t *testing.T) {
		f, err := prepareTestBook2()
		assert.NoError(t, err)
792 793 794
		assert.NoError(t, f.MergeCell(sheet, "B2", "C2"))
		assert.NoError(t, f.MergeCell(sheet, "C6", "C8"))

795 796
		assert.NoError(t, f.DuplicateRowTo(sheet, 2, 1))
		assert.NoError(t, f.DuplicateRowTo(sheet, 1, 8))
797

798
		if !assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, "InsertBeforeWithMergeCells"))) {
799 800 801 802 803 804 805 806 807
			t.FailNow()
		}

		expect := []MergeCell{
			{"B3:C3", "B2 Value"},
			{"C7:C10", ""},
			{"B1:C1", "B2 Value"},
		}

808
		mergeCells, err := f.GetMergeCells(sheet)
809 810 811 812 813 814 815 816 817
		assert.NoError(t, err)
		for idx, val := range expect {
			if !assert.Equal(t, val, mergeCells[idx]) {
				t.FailNow()
			}
		}
	})
}

818
func TestDuplicateRowInvalidRowNum(t *testing.T) {
819
	const sheet = "Sheet1"
820
	outFile := filepath.Join("test", "TestDuplicateRow.InvalidRowNum.%s.xlsx")
821 822 823 824 825 826 827 828 829 830

	cells := map[string]string{
		"A1": "A1 Value",
		"A2": "A2 Value",
		"A3": "A3 Value",
		"B1": "B1 Value",
		"B2": "B2 Value",
		"B3": "B3 Value",
	}

831
	invalidIndexes := []int{-100, -2, -1, 0}
832

833 834
	for _, row := range invalidIndexes {
		name := fmt.Sprintf("%d", row)
835
		t.Run(name, func(t *testing.T) {
836
			f := NewFile()
837
			for col, val := range cells {
838
				assert.NoError(t, f.SetCellStr(sheet, col, val))
839
			}
840

841
			assert.EqualError(t, f.DuplicateRow(sheet, row), newInvalidRowNumberError(row).Error())
842 843

			for col, val := range cells {
844
				v, err := f.GetCellValue(sheet, col)
845 846
				assert.NoError(t, err)
				if !assert.Equal(t, val, v) {
847 848 849
					t.FailNow()
				}
			}
850
			assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, name)))
851 852 853
		})
	}

854 855 856 857
	for _, row1 := range invalidIndexes {
		for _, row2 := range invalidIndexes {
			name := fmt.Sprintf("[%d,%d]", row1, row2)
			t.Run(name, func(t *testing.T) {
858
				f := NewFile()
859
				for col, val := range cells {
860
					assert.NoError(t, f.SetCellStr(sheet, col, val))
861
				}
862

863
				assert.EqualError(t, f.DuplicateRowTo(sheet, row1, row2), newInvalidRowNumberError(row1).Error())
864 865

				for col, val := range cells {
866
					v, err := f.GetCellValue(sheet, col)
867 868
					assert.NoError(t, err)
					if !assert.Equal(t, val, v) {
869 870
						t.FailNow()
					}
871
				}
872
				assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, name)))
873 874
			})
		}
875 876 877
	}
}

878 879 880 881
func TestDuplicateRow(t *testing.T) {
	f := NewFile()
	// Test duplicate row with invalid sheet name
	assert.EqualError(t, f.DuplicateRowTo("Sheet:1", 1, 2), ErrSheetNameInvalid.Error())
R
rjtee 已提交
882 883 884 885 886 887 888 889

	f = NewFile()
	assert.NoError(t, f.SetDefinedName(&DefinedName{
		Name:     "Amount",
		RefersTo: "Sheet1!$B$1",
	}))
	assert.NoError(t, f.SetCellFormula("Sheet1", "A1", "Amount+C1"))
	assert.NoError(t, f.SetCellValue("Sheet1", "A10", "A10"))
890 891 892 893 894

	format, err := f.NewConditionalStyle(&Style{Font: &Font{Color: "9A0511"}, Fill: Fill{Type: "pattern", Color: []string{"FEC7CE"}, Pattern: 1}})
	assert.NoError(t, err)

	expected := []ConditionalFormatOptions{
895
		{Type: "cell", Criteria: "greater than", Format: &format, Value: "0"},
896 897 898 899 900 901 902 903 904 905 906
	}
	assert.NoError(t, f.SetConditionalFormat("Sheet1", "A1", expected))

	dv := NewDataValidation(true)
	dv.Sqref = "A1"
	assert.NoError(t, dv.SetDropList([]string{"1", "2", "3"}))
	assert.NoError(t, f.AddDataValidation("Sheet1", dv))
	ws, ok := f.Sheet.Load("xl/worksheets/sheet1.xml")
	assert.True(t, ok)
	ws.(*xlsxWorksheet).DataValidations.DataValidation[0].Sqref = "A1"

R
rjtee 已提交
907 908 909 910 911 912 913
	assert.NoError(t, f.DuplicateRowTo("Sheet1", 1, 10))
	formula, err := f.GetCellFormula("Sheet1", "A10")
	assert.NoError(t, err)
	assert.Equal(t, "Amount+C10", formula)
	value, err := f.GetCellValue("Sheet1", "A11")
	assert.NoError(t, err)
	assert.Equal(t, "A10", value)
914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935

	cfs, err := f.GetConditionalFormats("Sheet1")
	assert.NoError(t, err)
	assert.Len(t, cfs, 2)
	assert.Equal(t, expected, cfs["A10:A10"])

	dvs, err := f.GetDataValidations("Sheet1")
	assert.NoError(t, err)
	assert.Len(t, dvs, 2)
	assert.Equal(t, "A10:A10", dvs[1].Sqref)

	// Test duplicate data validation with row number exceeds maximum limit
	assert.Equal(t, ErrMaxRows, f.duplicateDataValidations(ws.(*xlsxWorksheet), "Sheet1", 1, TotalRows+1))
	// Test duplicate data validation with invalid range reference
	ws.(*xlsxWorksheet).DataValidations.DataValidation[0].Sqref = "A"
	assert.Equal(t, newCellNameToCoordinatesError("A", newInvalidCellNameError("A")), f.duplicateDataValidations(ws.(*xlsxWorksheet), "Sheet1", 1, 10))

	// Test duplicate conditional formatting with row number exceeds maximum limit
	assert.Equal(t, ErrMaxRows, f.duplicateConditionalFormat(ws.(*xlsxWorksheet), "Sheet1", 1, TotalRows+1))
	// Test duplicate conditional formatting with invalid range reference
	ws.(*xlsxWorksheet).ConditionalFormatting[0].SQRef = "A"
	assert.Equal(t, newCellNameToCoordinatesError("A", newInvalidCellNameError("A")), f.duplicateConditionalFormat(ws.(*xlsxWorksheet), "Sheet1", 1, 10))
936 937
}

938
func TestDuplicateRowTo(t *testing.T) {
939 940 941 942 943 944 945
	f, sheetName := NewFile(), "Sheet1"
	// Test duplicate row with invalid target row number
	assert.Equal(t, nil, f.DuplicateRowTo(sheetName, 1, 0))
	// Test duplicate row with equal source and target row number
	assert.Equal(t, nil, f.DuplicateRowTo(sheetName, 1, 1))
	// Test duplicate row on the blank worksheet
	assert.Equal(t, nil, f.DuplicateRowTo(sheetName, 1, 2))
946
	// Test duplicate row on the worksheet with illegal cell reference
947
	f.Sheet.Store("xl/worksheets/sheet1.xml", &xlsxWorksheet{
948 949
		MergeCells: &xlsxMergeCells{Cells: []*xlsxMergeCell{{Ref: "A:B1"}}},
	})
950 951
	assert.EqualError(t, f.DuplicateRowTo(sheetName, 1, 2), newCellNameToCoordinatesError("A", newInvalidCellNameError("A")).Error())
	// Test duplicate row on not exists worksheet
952
	assert.EqualError(t, f.DuplicateRowTo("SheetN", 1, 2), "sheet SheetN does not exist")
953 954
	// Test duplicate row with invalid sheet name
	assert.EqualError(t, f.DuplicateRowTo("Sheet:1", 1, 2), ErrSheetNameInvalid.Error())
955 956 957 958
}

func TestDuplicateMergeCells(t *testing.T) {
	f := File{}
959
	ws := &xlsxWorksheet{MergeCells: &xlsxMergeCells{
960
		Cells: []*xlsxMergeCell{{Ref: "A1:-"}},
961
	}}
962
	assert.EqualError(t, f.duplicateMergeCells(ws, "Sheet1", 0, 0), `cannot convert cell "-" to coordinates: invalid cell name "-"`)
963
	ws.MergeCells.Cells[0].Ref = "A1:B1"
964
	assert.EqualError(t, f.duplicateMergeCells(ws, "SheetN", 1, 2), "sheet SheetN does not exist")
965 966
}

967
func TestGetValueFromInlineStr(t *testing.T) {
xurime's avatar
xurime 已提交
968 969 970
	c := &xlsxC{T: "inlineStr"}
	f := NewFile()
	d := &xlsxSST{}
xurime's avatar
xurime 已提交
971
	val, err := c.getValueFrom(f, d, false)
xurime's avatar
xurime 已提交
972 973 974 975
	assert.NoError(t, err)
	assert.Equal(t, "", val)
}

976
func TestGetValueFromNumber(t *testing.T) {
977
	c := &xlsxC{T: "n"}
978 979
	f := NewFile()
	d := &xlsxSST{}
980 981 982 983 984 985 986 987 988 989 990
	for input, expected := range map[string]string{
		"2.2.":                     "2.2.",
		"1.1000000000000001":       "1.1",
		"2.2200000000000002":       "2.22",
		"28.552":                   "28.552",
		"27.399000000000001":       "27.399",
		"26.245999999999999":       "26.246",
		"2422.3000000000002":       "2422.3",
		"2.220000ddsf0000000002-r": "2.220000ddsf0000000002-r",
	} {
		c.V = input
xurime's avatar
xurime 已提交
991
		val, err := c.getValueFrom(f, d, false)
992 993 994
		assert.NoError(t, err)
		assert.Equal(t, expected, val)
	}
995 996
}

997
func TestErrSheetNotExistError(t *testing.T) {
998
	assert.Equal(t, "sheet Sheet1 does not exist", ErrSheetNotExist{"Sheet1"}.Error())
999 1000
}

1001 1002
func TestCheckRow(t *testing.T) {
	f := NewFile()
1003
	f.Pkg.Store("xl/worksheets/sheet1.xml", []byte(xml.Header+`<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" ><sheetData><row r="2"><c><v>1</v></c><c r="F2"><v>2</v></c><c><v>3</v></c><c><v>4</v></c><c r="M2"><v>5</v></c></row></sheetData></worksheet>`))
1004 1005 1006 1007
	_, err := f.GetRows("Sheet1")
	assert.NoError(t, err)
	assert.NoError(t, f.SetCellValue("Sheet1", "A1", false))
	f = NewFile()
1008
	f.Pkg.Store("xl/worksheets/sheet1.xml", []byte(xml.Header+`<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" ><sheetData><row r="2"><c><v>1</v></c><c r="-"><v>2</v></c><c><v>3</v></c><c><v>4</v></c><c r="M2"><v>5</v></c></row></sheetData></worksheet>`))
1009
	f.Sheet.Delete("xl/worksheets/sheet1.xml")
xurime's avatar
xurime 已提交
1010
	f.checked.Delete("xl/worksheets/sheet1.xml")
1011
	assert.EqualError(t, f.SetCellValue("Sheet1", "A1", false), newCellNameToCoordinatesError("-", newInvalidCellNameError("-")).Error())
1012 1013
}

xurime's avatar
xurime 已提交
1014 1015
func TestSetRowStyle(t *testing.T) {
	f := NewFile()
1016
	style1, err := f.NewStyle(&Style{Fill: Fill{Type: "pattern", Color: []string{"63BE7B"}, Pattern: 1}})
xurime's avatar
xurime 已提交
1017
	assert.NoError(t, err)
1018
	style2, err := f.NewStyle(&Style{Fill: Fill{Type: "pattern", Color: []string{"E0EBF5"}, Pattern: 1}})
1019 1020 1021 1022
	assert.NoError(t, err)
	assert.NoError(t, f.SetCellStyle("Sheet1", "B2", "B2", style1))
	assert.EqualError(t, f.SetRowStyle("Sheet1", 5, -1, style2), newInvalidRowNumberError(-1).Error())
	assert.EqualError(t, f.SetRowStyle("Sheet1", 1, TotalRows+1, style2), ErrMaxRows.Error())
1023
	// Test set row style with invalid style ID
xurime's avatar
xurime 已提交
1024
	assert.EqualError(t, f.SetRowStyle("Sheet1", 1, 1, -1), newInvalidStyleID(-1).Error())
1025
	// Test set row style with not exists style ID
1026
	assert.EqualError(t, f.SetRowStyle("Sheet1", 1, 1, 10), newInvalidStyleID(10).Error())
1027
	assert.EqualError(t, f.SetRowStyle("SheetN", 1, 1, style2), "sheet SheetN does not exist")
1028 1029
	// Test set row style with invalid sheet name
	assert.EqualError(t, f.SetRowStyle("Sheet:1", 1, 1, 0), ErrSheetNameInvalid.Error())
1030
	assert.NoError(t, f.SetRowStyle("Sheet1", 5, 1, style2))
1031 1032
	cellStyleID, err := f.GetCellStyle("Sheet1", "B2")
	assert.NoError(t, err)
1033
	assert.Equal(t, style2, cellStyleID)
1034
	// Test cell inheritance rows style
1035 1036 1037 1038
	assert.NoError(t, f.SetCellValue("Sheet1", "C1", nil))
	cellStyleID, err = f.GetCellStyle("Sheet1", "C1")
	assert.NoError(t, err)
	assert.Equal(t, style2, cellStyleID)
xurime's avatar
xurime 已提交
1039
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetRowStyle.xlsx")))
1040
	// Test set row style with unsupported charset style sheet
1041 1042 1043
	f.Styles = nil
	f.Pkg.Store(defaultXMLPathStyles, MacintoshCyrillicCharset)
	assert.EqualError(t, f.SetRowStyle("Sheet1", 1, 1, cellStyleID), "XML syntax error on line 1: invalid UTF-8")
xurime's avatar
xurime 已提交
1044 1045
}

1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061
func TestSetRowHeight(t *testing.T) {
	f := NewFile()
	// Test hidden row by set row height to 0
	assert.NoError(t, f.SetRowHeight("Sheet1", 2, 0))
	ht, err := f.GetRowHeight("Sheet1", 2)
	assert.NoError(t, err)
	assert.Empty(t, ht)
	// Test unset custom row height
	assert.NoError(t, f.SetRowHeight("Sheet1", 2, -1))
	ht, err = f.GetRowHeight("Sheet1", 2)
	assert.NoError(t, err)
	assert.Equal(t, defaultRowHeight, ht)
	// Test set row height with invalid height value
	assert.Equal(t, ErrParameterInvalid, f.SetRowHeight("Sheet1", 2, -2))
}

1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080
func TestNumberFormats(t *testing.T) {
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	cells := make([][]string, 0)
	cols, err := f.Cols("Sheet2")
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	for cols.Next() {
		col, err := cols.Rows()
		assert.NoError(t, err)
		if err != nil {
			break
		}
		cells = append(cells, col)
	}
	assert.Equal(t, []string{"", "200", "450", "200", "510", "315", "127", "89", "348", "53", "37"}, cells[3])
1081
	assert.NoError(t, f.Close())
1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093

	f = NewFile()
	numFmt1, err := f.NewStyle(&Style{NumFmt: 1})
	assert.NoError(t, err)
	numFmt2, err := f.NewStyle(&Style{NumFmt: 2})
	assert.NoError(t, err)
	numFmt3, err := f.NewStyle(&Style{NumFmt: 3})
	assert.NoError(t, err)
	numFmt9, err := f.NewStyle(&Style{NumFmt: 9})
	assert.NoError(t, err)
	numFmt10, err := f.NewStyle(&Style{NumFmt: 10})
	assert.NoError(t, err)
1094 1095
	numFmt21, err := f.NewStyle(&Style{NumFmt: 21})
	assert.NoError(t, err)
1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136
	numFmt37, err := f.NewStyle(&Style{NumFmt: 37})
	assert.NoError(t, err)
	numFmt38, err := f.NewStyle(&Style{NumFmt: 38})
	assert.NoError(t, err)
	numFmt39, err := f.NewStyle(&Style{NumFmt: 39})
	assert.NoError(t, err)
	numFmt40, err := f.NewStyle(&Style{NumFmt: 40})
	assert.NoError(t, err)
	for _, cases := range [][]interface{}{
		{"A1", numFmt1, 8.8888666665555493e+19, "88888666665555500000"},
		{"A2", numFmt1, 8.8888666665555487, "9"},
		{"A3", numFmt2, 8.8888666665555493e+19, "88888666665555500000.00"},
		{"A4", numFmt2, 8.8888666665555487, "8.89"},
		{"A5", numFmt3, 8.8888666665555493e+19, "88,888,666,665,555,500,000"},
		{"A6", numFmt3, 8.8888666665555487, "9"},
		{"A7", numFmt3, 123, "123"},
		{"A8", numFmt3, -1234, "-1,234"},
		{"A9", numFmt9, 8.8888666665555493e+19, "8888866666555550000000%"},
		{"A10", numFmt9, -8.8888666665555493e+19, "-8888866666555550000000%"},
		{"A11", numFmt9, 8.8888666665555487, "889%"},
		{"A12", numFmt9, -8.8888666665555487, "-889%"},
		{"A13", numFmt10, 8.8888666665555493e+19, "8888866666555550000000.00%"},
		{"A14", numFmt10, -8.8888666665555493e+19, "-8888866666555550000000.00%"},
		{"A15", numFmt10, 8.8888666665555487, "888.89%"},
		{"A16", numFmt10, -8.8888666665555487, "-888.89%"},
		{"A17", numFmt37, 8.8888666665555493e+19, "88,888,666,665,555,500,000 "},
		{"A18", numFmt37, -8.8888666665555493e+19, "(88,888,666,665,555,500,000)"},
		{"A19", numFmt37, 8.8888666665555487, "9 "},
		{"A20", numFmt37, -8.8888666665555487, "(9)"},
		{"A21", numFmt38, 8.8888666665555493e+19, "88,888,666,665,555,500,000 "},
		{"A22", numFmt38, -8.8888666665555493e+19, "(88,888,666,665,555,500,000)"},
		{"A23", numFmt38, 8.8888666665555487, "9 "},
		{"A24", numFmt38, -8.8888666665555487, "(9)"},
		{"A25", numFmt39, 8.8888666665555493e+19, "88,888,666,665,555,500,000.00 "},
		{"A26", numFmt39, -8.8888666665555493e+19, "(88,888,666,665,555,500,000.00)"},
		{"A27", numFmt39, 8.8888666665555487, "8.89 "},
		{"A28", numFmt39, -8.8888666665555487, "(8.89)"},
		{"A29", numFmt40, 8.8888666665555493e+19, "88,888,666,665,555,500,000.00 "},
		{"A30", numFmt40, -8.8888666665555493e+19, "(88,888,666,665,555,500,000.00)"},
		{"A31", numFmt40, 8.8888666665555487, "8.89 "},
		{"A32", numFmt40, -8.8888666665555487, "(8.89)"},
1137 1138 1139
		{"A33", numFmt21, 44729.999988368058, "23:59:59"},
		{"A34", numFmt21, 44944.375005787035, "09:00:00"},
		{"A35", numFmt21, 44944.375005798611, "09:00:01"},
1140 1141
	} {
		cell, styleID, value, expected := cases[0].(string), cases[1].(int), cases[2], cases[3].(string)
1142
		assert.NoError(t, f.SetCellStyle("Sheet1", cell, cell, styleID))
1143 1144 1145
		assert.NoError(t, f.SetCellValue("Sheet1", cell, value))
		result, err := f.GetCellValue("Sheet1", cell)
		assert.NoError(t, err)
1146
		assert.Equal(t, expected, result, cell)
1147 1148
	}
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestNumberFormats.xlsx")))
1149

1150
	f = NewFile(Options{ShortDatePattern: "yyyy/m/d"})
1151 1152 1153 1154 1155 1156 1157
	assert.NoError(t, f.SetCellValue("Sheet1", "A1", 43543.503206018519))
	numFmt14, err := f.NewStyle(&Style{NumFmt: 14})
	assert.NoError(t, err)
	assert.NoError(t, f.SetCellStyle("Sheet1", "A1", "A1", numFmt14))
	result, err := f.GetCellValue("Sheet1", "A1")
	assert.NoError(t, err)
	assert.Equal(t, "2019/3/19", result, "A1")
1158 1159
}

H
Harris 已提交
1160
func BenchmarkRows(b *testing.B) {
1161
	f, _ := OpenFile(filepath.Join("test", "Book1.xlsx"))
H
Harris 已提交
1162 1163 1164 1165 1166 1167 1168 1169 1170 1171
	for i := 0; i < b.N; i++ {
		rows, _ := f.Rows("Sheet2")
		for rows.Next() {
			row, _ := rows.Columns()
			for i := range row {
				if i >= 0 {
					continue
				}
			}
		}
1172 1173 1174 1175 1176 1177
		if err := rows.Close(); err != nil {
			b.Error(err)
		}
	}
	if err := f.Close(); err != nil {
		b.Error(err)
H
Harris 已提交
1178 1179 1180
	}
}

1181
// trimSliceSpace trim continually blank element in the tail of slice.
1182 1183 1184 1185 1186 1187 1188 1189 1190 1191
func trimSliceSpace(s []string) []string {
	for {
		if len(s) > 0 && s[len(s)-1] == "" {
			s = s[:len(s)-1]
		} else {
			break
		}
	}
	return s
}