excelize_test.go 47.5 KB
Newer Older
xurime's avatar
xurime 已提交
1 2 3
package excelize

import (
4
	"bytes"
xurime's avatar
xurime 已提交
5 6
	"compress/gzip"
	"encoding/xml"
7
	"fmt"
xurime's avatar
xurime 已提交
8
	"image/color"
9 10 11
	_ "image/gif"
	_ "image/jpeg"
	_ "image/png"
12
	"io/ioutil"
13
	"math"
V
Veniamin Albaev 已提交
14
	"os"
15
	"path/filepath"
xurime's avatar
xurime 已提交
16
	"strconv"
xurime's avatar
xurime 已提交
17
	"strings"
xurime's avatar
xurime 已提交
18
	"testing"
19
	"time"
V
Veniamin Albaev 已提交
20 21

	"github.com/stretchr/testify/assert"
xurime's avatar
xurime 已提交
22 23
)

24
func TestOpenFile(t *testing.T) {
xurime's avatar
xurime 已提交
25
	// Test update a XLSX file.
26
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
27 28
	if !assert.NoError(t, err) {
		t.FailNow()
29
	}
V
Veniamin Albaev 已提交
30

31
	// Test get all the rows in a not exists worksheet.
32
	f.GetRows("Sheet4")
33
	// Test get all the rows in a worksheet.
34
	rows, err := f.GetRows("Sheet2")
35
	assert.NoError(t, err)
36 37 38 39 40 41
	for _, row := range rows {
		for _, cell := range row {
			t.Log(cell, "\t")
		}
		t.Log("\r\n")
	}
42
	f.UpdateLinkedValue()
43

44 45
	f.SetCellDefault("Sheet2", "A1", strconv.FormatFloat(float64(100.1588), 'f', -1, 32))
	f.SetCellDefault("Sheet2", "A1", strconv.FormatFloat(float64(-100.1588), 'f', -1, 64))
46

47
	// Test set cell value with illegal row number.
48
	assert.EqualError(t, f.SetCellDefault("Sheet2", "A", strconv.FormatFloat(float64(-100.1588), 'f', -1, 64)),
49
		`cannot convert cell "A" to coordinates: invalid cell name "A"`)
50

51
	assert.NoError(t, f.SetCellInt("Sheet2", "A1", 100))
52

53
	// Test set cell integer value with illegal row number.
54
	assert.EqualError(t, f.SetCellInt("Sheet2", "A", 100), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
55

56
	f.SetCellStr("Sheet2", "C11", "Knowns")
57
	// Test max characters in a cell.
58 59
	f.SetCellStr("Sheet2", "D11", strings.Repeat("c", 32769))
	f.NewSheet(":\\/?*[]Maximum 31 characters allowed in sheet title.")
60
	// Test set worksheet name with illegal name.
61 62 63 64
	f.SetSheetName("Maximum 31 characters allowed i", "[Rename]:\\/?* Maximum 31 characters allowed in sheet title.")
	f.SetCellInt("Sheet3", "A23", 10)
	f.SetCellStr("Sheet3", "b230", "10")
	assert.EqualError(t, f.SetCellStr("Sheet10", "b230", "10"), "sheet Sheet10 is not exist")
65

66
	// Test set cell string value with illegal row number.
67
	assert.EqualError(t, f.SetCellStr("Sheet1", "A", "10"), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
68

69
	f.SetActiveSheet(2)
70
	// Test get cell formula with given rows number.
71
	_, err = f.GetCellFormula("Sheet1", "B19")
72
	assert.NoError(t, err)
73
	// Test get cell formula with illegal worksheet name.
74
	_, err = f.GetCellFormula("Sheet2", "B20")
75
	assert.NoError(t, err)
76
	_, err = f.GetCellFormula("Sheet1", "B20")
77
	assert.NoError(t, err)
78 79

	// Test get cell formula with illegal rows number.
80
	_, err = f.GetCellFormula("Sheet1", "B")
81
	assert.EqualError(t, err, `cannot convert cell "B" to coordinates: invalid cell name "B"`)
82
	// Test get shared cell formula
83 84 85 86
	_, err = f.GetCellFormula("Sheet2", "H11")
	assert.NoError(t, err)
	_, err = f.GetCellFormula("Sheet2", "I11")
	assert.NoError(t, err)
87
	getSharedForumula(&xlsxWorksheet{}, "")
88

xurime's avatar
xurime 已提交
89
	// Test read cell value with given illegal rows number.
90
	_, err = f.GetCellValue("Sheet2", "a-1")
91
	assert.EqualError(t, err, `cannot convert cell "A-1" to coordinates: invalid cell name "A-1"`)
92
	_, err = f.GetCellValue("Sheet2", "A")
93
	assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
94

xurime's avatar
xurime 已提交
95
	// Test read cell value with given lowercase column number.
96 97 98 99 100 101 102 103
	_, err = f.GetCellValue("Sheet2", "a5")
	assert.NoError(t, err)
	_, err = f.GetCellValue("Sheet2", "C11")
	assert.NoError(t, err)
	_, err = f.GetCellValue("Sheet2", "D11")
	assert.NoError(t, err)
	_, err = f.GetCellValue("Sheet2", "D12")
	assert.NoError(t, err)
xurime's avatar
xurime 已提交
104
	// Test SetCellValue function.
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
	assert.NoError(t, f.SetCellValue("Sheet2", "F1", " Hello"))
	assert.NoError(t, f.SetCellValue("Sheet2", "G1", []byte("World")))
	assert.NoError(t, f.SetCellValue("Sheet2", "F2", 42))
	assert.NoError(t, f.SetCellValue("Sheet2", "F3", int8(1<<8/2-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F4", int16(1<<16/2-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F5", int32(1<<32/2-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F6", int64(1<<32/2-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F7", float32(42.65418)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F8", float64(-42.65418)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F9", float32(42)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F10", float64(42)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F11", uint(1<<32-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F12", uint8(1<<8-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F13", uint16(1<<16-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F14", uint32(1<<32-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F15", uint64(1<<32-1)))
	assert.NoError(t, f.SetCellValue("Sheet2", "F16", true))
	assert.NoError(t, f.SetCellValue("Sheet2", "F17", complex64(5+10i)))
xurime's avatar
xurime 已提交
123 124

	// Test on not exists worksheet.
125 126 127 128 129
	assert.EqualError(t, f.SetCellDefault("SheetN", "A1", ""), "sheet SheetN is not exist")
	assert.EqualError(t, f.SetCellFloat("SheetN", "A1", 42.65418, 2, 32), "sheet SheetN is not exist")
	assert.EqualError(t, f.SetCellBool("SheetN", "A1", true), "sheet SheetN is not exist")
	assert.EqualError(t, f.SetCellFormula("SheetN", "A1", ""), "sheet SheetN is not exist")
	assert.EqualError(t, f.SetCellHyperLink("SheetN", "A1", "Sheet1!A40", "Location"), "sheet SheetN is not exist")
130

131 132 133 134 135 136 137 138 139
	// Test boolean write
	booltest := []struct {
		value    bool
		expected string
	}{
		{false, "0"},
		{true, "1"},
	}
	for _, test := range booltest {
140 141
		f.SetCellValue("Sheet2", "F16", test.value)
		val, err := f.GetCellValue("Sheet2", "F16")
142 143
		assert.NoError(t, err)
		assert.Equal(t, test.expected, val)
144
	}
145

146
	f.SetCellValue("Sheet2", "G2", nil)
147

148
	assert.EqualError(t, f.SetCellValue("Sheet2", "G4", time.Now()), "only UTC time expected")
149

150
	f.SetCellValue("Sheet2", "G4", time.Now().UTC())
151
	// 02:46:40
152
	f.SetCellValue("Sheet2", "G5", time.Duration(1e13))
xurime's avatar
xurime 已提交
153
	// Test completion column.
154
	f.SetCellValue("Sheet2", "M2", nil)
xurime's avatar
xurime 已提交
155
	// Test read cell value with given axis large than exists row.
156 157
	_, err = f.GetCellValue("Sheet2", "E231")
	assert.NoError(t, err)
158
	// Test get active worksheet of XLSX and get worksheet name of XLSX by given worksheet index.
159
	f.GetSheetName(f.GetActiveSheetIndex())
160
	// Test get worksheet index of XLSX by given worksheet name.
161
	f.GetSheetIndex("Sheet1")
162
	// Test get worksheet name of XLSX by given invalid worksheet index.
163 164 165
	f.GetSheetName(4)
	// Test get worksheet map of f.
	f.GetSheetMap()
166
	for i := 1; i <= 300; i++ {
167
		f.SetCellStr("Sheet3", "c"+strconv.Itoa(i), strconv.Itoa(i))
168
	}
169
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestOpenFile.xlsx")))
V
Veniamin Albaev 已提交
170 171
}

xurime's avatar
xurime 已提交
172
func TestSaveFile(t *testing.T) {
173
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
xurime's avatar
xurime 已提交
174 175 176
	if !assert.NoError(t, err) {
		t.FailNow()
	}
177 178
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSaveFile.xlsx")))
	f, err = OpenFile(filepath.Join("test", "TestSaveFile.xlsx"))
xurime's avatar
xurime 已提交
179 180 181
	if !assert.NoError(t, err) {
		t.FailNow()
	}
182
	assert.NoError(t, f.Save())
xurime's avatar
xurime 已提交
183 184
}

V
Veniamin Albaev 已提交
185
func TestSaveAsWrongPath(t *testing.T) {
186
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
187 188
	if assert.NoError(t, err) {
		// Test write file to not exist directory.
189
		err = f.SaveAs("")
V
Veniamin Albaev 已提交
190 191 192
		if assert.Error(t, err) {
			assert.True(t, os.IsNotExist(err), "Error: %v: Expected os.IsNotExists(err) == true", err)
		}
xurime's avatar
xurime 已提交
193
	}
194 195
}

xurime's avatar
xurime 已提交
196 197 198 199 200
func TestCharsetTranscoder(t *testing.T) {
	f := NewFile()
	f.CharsetTranscoder(*new(charsetTranscoderFn))
}

201 202 203
func TestOpenReader(t *testing.T) {
	_, err := OpenReader(strings.NewReader(""))
	assert.EqualError(t, err, "zip: not a valid zip file")
204 205 206 207 208 209 210 211
	_, err = OpenReader(bytes.NewReader([]byte{
		0x3c, 0x00, 0x00, 0x00, 0x4d, 0x00, 0x69, 0x00, 0x63, 0x00, 0x72, 0x00, 0x6f, 0x00, 0x73, 0x00,
		0x6f, 0x00, 0x66, 0x00, 0x74, 0x00, 0x2e, 0x00, 0x43, 0x00, 0x6f, 0x00, 0x6e, 0x00, 0x74, 0x00,
		0x61, 0x00, 0x69, 0x00, 0x6e, 0x00, 0x65, 0x00, 0x72, 0x00, 0x2e, 0x00, 0x44, 0x00, 0x61, 0x00,
		0x74, 0x00, 0x61, 0x00, 0x53, 0x00, 0x70, 0x00, 0x61, 0x00, 0x63, 0x00, 0x65, 0x00, 0x73, 0x00,
		0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
	}))
	assert.EqualError(t, err, "not support encrypted file currently")
xurime's avatar
xurime 已提交
212 213 214 215 216 217 218 219 220 221 222 223

	// Test unexpected EOF.
	var b bytes.Buffer
	w := gzip.NewWriter(&b)
	defer w.Close()
	w.Flush()

	r, _ := gzip.NewReader(&b)
	defer r.Close()

	_, err = OpenReader(r)
	assert.EqualError(t, err, "unexpected EOF")
224 225
}

226
func TestBrokenFile(t *testing.T) {
xurime's avatar
xurime 已提交
227
	// Test write file with broken file struct.
228
	f := File{}
xurime's avatar
xurime 已提交
229

V
Veniamin Albaev 已提交
230
	t.Run("SaveWithoutName", func(t *testing.T) {
231
		assert.EqualError(t, f.Save(), "no path defined for file, consider File.WriteTo or File.Write")
V
Veniamin Albaev 已提交
232
	})
233

V
Veniamin Albaev 已提交
234 235
	t.Run("SaveAsEmptyStruct", func(t *testing.T) {
		// Test write file with broken file struct with given path.
236
		assert.NoError(t, f.SaveAs(filepath.Join("test", "BrokenFile.SaveAsEmptyStruct.xlsx")))
V
Veniamin Albaev 已提交
237 238 239 240
	})

	t.Run("OpenBadWorkbook", func(t *testing.T) {
		// Test set active sheet without BookViews and Sheets maps in xl/workbook.xml.
241
		f3, err := OpenFile(filepath.Join("test", "BadWorkbook.xlsx"))
V
Veniamin Albaev 已提交
242 243 244 245 246 247 248
		f3.GetActiveSheetIndex()
		f3.SetActiveSheet(2)
		assert.NoError(t, err)
	})

	t.Run("OpenNotExistsFile", func(t *testing.T) {
		// Test open a XLSX file with given illegal path.
249
		_, err := OpenFile(filepath.Join("test", "NotExistsFile.xlsx"))
V
Veniamin Albaev 已提交
250 251 252 253
		if assert.Error(t, err) {
			assert.True(t, os.IsNotExist(err), "Expected os.IsNotExists(err) == true")
		}
	})
254 255
}

256
func TestNewFile(t *testing.T) {
257
	// Test create a XLSX file.
258 259 260 261 262 263 264
	f := NewFile()
	f.NewSheet("Sheet1")
	f.NewSheet("XLSXSheet2")
	f.NewSheet("XLSXSheet3")
	f.SetCellInt("XLSXSheet2", "A23", 56)
	f.SetCellStr("Sheet1", "B20", "42")
	f.SetActiveSheet(0)
V
Veniamin Albaev 已提交
265

266
	// Test add picture to sheet with scaling and positioning.
267
	err := f.AddPicture("Sheet1", "H2", filepath.Join("test", "images", "excel.gif"),
268
		`{"x_scale": 0.5, "y_scale": 0.5, "positioning": "absolute"}`)
V
Veniamin Albaev 已提交
269 270
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
271
	}
V
Veniamin Albaev 已提交
272

273
	// Test add picture to worksheet without formatset.
274
	err = f.AddPicture("Sheet1", "C2", filepath.Join("test", "images", "excel.png"), "")
V
Veniamin Albaev 已提交
275 276
	if !assert.NoError(t, err) {
		t.FailNow()
277
	}
V
Veniamin Albaev 已提交
278

279
	// Test add picture to worksheet with invalid formatset.
280
	err = f.AddPicture("Sheet1", "C2", filepath.Join("test", "images", "excel.png"), `{`)
V
Veniamin Albaev 已提交
281 282
	if !assert.Error(t, err) {
		t.FailNow()
283
	}
V
Veniamin Albaev 已提交
284

285
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestNewFile.xlsx")))
xurime's avatar
xurime 已提交
286
}
287

288 289 290 291 292 293
func TestAddDrawingVML(t *testing.T) {
	// Test addDrawingVML with illegal cell coordinates.
	f := NewFile()
	assert.EqualError(t, f.addDrawingVML(0, "", "*", 0, 0), `cannot convert cell "*" to coordinates: invalid cell name "*"`)
}

294
func TestSetCellHyperLink(t *testing.T) {
295
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
296 297 298 299
	if err != nil {
		t.Log(err)
	}
	// Test set cell hyperlink in a work sheet already have hyperlinks.
300
	assert.NoError(t, f.SetCellHyperLink("Sheet1", "B19", "https://github.com/360EntSecGroup-Skylar/excelize", "External"))
301
	// Test add first hyperlink in a work sheet.
302
	assert.NoError(t, f.SetCellHyperLink("Sheet2", "C1", "https://github.com/360EntSecGroup-Skylar/excelize", "External"))
303
	// Test add Location hyperlink in a work sheet.
304
	assert.NoError(t, f.SetCellHyperLink("Sheet2", "D6", "Sheet1!D8", "Location"))
305

306
	assert.EqualError(t, f.SetCellHyperLink("Sheet2", "C3", "Sheet1!D8", ""), `invalid link type ""`)
307

308
	assert.EqualError(t, f.SetCellHyperLink("Sheet2", "", "Sheet1!D60", "Location"), `invalid cell name ""`)
309

310 311
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellHyperLink.xlsx")))

312 313 314 315
	f = NewFile()
	f.workSheetReader("Sheet1")
	f.Sheet["xl/worksheets/sheet1.xml"].Hyperlinks = &xlsxHyperlinks{Hyperlink: make([]xlsxHyperlink, 65530)}
	assert.EqualError(t, f.SetCellHyperLink("Sheet1", "A65531", "https://github.com/360EntSecGroup-Skylar/excelize", "External"), "over maximum limit hyperlinks in a worksheet")
316 317 318 319 320 321

	f = NewFile()
	f.workSheetReader("Sheet1")
	f.Sheet["xl/worksheets/sheet1.xml"].MergeCells = &xlsxMergeCells{Cells: []*xlsxMergeCell{{Ref: "A:A"}}}
	err = f.SetCellHyperLink("Sheet1", "A1", "https://github.com/360EntSecGroup-Skylar/excelize", "External")
	assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
322 323
}

324
func TestGetCellHyperLink(t *testing.T) {
325
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
326 327
	if !assert.NoError(t, err) {
		t.FailNow()
328
	}
V
Veniamin Albaev 已提交
329

330
	_, _, err = f.GetCellHyperLink("Sheet1", "")
331
	assert.EqualError(t, err, `invalid cell name ""`)
332

333
	link, target, err := f.GetCellHyperLink("Sheet1", "A22")
334
	assert.NoError(t, err)
335
	t.Log(link, target)
336
	link, target, err = f.GetCellHyperLink("Sheet2", "D6")
337
	assert.NoError(t, err)
338
	t.Log(link, target)
339
	link, target, err = f.GetCellHyperLink("Sheet3", "H3")
xurime's avatar
xurime 已提交
340
	assert.EqualError(t, err, "sheet Sheet3 is not exist")
341
	t.Log(link, target)
342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358

	f = NewFile()
	f.workSheetReader("Sheet1")
	f.Sheet["xl/worksheets/sheet1.xml"].Hyperlinks = &xlsxHyperlinks{
		Hyperlink: []xlsxHyperlink{{Ref: "A1"}},
	}
	link, target, err = f.GetCellHyperLink("Sheet1", "A1")
	assert.NoError(t, err)
	assert.Equal(t, link, true)
	assert.Equal(t, target, "")

	f.Sheet["xl/worksheets/sheet1.xml"].MergeCells = &xlsxMergeCells{Cells: []*xlsxMergeCell{{Ref: "A:A"}}}
	link, target, err = f.GetCellHyperLink("Sheet1", "A1")
	assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.Equal(t, link, false)
	assert.Equal(t, target, "")

359 360
}

361
func TestSetCellFormula(t *testing.T) {
362
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
363 364
	if !assert.NoError(t, err) {
		t.FailNow()
365
	}
V
Veniamin Albaev 已提交
366

367 368
	f.SetCellFormula("Sheet1", "B19", "SUM(Sheet2!D2,Sheet2!D11)")
	f.SetCellFormula("Sheet1", "C19", "SUM(Sheet2!D2,Sheet2!D9)")
369

370
	// Test set cell formula with illegal rows number.
371
	assert.EqualError(t, f.SetCellFormula("Sheet1", "C", "SUM(Sheet2!D2,Sheet2!D9)"), `cannot convert cell "C" to coordinates: invalid cell name "C"`)
372

373
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellFormula1.xlsx")))
V
Veniamin Albaev 已提交
374

375
	f, err = OpenFile(filepath.Join("test", "CalcChain.xlsx"))
376 377 378 379
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	// Test remove cell formula.
380 381
	f.SetCellFormula("Sheet1", "A1", "")
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellFormula2.xlsx")))
382
	// Test remove all cell formula.
383 384
	f.SetCellFormula("Sheet1", "B1", "")
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellFormula3.xlsx")))
385
}
386 387

func TestSetSheetBackground(t *testing.T) {
388
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
389 390
	if !assert.NoError(t, err) {
		t.FailNow()
391
	}
V
Veniamin Albaev 已提交
392

393
	err = f.SetSheetBackground("Sheet2", filepath.Join("test", "images", "background.jpg"))
V
Veniamin Albaev 已提交
394 395
	if !assert.NoError(t, err) {
		t.FailNow()
396
	}
V
Veniamin Albaev 已提交
397

398
	err = f.SetSheetBackground("Sheet2", filepath.Join("test", "images", "background.jpg"))
V
Veniamin Albaev 已提交
399 400
	if !assert.NoError(t, err) {
		t.FailNow()
401
	}
V
Veniamin Albaev 已提交
402

403
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetSheetBackground.xlsx")))
V
Veniamin Albaev 已提交
404 405 406
}

func TestSetSheetBackgroundErrors(t *testing.T) {
407
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
408 409
	if !assert.NoError(t, err) {
		t.FailNow()
410
	}
V
Veniamin Albaev 已提交
411

412
	err = f.SetSheetBackground("Sheet2", filepath.Join("test", "not_exists", "not_exists.png"))
V
Veniamin Albaev 已提交
413 414 415 416
	if assert.Error(t, err) {
		assert.True(t, os.IsNotExist(err), "Expected os.IsNotExists(err) == true")
	}

417
	err = f.SetSheetBackground("Sheet2", filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
418
	assert.EqualError(t, err, "unsupported image extension")
419
}
xurime's avatar
xurime 已提交
420

421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526
// TestWriteArrayFormula tests the extended options of SetCellFormula by writing an array function
// to a workbook. In the resulting file, the lines 2 and 3 as well as 4 and 5 should have matching
// contents.
func TestWriteArrayFormula(t *testing.T) {
	cell := func(col, row int) string {
		c, err := CoordinatesToCellName(col, row)
		if err != nil {
			t.Fatal(err)
		}

		return c
	}

	f := NewFile()

	sample := []string{"Sample 1", "Sample 2", "Sample 3"}
	values := []int{1855, 1709, 1462, 1115, 1524, 625, 773, 126, 1027, 1696, 1078, 1917, 1109, 1753, 1884, 659, 994, 1911, 1925, 899, 196, 244, 1488, 1056, 1986, 66, 784, 725, 767, 1722, 1541, 1026, 1455, 264, 1538, 877, 1581, 1098, 383, 762, 237, 493, 29, 1923, 474, 430, 585, 688, 308, 200, 1259, 622, 798, 1048, 996, 601, 582, 332, 377, 805, 250, 1860, 1360, 840, 911, 1346, 1651, 1651, 665, 584, 1057, 1145, 925, 1752, 202, 149, 1917, 1398, 1894, 818, 714, 624, 1085, 1566, 635, 78, 313, 1686, 1820, 494, 614, 1913, 271, 1016, 338, 1301, 489, 1733, 1483, 1141}
	assoc := []int{2, 0, 0, 0, 0, 1, 1, 0, 0, 1, 2, 2, 2, 1, 1, 1, 1, 0, 0, 0, 1, 0, 2, 0, 2, 1, 2, 2, 2, 1, 0, 1, 0, 1, 1, 2, 0, 2, 1, 0, 2, 1, 0, 1, 0, 0, 2, 0, 2, 2, 1, 2, 2, 1, 2, 2, 1, 2, 1, 2, 2, 1, 1, 1, 0, 1, 0, 2, 0, 0, 1, 2, 1, 0, 1, 0, 0, 2, 1, 1, 2, 0, 2, 1, 0, 2, 2, 2, 1, 0, 0, 1, 1, 1, 2, 0, 2, 0, 1, 1}
	if len(values) != len(assoc) {
		t.Fatal("values and assoc must be of same length")
	}

	// Average calculates the average of the n-th sample (0 <= n < len(sample)).
	average := func(n int) int {
		sum := 0
		count := 0
		for i := 0; i != len(values); i++ {
			if assoc[i] == n {
				sum += values[i]
				count++
			}
		}

		return int(math.Round(float64(sum) / float64(count)))
	}

	// Stdev calculates the standard deviation of the n-th sample (0 <= n < len(sample)).
	stdev := func(n int) int {
		avg := average(n)

		sum := 0
		count := 0
		for i := 0; i != len(values); i++ {
			if assoc[i] == n {
				sum += (values[i] - avg) * (values[i] - avg)
				count++
			}
		}

		return int(math.Round(math.Sqrt(float64(sum) / float64(count))))
	}

	// Line 2 contains the results of AVERAGEIF
	f.SetCellStr("Sheet1", "A2", "Average")

	// Line 3 contains the average that was calculated in Go
	f.SetCellStr("Sheet1", "A3", "Average (calculated)")

	// Line 4 contains the results of the array function that calculates the standard deviation
	f.SetCellStr("Sheet1", "A4", "Std. deviation")

	// Line 5 contains the standard deviations calculated in Go
	f.SetCellStr("Sheet1", "A5", "Std. deviation (calculated)")

	f.SetCellStr("Sheet1", "B1", sample[0])
	f.SetCellStr("Sheet1", "C1", sample[1])
	f.SetCellStr("Sheet1", "D1", sample[2])

	firstResLine := 8
	f.SetCellStr("Sheet1", cell(1, firstResLine-1), "Result Values")
	f.SetCellStr("Sheet1", cell(2, firstResLine-1), "Sample")

	for i := 0; i != len(values); i++ {
		valCell := cell(1, i+firstResLine)
		assocCell := cell(2, i+firstResLine)

		f.SetCellInt("Sheet1", valCell, values[i])
		f.SetCellStr("Sheet1", assocCell, sample[assoc[i]])
	}

	valRange := fmt.Sprintf("$A$%d:$A$%d", firstResLine, len(values)+firstResLine-1)
	assocRange := fmt.Sprintf("$B$%d:$B$%d", firstResLine, len(values)+firstResLine-1)

	for i := 0; i != len(sample); i++ {
		nameCell := cell(i+2, 1)
		avgCell := cell(i+2, 2)
		calcAvgCell := cell(i+2, 3)
		stdevCell := cell(i+2, 4)
		calcStdevCell := cell(i+2, 5)

		f.SetCellInt("Sheet1", calcAvgCell, average(i))
		f.SetCellInt("Sheet1", calcStdevCell, stdev(i))

		// Average can be done with AVERAGEIF
		f.SetCellFormula("Sheet1", avgCell, fmt.Sprintf("ROUND(AVERAGEIF(%s,%s,%s),0)", assocRange, nameCell, valRange))

		ref := stdevCell + ":" + stdevCell
		t := STCellFormulaTypeArray
		// Use an array formula for standard deviation
		f.SetCellFormula("Sheet1", stdevCell, fmt.Sprintf("ROUND(STDEVP(IF(%s=%s,%s)),0)", assocRange, nameCell, valRange),
			FormulaOpts{}, FormulaOpts{Type: &t}, FormulaOpts{Ref: &ref})
	}

	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestWriteArrayFormula.xlsx")))
}

527
func TestSetCellStyleAlignment(t *testing.T) {
528
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
529 530
	if !assert.NoError(t, err) {
		t.FailNow()
531
	}
V
Veniamin Albaev 已提交
532

533
	var style int
534
	style, err = f.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}}`)
V
Veniamin Albaev 已提交
535 536
	if !assert.NoError(t, err) {
		t.FailNow()
537
	}
V
Veniamin Albaev 已提交
538

539
	assert.NoError(t, f.SetCellStyle("Sheet1", "A22", "A22", style))
540

541
	// Test set cell style with given illegal rows number.
542 543
	assert.EqualError(t, f.SetCellStyle("Sheet1", "A", "A22", style), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.EqualError(t, f.SetCellStyle("Sheet1", "A22", "A", style), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
544

545
	// Test get cell style with given illegal rows number.
546
	index, err := f.GetCellStyle("Sheet1", "A")
547 548
	assert.Equal(t, 0, index)
	assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
V
Veniamin Albaev 已提交
549

550
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleAlignment.xlsx")))
551 552 553
}

func TestSetCellStyleBorder(t *testing.T) {
554
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
555 556
	if !assert.NoError(t, err) {
		t.FailNow()
557
	}
558

V
Veniamin Albaev 已提交
559
	var style int
560

561
	// Test set border on overlapping area with vertical variants shading styles gradient fill.
562
	style, err = f.NewStyle(`{"border":[{"type":"left","color":"0000FF","style":2},{"type":"top","color":"00FF00","style":12},{"type":"bottom","color":"FFFF00","style":5},{"type":"right","color":"FF0000","style":6},{"type":"diagonalDown","color":"A020F0","style":9},{"type":"diagonalUp","color":"A020F0","style":8}]}`)
V
Veniamin Albaev 已提交
563 564
	if !assert.NoError(t, err) {
		t.FailNow()
565
	}
566
	assert.NoError(t, f.SetCellStyle("Sheet1", "J21", "L25", style))
567

568
	style, err = f.NewStyle(`{"border":[{"type":"left","color":"0000FF","style":2},{"type":"top","color":"00FF00","style":3},{"type":"bottom","color":"FFFF00","style":4},{"type":"right","color":"FF0000","style":5},{"type":"diagonalDown","color":"A020F0","style":6},{"type":"diagonalUp","color":"A020F0","style":7}],"fill":{"type":"gradient","color":["#FFFFFF","#E0EBF5"],"shading":1}}`)
V
Veniamin Albaev 已提交
569 570
	if !assert.NoError(t, err) {
		t.FailNow()
571
	}
572
	assert.NoError(t, f.SetCellStyle("Sheet1", "M28", "K24", style))
573

574
	style, err = f.NewStyle(`{"border":[{"type":"left","color":"0000FF","style":2},{"type":"top","color":"00FF00","style":3},{"type":"bottom","color":"FFFF00","style":4},{"type":"right","color":"FF0000","style":5},{"type":"diagonalDown","color":"A020F0","style":6},{"type":"diagonalUp","color":"A020F0","style":7}],"fill":{"type":"gradient","color":["#FFFFFF","#E0EBF5"],"shading":4}}`)
V
Veniamin Albaev 已提交
575 576
	if !assert.NoError(t, err) {
		t.FailNow()
577
	}
578
	assert.NoError(t, f.SetCellStyle("Sheet1", "M28", "K24", style))
579

580
	// Test set border and solid style pattern fill for a single cell.
581
	style, err = f.NewStyle(`{"border":[{"type":"left","color":"0000FF","style":8},{"type":"top","color":"00FF00","style":9},{"type":"bottom","color":"FFFF00","style":10},{"type":"right","color":"FF0000","style":11},{"type":"diagonalDown","color":"A020F0","style":12},{"type":"diagonalUp","color":"A020F0","style":13}],"fill":{"type":"pattern","color":["#E0EBF5"],"pattern":1}}`)
V
Veniamin Albaev 已提交
582 583
	if !assert.NoError(t, err) {
		t.FailNow()
584
	}
585

586
	assert.NoError(t, f.SetCellStyle("Sheet1", "O22", "O22", style))
V
Veniamin Albaev 已提交
587

588
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleBorder.xlsx")))
V
Veniamin Albaev 已提交
589 590 591
}

func TestSetCellStyleBorderErrors(t *testing.T) {
592
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
593 594 595 596 597
	if !assert.NoError(t, err) {
		t.FailNow()
	}

	// Set border with invalid style parameter.
598
	_, err = f.NewStyle("")
V
Veniamin Albaev 已提交
599 600 601 602 603
	if !assert.EqualError(t, err, "unexpected end of JSON input") {
		t.FailNow()
	}

	// Set border with invalid style index number.
604
	_, err = f.NewStyle(`{"border":[{"type":"left","color":"0000FF","style":-1},{"type":"top","color":"00FF00","style":14},{"type":"bottom","color":"FFFF00","style":5},{"type":"right","color":"FF0000","style":6},{"type":"diagonalDown","color":"A020F0","style":9},{"type":"diagonalUp","color":"A020F0","style":8}]}`)
V
Veniamin Albaev 已提交
605 606
	if !assert.NoError(t, err) {
		t.FailNow()
607 608 609
	}
}

610
func TestSetCellStyleNumberFormat(t *testing.T) {
611
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
612 613
	if !assert.NoError(t, err) {
		t.FailNow()
614
	}
V
Veniamin Albaev 已提交
615

xurime's avatar
xurime 已提交
616
	// Test only set fill and number format for a cell.
617 618 619 620 621 622 623 624 625
	col := []string{"L", "M", "N", "O", "P"}
	data := []int{0, 1, 2, 3, 4, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49}
	value := []string{"37947.7500001", "-37947.7500001", "0.007", "2.1", "String"}
	for i, v := range value {
		for k, d := range data {
			c := col[i] + strconv.Itoa(k+1)
			var val float64
			val, err = strconv.ParseFloat(v, 64)
			if err != nil {
626
				f.SetCellValue("Sheet2", c, v)
627
			} else {
628
				f.SetCellValue("Sheet2", c, val)
629
			}
630
			style, err := f.NewStyle(`{"fill":{"type":"gradient","color":["#FFFFFF","#E0EBF5"],"shading":5},"number_format": ` + strconv.Itoa(d) + `}`)
V
Veniamin Albaev 已提交
631 632
			if !assert.NoError(t, err) {
				t.FailNow()
633
			}
634 635
			assert.NoError(t, f.SetCellStyle("Sheet2", c, c, style))
			t.Log(f.GetCellValue("Sheet2", c))
636 637
		}
	}
638
	var style int
639
	style, err = f.NewStyle(`{"number_format":-1}`)
V
Veniamin Albaev 已提交
640 641
	if !assert.NoError(t, err) {
		t.FailNow()
642
	}
643
	assert.NoError(t, f.SetCellStyle("Sheet2", "L33", "L33", style))
V
Veniamin Albaev 已提交
644

645
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleNumberFormat.xlsx")))
646 647
}

648
func TestSetCellStyleCurrencyNumberFormat(t *testing.T) {
V
Veniamin Albaev 已提交
649
	t.Run("TestBook3", func(t *testing.T) {
650
		f, err := prepareTestBook3()
V
Veniamin Albaev 已提交
651 652 653
		if !assert.NoError(t, err) {
			t.FailNow()
		}
654

655 656
		f.SetCellValue("Sheet1", "A1", 56)
		f.SetCellValue("Sheet1", "A2", -32.3)
V
Veniamin Albaev 已提交
657
		var style int
658
		style, err = f.NewStyle(`{"number_format": 188, "decimal_places": -1}`)
V
Veniamin Albaev 已提交
659 660 661
		if !assert.NoError(t, err) {
			t.FailNow()
		}
662

663 664
		assert.NoError(t, f.SetCellStyle("Sheet1", "A1", "A1", style))
		style, err = f.NewStyle(`{"number_format": 188, "decimal_places": 31, "negred": true}`)
V
Veniamin Albaev 已提交
665 666 667
		if !assert.NoError(t, err) {
			t.FailNow()
		}
668

669
		assert.NoError(t, f.SetCellStyle("Sheet1", "A2", "A2", style))
670

671
		assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleCurrencyNumberFormat.TestBook3.xlsx")))
V
Veniamin Albaev 已提交
672
	})
673

V
Veniamin Albaev 已提交
674
	t.Run("TestBook4", func(t *testing.T) {
675
		f, err := prepareTestBook4()
V
Veniamin Albaev 已提交
676 677 678
		if !assert.NoError(t, err) {
			t.FailNow()
		}
679 680
		f.SetCellValue("Sheet1", "A1", 42920.5)
		f.SetCellValue("Sheet1", "A2", 42920.5)
V
Veniamin Albaev 已提交
681

682
		_, err = f.NewStyle(`{"number_format": 26, "lang": "zh-tw"}`)
V
Veniamin Albaev 已提交
683 684 685 686
		if !assert.NoError(t, err) {
			t.FailNow()
		}

687
		style, err := f.NewStyle(`{"number_format": 27}`)
V
Veniamin Albaev 已提交
688 689 690 691
		if !assert.NoError(t, err) {
			t.FailNow()
		}

692 693
		assert.NoError(t, f.SetCellStyle("Sheet1", "A1", "A1", style))
		style, err = f.NewStyle(`{"number_format": 31, "lang": "ko-kr"}`)
V
Veniamin Albaev 已提交
694 695 696 697
		if !assert.NoError(t, err) {
			t.FailNow()
		}

698
		assert.NoError(t, f.SetCellStyle("Sheet1", "A2", "A2", style))
V
Veniamin Albaev 已提交
699

700
		style, err = f.NewStyle(`{"number_format": 71, "lang": "th-th"}`)
V
Veniamin Albaev 已提交
701 702 703
		if !assert.NoError(t, err) {
			t.FailNow()
		}
704
		assert.NoError(t, f.SetCellStyle("Sheet1", "A2", "A2", style))
V
Veniamin Albaev 已提交
705

706
		assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleCurrencyNumberFormat.TestBook4.xlsx")))
V
Veniamin Albaev 已提交
707
	})
708 709
}

710
func TestSetCellStyleCustomNumberFormat(t *testing.T) {
711 712 713 714
	f := NewFile()
	f.SetCellValue("Sheet1", "A1", 42920.5)
	f.SetCellValue("Sheet1", "A2", 42920.5)
	style, err := f.NewStyle(`{"custom_number_format": "[$-380A]dddd\\,\\ dd\" de \"mmmm\" de \"yyyy;@"}`)
715 716 717
	if err != nil {
		t.Log(err)
	}
718 719
	assert.NoError(t, f.SetCellStyle("Sheet1", "A1", "A1", style))
	style, err = f.NewStyle(`{"custom_number_format": "[$-380A]dddd\\,\\ dd\" de \"mmmm\" de \"yyyy;@"}`)
720 721 722
	if err != nil {
		t.Log(err)
	}
723
	assert.NoError(t, f.SetCellStyle("Sheet1", "A2", "A2", style))
V
Veniamin Albaev 已提交
724

725
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleCustomNumberFormat.xlsx")))
726 727
}

728
func TestSetCellStyleFill(t *testing.T) {
729
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
730 731
	if !assert.NoError(t, err) {
		t.FailNow()
732
	}
V
Veniamin Albaev 已提交
733

734
	var style int
735
	// Test set fill for cell with invalid parameter.
736
	style, err = f.NewStyle(`{"fill":{"type":"gradient","color":["#FFFFFF","#E0EBF5"],"shading":6}}`)
V
Veniamin Albaev 已提交
737 738
	if !assert.NoError(t, err) {
		t.FailNow()
739
	}
740
	assert.NoError(t, f.SetCellStyle("Sheet1", "O23", "O23", style))
741

742
	style, err = f.NewStyle(`{"fill":{"type":"gradient","color":["#FFFFFF"],"shading":1}}`)
V
Veniamin Albaev 已提交
743 744
	if !assert.NoError(t, err) {
		t.FailNow()
745
	}
746
	assert.NoError(t, f.SetCellStyle("Sheet1", "O23", "O23", style))
747

748
	style, err = f.NewStyle(`{"fill":{"type":"pattern","color":[],"pattern":1}}`)
V
Veniamin Albaev 已提交
749 750
	if !assert.NoError(t, err) {
		t.FailNow()
751
	}
752
	assert.NoError(t, f.SetCellStyle("Sheet1", "O23", "O23", style))
xurime's avatar
xurime 已提交
753

754
	style, err = f.NewStyle(`{"fill":{"type":"pattern","color":["#E0EBF5"],"pattern":19}}`)
V
Veniamin Albaev 已提交
755 756
	if !assert.NoError(t, err) {
		t.FailNow()
757
	}
758
	assert.NoError(t, f.SetCellStyle("Sheet1", "O23", "O23", style))
759

760
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleFill.xlsx")))
761 762 763
}

func TestSetCellStyleFont(t *testing.T) {
764
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
765 766
	if !assert.NoError(t, err) {
		t.FailNow()
767
	}
V
Veniamin Albaev 已提交
768

769
	var style int
xurime's avatar
xurime 已提交
770
	style, err = f.NewStyle(`{"font":{"bold":true,"italic":true,"family":"Times New Roman","size":36,"color":"#777777","underline":"single"}}`)
V
Veniamin Albaev 已提交
771 772
	if !assert.NoError(t, err) {
		t.FailNow()
773
	}
V
Veniamin Albaev 已提交
774

775
	assert.NoError(t, f.SetCellStyle("Sheet2", "A1", "A1", style))
776

777
	style, err = f.NewStyle(`{"font":{"italic":true,"underline":"double"}}`)
V
Veniamin Albaev 已提交
778 779
	if !assert.NoError(t, err) {
		t.FailNow()
780
	}
V
Veniamin Albaev 已提交
781

782
	assert.NoError(t, f.SetCellStyle("Sheet2", "A2", "A2", style))
783

784
	style, err = f.NewStyle(`{"font":{"bold":true}}`)
V
Veniamin Albaev 已提交
785 786
	if !assert.NoError(t, err) {
		t.FailNow()
787
	}
V
Veniamin Albaev 已提交
788

789
	assert.NoError(t, f.SetCellStyle("Sheet2", "A3", "A3", style))
790

791
	style, err = f.NewStyle(`{"font":{"bold":true,"family":"","size":0,"color":"","underline":""}}`)
V
Veniamin Albaev 已提交
792 793
	if !assert.NoError(t, err) {
		t.FailNow()
794
	}
V
Veniamin Albaev 已提交
795

796
	assert.NoError(t, f.SetCellStyle("Sheet2", "A4", "A4", style))
797

xurime's avatar
xurime 已提交
798
	style, err = f.NewStyle(`{"font":{"color":"#777777","strike":true}}`)
V
Veniamin Albaev 已提交
799 800
	if !assert.NoError(t, err) {
		t.FailNow()
801
	}
V
Veniamin Albaev 已提交
802

803
	assert.NoError(t, f.SetCellStyle("Sheet2", "A5", "A5", style))
V
Veniamin Albaev 已提交
804

805
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetCellStyleFont.xlsx")))
806
}
807

808
func TestSetCellStyleProtection(t *testing.T) {
809
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
810 811
	if !assert.NoError(t, err) {
		t.FailNow()
812
	}
V
Veniamin Albaev 已提交
813

814
	var style int
815
	style, err = f.NewStyle(`{"protection":{"hidden":true, "locked":true}}`)
V
Veniamin Albaev 已提交
816 817
	if !assert.NoError(t, err) {
		t.FailNow()
818
	}
V
Veniamin Albaev 已提交
819

820 821
	assert.NoError(t, f.SetCellStyle("Sheet2", "A6", "A6", style))
	err = f.SaveAs(filepath.Join("test", "TestSetCellStyleProtection.xlsx"))
V
Veniamin Albaev 已提交
822 823
	if !assert.NoError(t, err) {
		t.FailNow()
824 825 826
	}
}

V
Veniamin Albaev 已提交
827 828
func TestSetDeleteSheet(t *testing.T) {
	t.Run("TestBook3", func(t *testing.T) {
829
		f, err := prepareTestBook3()
V
Veniamin Albaev 已提交
830 831 832 833
		if !assert.NoError(t, err) {
			t.FailNow()
		}

834 835
		f.DeleteSheet("XLSXSheet3")
		assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetDeleteSheet.TestBook3.xlsx")))
V
Veniamin Albaev 已提交
836 837 838
	})

	t.Run("TestBook4", func(t *testing.T) {
839
		f, err := prepareTestBook4()
V
Veniamin Albaev 已提交
840 841 842
		if !assert.NoError(t, err) {
			t.FailNow()
		}
843 844 845 846
		f.DeleteSheet("Sheet1")
		f.AddComment("Sheet1", "A1", "")
		f.AddComment("Sheet1", "A1", `{"author":"Excelize: ","text":"This is a comment."}`)
		assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetDeleteSheet.TestBook4.xlsx")))
V
Veniamin Albaev 已提交
847
	})
848
}
849

850
func TestSheetVisibility(t *testing.T) {
851
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
852 853
	if !assert.NoError(t, err) {
		t.FailNow()
854
	}
V
Veniamin Albaev 已提交
855

856 857 858 859
	f.SetSheetVisible("Sheet2", false)
	f.SetSheetVisible("Sheet1", false)
	f.SetSheetVisible("Sheet1", true)
	f.GetSheetVisible("Sheet1")
V
Veniamin Albaev 已提交
860

861
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSheetVisibility.xlsx")))
862 863
}

864
func TestCopySheet(t *testing.T) {
865
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
866 867
	if !assert.NoError(t, err) {
		t.FailNow()
868
	}
V
Veniamin Albaev 已提交
869

870
	idx := f.NewSheet("CopySheet")
871
	assert.NoError(t, f.CopySheet(1, idx))
V
Veniamin Albaev 已提交
872

873
	f.SetCellValue("CopySheet", "F1", "Hello")
874
	val, err := f.GetCellValue("Sheet1", "F1")
875 876
	assert.NoError(t, err)
	assert.NotEqual(t, "Hello", val)
V
Veniamin Albaev 已提交
877

878
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestCopySheet.xlsx")))
V
Veniamin Albaev 已提交
879 880 881
}

func TestCopySheetError(t *testing.T) {
882
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
883 884
	if !assert.NoError(t, err) {
		t.FailNow()
885
	}
V
Veniamin Albaev 已提交
886

887 888
	assert.EqualError(t, f.copySheet(0, -1), "sheet  is not exist")
	if !assert.EqualError(t, f.CopySheet(0, -1), "invalid worksheet index") {
V
Veniamin Albaev 已提交
889
		t.FailNow()
890
	}
V
Veniamin Albaev 已提交
891

892
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestCopySheetError.xlsx")))
893
}
894

xurime's avatar
xurime 已提交
895
func TestAddTable(t *testing.T) {
896
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
897 898
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
899
	}
V
Veniamin Albaev 已提交
900

901
	err = f.AddTable("Sheet1", "B26", "A21", `{}`)
V
Veniamin Albaev 已提交
902 903
	if !assert.NoError(t, err) {
		t.FailNow()
904
	}
V
Veniamin Albaev 已提交
905

906
	err = f.AddTable("Sheet2", "A2", "B5", `{"table_name":"table","table_style":"TableStyleMedium2", "show_first_column":true,"show_last_column":true,"show_row_stripes":false,"show_column_stripes":true}`)
V
Veniamin Albaev 已提交
907 908
	if !assert.NoError(t, err) {
		t.FailNow()
909
	}
V
Veniamin Albaev 已提交
910

911
	err = f.AddTable("Sheet2", "F1", "F1", `{"table_style":"TableStyleMedium8"}`)
V
Veniamin Albaev 已提交
912 913
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
914
	}
V
Veniamin Albaev 已提交
915

916
	// Test add table with illegal formatset.
917
	assert.EqualError(t, f.AddTable("Sheet1", "B26", "A21", `{x}`), "invalid character 'x' looking for beginning of object key string")
918
	// Test add table with illegal cell coordinates.
919 920
	assert.EqualError(t, f.AddTable("Sheet1", "A", "B1", `{}`), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.EqualError(t, f.AddTable("Sheet1", "A1", "B", `{}`), `cannot convert cell "B" to coordinates: invalid cell name "B"`)
921

922
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddTable.xlsx")))
923 924

	// Test addTable with illegal cell coordinates.
925
	f = NewFile()
926 927
	assert.EqualError(t, f.addTable("sheet1", "", 0, 0, 0, 0, 0, nil), "invalid cell coordinates [0, 0]")
	assert.EqualError(t, f.addTable("sheet1", "", 1, 1, 0, 0, 0, nil), "invalid cell coordinates [0, 0]")
xurime's avatar
xurime 已提交
928 929
}

930
func TestAddShape(t *testing.T) {
931
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
932 933
	if !assert.NoError(t, err) {
		t.FailNow()
934
	}
V
Veniamin Albaev 已提交
935

936 937 938 939 940 941 942
	assert.NoError(t, f.AddShape("Sheet1", "A30", `{"type":"rect","paragraph":[{"text":"Rectangle","font":{"color":"CD5C5C"}},{"text":"Shape","font":{"bold":true,"color":"2980B9"}}]}`))
	assert.NoError(t, f.AddShape("Sheet1", "B30", `{"type":"rect","paragraph":[{"text":"Rectangle"},{}]}`))
	assert.NoError(t, f.AddShape("Sheet1", "C30", `{"type":"rect","paragraph":[]}`))
	assert.EqualError(t, f.AddShape("Sheet3", "H1", `{"type":"ellipseRibbon", "color":{"line":"#4286f4","fill":"#8eb9ff"}, "paragraph":[{"font":{"bold":true,"italic":true,"family":"Times New Roman","size":36,"color":"#777777","underline":"single"}}], "height": 90}`), "sheet Sheet3 is not exist")
	assert.EqualError(t, f.AddShape("Sheet3", "H1", ""), "unexpected end of JSON input")
	assert.EqualError(t, f.AddShape("Sheet1", "A", `{"type":"rect","paragraph":[{"text":"Rectangle","font":{"color":"CD5C5C"}},{"text":"Shape","font":{"bold":true,"color":"2980B9"}}]}`), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddShape1.xlsx")))
V
Veniamin Albaev 已提交
943

944 945 946 947
	// Test add first shape for given sheet.
	f = NewFile()
	assert.NoError(t, f.AddShape("Sheet1", "A1", `{"type":"ellipseRibbon", "color":{"line":"#4286f4","fill":"#8eb9ff"}, "paragraph":[{"font":{"bold":true,"italic":true,"family":"Times New Roman","size":36,"color":"#777777","underline":"single"}}], "height": 90}`))
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddShape2.xlsx")))
948 949
}

950 951 952 953 954
func TestGetSheetComments(t *testing.T) {
	f := NewFile()
	assert.Equal(t, "", f.getSheetComments(0))
}

955
func TestAutoFilter(t *testing.T) {
956 957
	outFile := filepath.Join("test", "TestAutoFilter%d.xlsx")

958
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
959 960
	if !assert.NoError(t, err) {
		t.FailNow()
961
	}
V
Veniamin Albaev 已提交
962 963 964

	formats := []string{
		``,
965 966 967 968 969 970 971
		`{"column":"B","expression":"x != blanks"}`,
		`{"column":"B","expression":"x == blanks"}`,
		`{"column":"B","expression":"x != nonblanks"}`,
		`{"column":"B","expression":"x == nonblanks"}`,
		`{"column":"B","expression":"x <= 1 and x >= 2"}`,
		`{"column":"B","expression":"x == 1 or x == 2"}`,
		`{"column":"B","expression":"x == 1 or x == 2*"}`,
V
Veniamin Albaev 已提交
972 973 974 975
	}

	for i, format := range formats {
		t.Run(fmt.Sprintf("Expression%d", i+1), func(t *testing.T) {
976
			err = f.AutoFilter("Sheet1", "D4", "B1", format)
xurime's avatar
xurime 已提交
977
			assert.NoError(t, err)
978
			assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, i+1)))
V
Veniamin Albaev 已提交
979 980 981
		})
	}

982
	// testing AutoFilter with illegal cell coordinates.
983 984
	assert.EqualError(t, f.AutoFilter("Sheet1", "A", "B1", ""), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.EqualError(t, f.AutoFilter("Sheet1", "A1", "B", ""), `cannot convert cell "B" to coordinates: invalid cell name "B"`)
V
Veniamin Albaev 已提交
985 986 987
}

func TestAutoFilterError(t *testing.T) {
988 989
	outFile := filepath.Join("test", "TestAutoFilterError%d.xlsx")

990
	f, err := prepareTestBook1()
V
Veniamin Albaev 已提交
991 992 993 994 995
	if !assert.NoError(t, err) {
		t.FailNow()
	}

	formats := []string{
996 997 998 999 1000 1001 1002
		`{"column":"B","expression":"x <= 1 and x >= blanks"}`,
		`{"column":"B","expression":"x -- y or x == *2*"}`,
		`{"column":"B","expression":"x != y or x ? *2"}`,
		`{"column":"B","expression":"x -- y o r x == *2"}`,
		`{"column":"B","expression":"x -- y"}`,
		`{"column":"A","expression":"x -- y"}`,
	}
V
Veniamin Albaev 已提交
1003 1004
	for i, format := range formats {
		t.Run(fmt.Sprintf("Expression%d", i+1), func(t *testing.T) {
1005
			err = f.AutoFilter("Sheet3", "D4", "B1", format)
V
Veniamin Albaev 已提交
1006
			if assert.Error(t, err) {
1007
				assert.NoError(t, f.SaveAs(fmt.Sprintf(outFile, i+1)))
V
Veniamin Albaev 已提交
1008 1009
			}
		})
1010 1011 1012
	}
}

xurime's avatar
xurime 已提交
1013 1014 1015 1016 1017 1018 1019
func TestSetActiveSheet(t *testing.T) {
	f := NewFile()
	f.WorkBook.BookViews = nil
	f.SetActiveSheet(1)
	f.WorkBook.BookViews = &xlsxBookViews{WorkBookView: []xlsxWorkBookView{}}
	f.Sheet["xl/worksheets/sheet1.xml"].SheetViews = &xlsxSheetViews{SheetView: []xlsxSheetView{}}
	f.SetActiveSheet(1)
1020 1021
	f.Sheet["xl/worksheets/sheet1.xml"].SheetViews = nil
	f.SetActiveSheet(1)
xurime's avatar
xurime 已提交
1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
}

func TestSetSheetVisible(t *testing.T) {
	f := NewFile()
	f.WorkBook.Sheets.Sheet[0].Name = "SheetN"
	assert.EqualError(t, f.SetSheetVisible("Sheet1", false), "sheet SheetN is not exist")
}

func TestGetActiveSheetIndex(t *testing.T) {
	f := NewFile()
	f.WorkBook.BookViews = nil
	assert.Equal(t, 1, f.GetActiveSheetIndex())
}

func TestRelsWriter(t *testing.T) {
	f := NewFile()
	f.Relationships["xl/worksheets/sheet/rels/sheet1.xml.rel"] = &xlsxRelationships{}
	f.relsWriter()
}

func TestGetSheetView(t *testing.T) {
1043
	f := NewFile()
xurime's avatar
xurime 已提交
1044 1045
	_, err := f.getSheetView("SheetN", 0)
	assert.EqualError(t, err, "sheet SheetN is not exist")
xurime's avatar
xurime 已提交
1046 1047
}

1048
func TestConditionalFormat(t *testing.T) {
1049 1050
	f := NewFile()
	sheet1 := f.GetSheetName(1)
1051

1052
	fillCells(f, sheet1, 10, 15)
1053

1054 1055 1056
	var format1, format2, format3 int
	var err error
	// Rose format for bad conditional.
1057
	format1, err = f.NewConditionalStyle(`{"font":{"color":"#9A0511"},"fill":{"type":"pattern","color":["#FEC7CE"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1058 1059 1060 1061
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1062
	// Light yellow format for neutral conditional.
1063
	format2, err = f.NewConditionalStyle(`{"fill":{"type":"pattern","color":["#FEEAA0"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1064 1065 1066 1067
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1068
	// Light green format for good conditional.
1069
	format3, err = f.NewConditionalStyle(`{"font":{"color":"#09600B"},"fill":{"type":"pattern","color":["#C7EECF"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1070 1071 1072 1073
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1074
	// Color scales: 2 color.
1075
	f.SetConditionalFormat(sheet1, "A1:A10", `[{"type":"2_color_scale","criteria":"=","min_type":"min","max_type":"max","min_color":"#F8696B","max_color":"#63BE7B"}]`)
1076
	// Color scales: 3 color.
1077
	f.SetConditionalFormat(sheet1, "B1:B10", `[{"type":"3_color_scale","criteria":"=","min_type":"min","mid_type":"percentile","max_type":"max","min_color":"#F8696B","mid_color":"#FFEB84","max_color":"#63BE7B"}]`)
1078
	// Hightlight cells rules: between...
1079
	f.SetConditionalFormat(sheet1, "C1:C10", fmt.Sprintf(`[{"type":"cell","criteria":"between","format":%d,"minimum":"6","maximum":"8"}]`, format1))
1080
	// Hightlight cells rules: Greater Than...
1081
	f.SetConditionalFormat(sheet1, "D1:D10", fmt.Sprintf(`[{"type":"cell","criteria":">","format":%d,"value":"6"}]`, format3))
1082
	// Hightlight cells rules: Equal To...
1083
	f.SetConditionalFormat(sheet1, "E1:E10", fmt.Sprintf(`[{"type":"top","criteria":"=","format":%d}]`, format3))
1084
	// Hightlight cells rules: Not Equal To...
1085
	f.SetConditionalFormat(sheet1, "F1:F10", fmt.Sprintf(`[{"type":"unique","criteria":"=","format":%d}]`, format2))
1086
	// Hightlight cells rules: Duplicate Values...
1087
	f.SetConditionalFormat(sheet1, "G1:G10", fmt.Sprintf(`[{"type":"duplicate","criteria":"=","format":%d}]`, format2))
1088
	// Top/Bottom rules: Top 10%.
1089
	f.SetConditionalFormat(sheet1, "H1:H10", fmt.Sprintf(`[{"type":"top","criteria":"=","format":%d,"value":"6","percent":true}]`, format1))
1090
	// Top/Bottom rules: Above Average...
1091
	f.SetConditionalFormat(sheet1, "I1:I10", fmt.Sprintf(`[{"type":"average","criteria":"=","format":%d, "above_average": true}]`, format3))
1092
	// Top/Bottom rules: Below Average...
1093
	f.SetConditionalFormat(sheet1, "J1:J10", fmt.Sprintf(`[{"type":"average","criteria":"=","format":%d, "above_average": false}]`, format1))
1094
	// Data Bars: Gradient Fill.
1095
	f.SetConditionalFormat(sheet1, "K1:K10", `[{"type":"data_bar", "criteria":"=", "min_type":"min","max_type":"max","bar_color":"#638EC6"}]`)
1096
	// Use a formula to determine which cells to format.
1097
	f.SetConditionalFormat(sheet1, "L1:L10", fmt.Sprintf(`[{"type":"formula", "criteria":"L2<3", "format":%d}]`, format1))
1098
	// Test set invalid format set in conditional format
1099
	f.SetConditionalFormat(sheet1, "L1:L10", "")
V
Veniamin Albaev 已提交
1100

1101
	err = f.SaveAs(filepath.Join("test", "TestConditionalFormat.xlsx"))
V
Veniamin Albaev 已提交
1102 1103
	if !assert.NoError(t, err) {
		t.FailNow()
1104 1105 1106
	}

	// Set conditional format with illegal valid type.
1107
	f.SetConditionalFormat(sheet1, "K1:K10", `[{"type":"", "criteria":"=", "min_type":"min","max_type":"max","bar_color":"#638EC6"}]`)
1108
	// Set conditional format with illegal criteria type.
1109
	f.SetConditionalFormat(sheet1, "K1:K10", `[{"type":"data_bar", "criteria":"", "min_type":"min","max_type":"max","bar_color":"#638EC6"}]`)
V
Veniamin Albaev 已提交
1110 1111

	// Set conditional format with file without dxfs element shold not return error.
1112
	f, err = OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1113 1114 1115 1116
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1117
	_, err = f.NewConditionalStyle(`{"font":{"color":"#9A0511"},"fill":{"type":"pattern","color":["#FEC7CE"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1118 1119 1120
	if !assert.NoError(t, err) {
		t.FailNow()
	}
1121
}
xurime's avatar
xurime 已提交
1122

V
Veniamin Albaev 已提交
1123
func TestConditionalFormatError(t *testing.T) {
1124 1125
	f := NewFile()
	sheet1 := f.GetSheetName(1)
1126

1127
	fillCells(f, sheet1, 10, 15)
V
Veniamin Albaev 已提交
1128 1129

	// Set conditional format with illegal JSON string should return error
1130
	_, err := f.NewConditionalStyle("")
V
Veniamin Albaev 已提交
1131 1132
	if !assert.EqualError(t, err, "unexpected end of JSON input") {
		t.FailNow()
xurime's avatar
xurime 已提交
1133 1134
	}
}
1135 1136

func TestSharedStrings(t *testing.T) {
1137
	f, err := OpenFile(filepath.Join("test", "SharedStrings.xlsx"))
V
Veniamin Albaev 已提交
1138 1139
	if !assert.NoError(t, err) {
		t.FailNow()
1140
	}
H
Harris 已提交
1141 1142 1143 1144 1145
	rows, err := f.GetRows("Sheet1")
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	assert.Equal(t, "A", rows[0][0])
H
Harris 已提交
1146 1147 1148 1149 1150
	rows, err = f.GetRows("Sheet2")
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	assert.Equal(t, "Test Weight (Kgs)", rows[0][0])
1151
}
1152 1153

func TestSetSheetRow(t *testing.T) {
1154
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1155 1156
	if !assert.NoError(t, err) {
		t.FailNow()
1157
	}
V
Veniamin Albaev 已提交
1158

1159
	f.SetSheetRow("Sheet1", "B27", &[]interface{}{"cell", nil, int32(42), float64(42), time.Now().UTC()})
1160

1161
	assert.EqualError(t, f.SetSheetRow("Sheet1", "", &[]interface{}{"cell", nil, 2}),
1162
		`cannot convert cell "" to coordinates: invalid cell name ""`)
V
Veniamin Albaev 已提交
1163

1164 1165 1166
	assert.EqualError(t, f.SetSheetRow("Sheet1", "B27", []interface{}{}), `pointer to slice expected`)
	assert.EqualError(t, f.SetSheetRow("Sheet1", "B27", &f), `pointer to slice expected`)
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestSetSheetRow.xlsx")))
1167
}
1168

xurime's avatar
xurime 已提交
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193
func TestThemeColor(t *testing.T) {
	t.Log(ThemeColor("000000", -0.1))
	t.Log(ThemeColor("000000", 0))
	t.Log(ThemeColor("000000", 1))
}

func TestHSL(t *testing.T) {
	var hsl HSL
	t.Log(hsl.RGBA())
	t.Log(hslModel(hsl))
	t.Log(hslModel(color.Gray16{Y: uint16(1)}))
	t.Log(HSLToRGB(0, 1, 0.4))
	t.Log(HSLToRGB(0, 1, 0.6))
	t.Log(hueToRGB(0, 0, -1))
	t.Log(hueToRGB(0, 0, 2))
	t.Log(hueToRGB(0, 0, 1.0/7))
	t.Log(hueToRGB(0, 0, 0.4))
	t.Log(hueToRGB(0, 0, 2.0/4))
	t.Log(RGBToHSL(255, 255, 0))
	t.Log(RGBToHSL(0, 255, 255))
	t.Log(RGBToHSL(250, 100, 50))
	t.Log(RGBToHSL(50, 100, 250))
	t.Log(RGBToHSL(250, 50, 100))
}

1194
func TestProtectSheet(t *testing.T) {
1195 1196 1197
	f := NewFile()
	f.ProtectSheet("Sheet1", nil)
	f.ProtectSheet("Sheet1", &FormatSheetProtection{
1198 1199 1200
		Password:      "password",
		EditScenarios: false,
	})
V
Veniamin Albaev 已提交
1201

1202
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestProtectSheet.xlsx")))
xurime's avatar
xurime 已提交
1203
	// Test protect not exists worksheet.
1204
	assert.EqualError(t, f.ProtectSheet("SheetN", nil), "sheet SheetN is not exist")
1205 1206
}

1207
func TestUnprotectSheet(t *testing.T) {
1208
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1209 1210
	if !assert.NoError(t, err) {
		t.FailNow()
1211
	}
xurime's avatar
xurime 已提交
1212
	// Test unprotect not exists worksheet.
1213
	assert.EqualError(t, f.UnprotectSheet("SheetN"), "sheet SheetN is not exist")
V
Veniamin Albaev 已提交
1214

1215 1216
	f.UnprotectSheet("Sheet1")
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestUnprotectSheet.xlsx")))
1217
}
xurime's avatar
xurime 已提交
1218 1219 1220 1221 1222 1223

func TestSetDefaultTimeStyle(t *testing.T) {
	f := NewFile()
	// Test set default time style on not exists worksheet.
	assert.EqualError(t, f.setDefaultTimeStyle("SheetN", "", 0), "sheet SheetN is not exist")
}
1224

1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
func TestAddVBAProject(t *testing.T) {
	f := NewFile()
	assert.NoError(t, f.SetSheetPrOptions("Sheet1", CodeName("Sheet1")))
	assert.EqualError(t, f.AddVBAProject("macros.bin"), "stat macros.bin: no such file or directory")
	assert.EqualError(t, f.AddVBAProject(filepath.Join("test", "Book1.xlsx")), "unsupported VBA project extension")
	assert.NoError(t, f.AddVBAProject(filepath.Join("test", "vbaProject.bin")))
	// Test add VBA project twice.
	assert.NoError(t, f.AddVBAProject(filepath.Join("test", "vbaProject.bin")))
	assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddVBAProject.xlsm")))
}

xurime's avatar
xurime 已提交
1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290
func TestContentTypesReader(t *testing.T) {
	// Test unsupport charset.
	f := NewFile()
	f.ContentTypes = nil
	f.XLSX["[Content_Types].xml"] = MacintoshCyrillicCharset
	f.contentTypesReader()
}

func TestWorkbookReader(t *testing.T) {
	// Test unsupport charset.
	f := NewFile()
	f.WorkBook = nil
	f.XLSX["xl/workbook.xml"] = MacintoshCyrillicCharset
	f.workbookReader()
}

func TestWorkSheetReader(t *testing.T) {
	// Test unsupport charset.
	f := NewFile()
	delete(f.Sheet, "xl/worksheets/sheet1.xml")
	f.XLSX["xl/worksheets/sheet1.xml"] = MacintoshCyrillicCharset
	_, err := f.workSheetReader("Sheet1")
	assert.EqualError(t, err, "xml decode error: XML syntax error on line 1: invalid UTF-8")

	// Test on no checked worksheet.
	f = NewFile()
	delete(f.Sheet, "xl/worksheets/sheet1.xml")
	f.XLSX["xl/worksheets/sheet1.xml"] = []byte(`<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main"><sheetData/></worksheet>`)
	f.checked = nil
	_, err = f.workSheetReader("Sheet1")
	assert.NoError(t, err)
}

func TestRelsReader(t *testing.T) {
	// Test unsupport charset.
	f := NewFile()
	rels := "xl/_rels/workbook.xml.rels"
	f.Relationships[rels] = nil
	f.XLSX[rels] = MacintoshCyrillicCharset
	f.relsReader(rels)
}

func TestDeleteSheetFromWorkbookRels(t *testing.T) {
	f := NewFile()
	rels := "xl/_rels/workbook.xml.rels"
	f.Relationships[rels] = nil
	assert.Equal(t, f.deleteSheetFromWorkbookRels("rID"), "")
}

func TestAttrValToInt(t *testing.T) {
	_, err := attrValToInt("r", []xml.Attr{
		{Name: xml.Name{Local: "r"}, Value: "s"}})
	assert.EqualError(t, err, `strconv.Atoi: parsing "s": invalid syntax`)
}

V
Veniamin Albaev 已提交
1291
func prepareTestBook1() (*File, error) {
1292
	f, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1293 1294 1295 1296
	if err != nil {
		return nil, err
	}

1297
	err = f.AddPicture("Sheet2", "I9", filepath.Join("test", "images", "excel.jpg"),
V
Veniamin Albaev 已提交
1298 1299 1300 1301 1302 1303
		`{"x_offset": 140, "y_offset": 120, "hyperlink": "#Sheet2!D8", "hyperlink_type": "Location"}`)
	if err != nil {
		return nil, err
	}

	// Test add picture to worksheet with offset, external hyperlink and positioning.
1304
	err = f.AddPicture("Sheet1", "F21", filepath.Join("test", "images", "excel.png"),
V
Veniamin Albaev 已提交
1305 1306 1307 1308 1309
		`{"x_offset": 10, "y_offset": 10, "hyperlink": "https://github.com/360EntSecGroup-Skylar/excelize", "hyperlink_type": "External", "positioning": "oneCell"}`)
	if err != nil {
		return nil, err
	}

1310
	file, err := ioutil.ReadFile(filepath.Join("test", "images", "excel.jpg"))
V
Veniamin Albaev 已提交
1311 1312 1313 1314
	if err != nil {
		return nil, err
	}

1315
	err = f.AddPictureFromBytes("Sheet1", "Q1", "", "Excel Logo", ".jpg", file)
V
Veniamin Albaev 已提交
1316 1317 1318 1319
	if err != nil {
		return nil, err
	}

1320
	return f, nil
V
Veniamin Albaev 已提交
1321 1322 1323
}

func prepareTestBook3() (*File, error) {
1324 1325 1326 1327 1328 1329 1330 1331 1332
	f := NewFile()
	f.NewSheet("Sheet1")
	f.NewSheet("XLSXSheet2")
	f.NewSheet("XLSXSheet3")
	f.SetCellInt("XLSXSheet2", "A23", 56)
	f.SetCellStr("Sheet1", "B20", "42")
	f.SetActiveSheet(0)

	err := f.AddPicture("Sheet1", "H2", filepath.Join("test", "images", "excel.gif"),
1333
		`{"x_scale": 0.5, "y_scale": 0.5, "positioning": "absolute"}`)
V
Veniamin Albaev 已提交
1334 1335 1336 1337
	if err != nil {
		return nil, err
	}

1338
	err = f.AddPicture("Sheet1", "C2", filepath.Join("test", "images", "excel.png"), "")
V
Veniamin Albaev 已提交
1339 1340 1341 1342
	if err != nil {
		return nil, err
	}

1343
	return f, nil
V
Veniamin Albaev 已提交
1344 1345 1346
}

func prepareTestBook4() (*File, error) {
1347 1348 1349 1350 1351
	f := NewFile()
	f.SetColWidth("Sheet1", "B", "A", 12)
	f.SetColWidth("Sheet1", "A", "B", 12)
	f.GetColWidth("Sheet1", "A")
	f.GetColWidth("Sheet1", "C")
V
Veniamin Albaev 已提交
1352

1353
	return f, nil
V
Veniamin Albaev 已提交
1354
}
1355

1356
func fillCells(f *File, sheet string, colCount, rowCount int) {
1357 1358
	for col := 1; col <= colCount; col++ {
		for row := 1; row <= rowCount; row++ {
1359
			cell, _ := CoordinatesToCellName(col, row)
1360
			f.SetCellStr(sheet, cell, cell)
1361 1362 1363
		}
	}
}
1364 1365 1366 1367 1368 1369

func BenchmarkOpenFile(b *testing.B) {
	for i := 0; i < b.N; i++ {
		OpenFile(filepath.Join("test", "Book1.xlsx"))
	}
}