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

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

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

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

27
	// Test get all the rows in a not exists worksheet.
xurime's avatar
xurime 已提交
28
	xlsx.GetRows("Sheet4")
29
	// Test get all the rows in a worksheet.
30 31
	rows, err := xlsx.GetRows("Sheet2")
	assert.NoError(t, err)
32 33 34 35 36 37
	for _, row := range rows {
		for _, cell := range row {
			t.Log(cell, "\t")
		}
		t.Log("\r\n")
	}
38
	xlsx.UpdateLinkedValue()
39

40 41
	xlsx.SetCellDefault("Sheet2", "A1", strconv.FormatFloat(float64(100.1588), 'f', -1, 32))
	xlsx.SetCellDefault("Sheet2", "A1", strconv.FormatFloat(float64(-100.1588), 'f', -1, 64))
42

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

47
	xlsx.SetCellInt("Sheet2", "A1", 100)
48

49
	// Test set cell integer value with illegal row number.
50
	assert.EqualError(t, xlsx.SetCellInt("Sheet2", "A", 100), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
51

52
	xlsx.SetCellStr("Sheet2", "C11", "Knowns")
53
	// Test max characters in a cell.
54 55 56
	xlsx.SetCellStr("Sheet2", "D11", strings.Repeat("c", 32769))
	xlsx.NewSheet(":\\/?*[]Maximum 31 characters allowed in sheet title.")
	// Test set worksheet name with illegal name.
57 58
	xlsx.SetSheetName("Maximum 31 characters allowed i", "[Rename]:\\/?* Maximum 31 characters allowed in sheet title.")
	xlsx.SetCellInt("Sheet3", "A23", 10)
59
	xlsx.SetCellStr("Sheet3", "b230", "10")
xurime's avatar
xurime 已提交
60
	assert.EqualError(t, xlsx.SetCellStr("Sheet10", "b230", "10"), "sheet Sheet10 is not exist")
61

62
	// Test set cell string value with illegal row number.
xurime's avatar
xurime 已提交
63
	assert.EqualError(t, xlsx.SetCellStr("Sheet1", "A", "10"), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
64

65
	xlsx.SetActiveSheet(2)
66
	// Test get cell formula with given rows number.
67 68
	_, err = xlsx.GetCellFormula("Sheet1", "B19")
	assert.NoError(t, err)
69
	// Test get cell formula with illegal worksheet name.
70 71 72 73
	_, err = xlsx.GetCellFormula("Sheet2", "B20")
	assert.NoError(t, err)
	_, err = xlsx.GetCellFormula("Sheet1", "B20")
	assert.NoError(t, err)
74 75

	// Test get cell formula with illegal rows number.
76 77
	_, err = xlsx.GetCellFormula("Sheet1", "B")
	assert.EqualError(t, err, `cannot convert cell "B" to coordinates: invalid cell name "B"`)
78 79 80 81
	// Test get shared cell formula
	xlsx.GetCellFormula("Sheet2", "H11")
	xlsx.GetCellFormula("Sheet2", "I11")
	getSharedForumula(&xlsxWorksheet{}, "")
82

xurime's avatar
xurime 已提交
83
	// Test read cell value with given illegal rows number.
84 85 86 87
	_, err = xlsx.GetCellValue("Sheet2", "a-1")
	assert.EqualError(t, err, `cannot convert cell "A-1" to coordinates: invalid cell name "A-1"`)
	_, err = xlsx.GetCellValue("Sheet2", "A")
	assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
88

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

	// Test on not exists worksheet.
	assert.EqualError(t, xlsx.SetCellDefault("SheetN", "A1", ""), "sheet SheetN is not exist")
	assert.EqualError(t, xlsx.SetCellFloat("SheetN", "A1", 42.65418, 2, 32), "sheet SheetN is not exist")
	assert.EqualError(t, xlsx.SetCellBool("SheetN", "A1", true), "sheet SheetN is not exist")
	assert.EqualError(t, xlsx.SetCellFormula("SheetN", "A1", ""), "sheet SheetN is not exist")
	assert.EqualError(t, xlsx.SetCellHyperLink("SheetN", "A1", "Sheet1!A40", "Location"), "sheet SheetN is not exist")
120

121 122 123 124 125 126 127 128 129 130
	// Test boolean write
	booltest := []struct {
		value    bool
		expected string
	}{
		{false, "0"},
		{true, "1"},
	}
	for _, test := range booltest {
		xlsx.SetCellValue("Sheet2", "F16", test.value)
131 132 133
		val, err := xlsx.GetCellValue("Sheet2", "F16")
		assert.NoError(t, err)
		assert.Equal(t, test.expected, val)
134
	}
135

136
	xlsx.SetCellValue("Sheet2", "G2", nil)
137

138
	assert.EqualError(t, xlsx.SetCellValue("Sheet2", "G4", time.Now()), "only UTC time expected")
139 140

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

xurime's avatar
xurime 已提交
161
func TestSaveFile(t *testing.T) {
162
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
xurime's avatar
xurime 已提交
163 164 165
	if !assert.NoError(t, err) {
		t.FailNow()
	}
166 167
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSaveFile.xlsx")))
	xlsx, err = OpenFile(filepath.Join("test", "TestSaveFile.xlsx"))
xurime's avatar
xurime 已提交
168 169 170 171 172 173
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	assert.NoError(t, xlsx.Save())
}

V
Veniamin Albaev 已提交
174
func TestSaveAsWrongPath(t *testing.T) {
175
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
176 177 178 179 180 181
	if assert.NoError(t, err) {
		// Test write file to not exist directory.
		err = xlsx.SaveAs("")
		if assert.Error(t, err) {
			assert.True(t, os.IsNotExist(err), "Error: %v: Expected os.IsNotExists(err) == true", err)
		}
xurime's avatar
xurime 已提交
182
	}
183 184
}

185
func TestBrokenFile(t *testing.T) {
xurime's avatar
xurime 已提交
186
	// Test write file with broken file struct.
187
	xlsx := File{}
xurime's avatar
xurime 已提交
188

V
Veniamin Albaev 已提交
189 190 191
	t.Run("SaveWithoutName", func(t *testing.T) {
		assert.EqualError(t, xlsx.Save(), "no path defined for file, consider File.WriteTo or File.Write")
	})
192

V
Veniamin Albaev 已提交
193 194
	t.Run("SaveAsEmptyStruct", func(t *testing.T) {
		// Test write file with broken file struct with given path.
195
		assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestBrokenFile.SaveAsEmptyStruct.xlsx")))
V
Veniamin Albaev 已提交
196 197 198 199
	})

	t.Run("OpenBadWorkbook", func(t *testing.T) {
		// Test set active sheet without BookViews and Sheets maps in xl/workbook.xml.
200
		f3, err := OpenFile(filepath.Join("test", "BadWorkbook.xlsx"))
V
Veniamin Albaev 已提交
201 202 203 204 205 206 207
		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.
208
		_, err := OpenFile(filepath.Join("test", "NotExistsFile.xlsx"))
V
Veniamin Albaev 已提交
209 210 211 212
		if assert.Error(t, err) {
			assert.True(t, os.IsNotExist(err), "Expected os.IsNotExists(err) == true")
		}
	})
213 214
}

215
func TestNewFile(t *testing.T) {
216
	// Test create a XLSX file.
217
	xlsx := NewFile()
218
	xlsx.NewSheet("Sheet1")
219 220 221 222
	xlsx.NewSheet("XLSXSheet2")
	xlsx.NewSheet("XLSXSheet3")
	xlsx.SetCellInt("XLSXSheet2", "A23", 56)
	xlsx.SetCellStr("Sheet1", "B20", "42")
223
	xlsx.SetActiveSheet(0)
V
Veniamin Albaev 已提交
224

225
	// Test add picture to sheet with scaling and positioning.
226 227
	err := xlsx.AddPicture("Sheet1", "H2", filepath.Join("test", "images", "excel.gif"),
		`{"x_scale": 0.5, "y_scale": 0.5, "positioning": "absolute"}`)
V
Veniamin Albaev 已提交
228 229
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
230
	}
V
Veniamin Albaev 已提交
231

232
	// Test add picture to worksheet without formatset.
233
	err = xlsx.AddPicture("Sheet1", "C2", filepath.Join("test", "images", "excel.png"), "")
V
Veniamin Albaev 已提交
234 235
	if !assert.NoError(t, err) {
		t.FailNow()
236
	}
V
Veniamin Albaev 已提交
237

238
	// Test add picture to worksheet with invalid formatset.
239
	err = xlsx.AddPicture("Sheet1", "C2", filepath.Join("test", "images", "excel.png"), `{`)
V
Veniamin Albaev 已提交
240 241
	if !assert.Error(t, err) {
		t.FailNow()
242
	}
V
Veniamin Albaev 已提交
243

244
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestNewFile.xlsx")))
xurime's avatar
xurime 已提交
245
}
246

247
func TestColWidth(t *testing.T) {
248
	xlsx := NewFile()
249 250 251 252
	xlsx.SetColWidth("Sheet1", "B", "A", 12)
	xlsx.SetColWidth("Sheet1", "A", "B", 12)
	xlsx.GetColWidth("Sheet1", "A")
	xlsx.GetColWidth("Sheet1", "C")
253 254 255 256 257 258 259

	// Test set and get column width with illegal cell coordinates.
	_, err := xlsx.GetColWidth("Sheet1", "*")
	assert.EqualError(t, err, `invalid column name "*"`)
	assert.EqualError(t, xlsx.SetColWidth("Sheet1", "*", "B", 1), `invalid column name "*"`)
	assert.EqualError(t, xlsx.SetColWidth("Sheet1", "A", "*", 1), `invalid column name "*"`)

xurime's avatar
xurime 已提交
260 261 262 263
	// Test get column width on not exists worksheet.
	_, err = xlsx.GetColWidth("SheetN", "A")
	assert.EqualError(t, err, "sheet SheetN is not exist")

264
	err = xlsx.SaveAs(filepath.Join("test", "TestColWidth.xlsx"))
265
	if err != nil {
266
		t.Error(err)
267
	}
268 269 270
	convertRowHeightToPixels(0)
}

271 272 273 274 275 276
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 "*"`)
}

277
func TestSetCellHyperLink(t *testing.T) {
278
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
279 280 281 282
	if err != nil {
		t.Log(err)
	}
	// Test set cell hyperlink in a work sheet already have hyperlinks.
283
	assert.NoError(t, xlsx.SetCellHyperLink("Sheet1", "B19", "https://github.com/360EntSecGroup-Skylar/excelize", "External"))
284
	// Test add first hyperlink in a work sheet.
285
	assert.NoError(t, xlsx.SetCellHyperLink("Sheet2", "C1", "https://github.com/360EntSecGroup-Skylar/excelize", "External"))
286
	// Test add Location hyperlink in a work sheet.
287
	assert.NoError(t, xlsx.SetCellHyperLink("Sheet2", "D6", "Sheet1!D8", "Location"))
288

289
	assert.EqualError(t, xlsx.SetCellHyperLink("Sheet2", "C3", "Sheet1!D8", ""), `invalid link type ""`)
290

291
	assert.EqualError(t, xlsx.SetCellHyperLink("Sheet2", "", "Sheet1!D60", "Location"), `invalid cell name ""`)
292

293
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellHyperLink.xlsx")))
294 295
}

296
func TestGetCellHyperLink(t *testing.T) {
297
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
298 299
	if !assert.NoError(t, err) {
		t.FailNow()
300
	}
V
Veniamin Albaev 已提交
301

xurime's avatar
xurime 已提交
302
	_, _, err = xlsx.GetCellHyperLink("Sheet1", "")
303
	assert.EqualError(t, err, `invalid cell name ""`)
304

xurime's avatar
xurime 已提交
305
	link, target, err := xlsx.GetCellHyperLink("Sheet1", "A22")
306
	assert.NoError(t, err)
307
	t.Log(link, target)
308 309
	link, target, err = xlsx.GetCellHyperLink("Sheet2", "D6")
	assert.NoError(t, err)
310
	t.Log(link, target)
311
	link, target, err = xlsx.GetCellHyperLink("Sheet3", "H3")
xurime's avatar
xurime 已提交
312
	assert.EqualError(t, err, "sheet Sheet3 is not exist")
313 314 315
	t.Log(link, target)
}

316
func TestSetCellFormula(t *testing.T) {
317
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
318 319
	if !assert.NoError(t, err) {
		t.FailNow()
320
	}
V
Veniamin Albaev 已提交
321

322 323
	xlsx.SetCellFormula("Sheet1", "B19", "SUM(Sheet2!D2,Sheet2!D11)")
	xlsx.SetCellFormula("Sheet1", "C19", "SUM(Sheet2!D2,Sheet2!D9)")
324

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

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

330 331 332 333 334 335 336 337 338 339
	xlsx, err = OpenFile(filepath.Join("test", "CalcChain.xlsx"))
	if !assert.NoError(t, err) {
		t.FailNow()
	}
	// Test remove cell formula.
	xlsx.SetCellFormula("Sheet1", "A1", "")
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellFormula2.xlsx")))
	// Test remove all cell formula.
	xlsx.SetCellFormula("Sheet1", "B1", "")
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellFormula3.xlsx")))
340
}
341 342

func TestSetSheetBackground(t *testing.T) {
343
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
344 345
	if !assert.NoError(t, err) {
		t.FailNow()
346
	}
V
Veniamin Albaev 已提交
347

348
	err = xlsx.SetSheetBackground("Sheet2", filepath.Join("test", "images", "background.jpg"))
V
Veniamin Albaev 已提交
349 350
	if !assert.NoError(t, err) {
		t.FailNow()
351
	}
V
Veniamin Albaev 已提交
352

353
	err = xlsx.SetSheetBackground("Sheet2", filepath.Join("test", "images", "background.jpg"))
V
Veniamin Albaev 已提交
354 355
	if !assert.NoError(t, err) {
		t.FailNow()
356
	}
V
Veniamin Albaev 已提交
357

358
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetSheetBackground.xlsx")))
V
Veniamin Albaev 已提交
359 360 361
}

func TestSetSheetBackgroundErrors(t *testing.T) {
362
	xlsx, 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
	err = xlsx.SetSheetBackground("Sheet2", filepath.Join("test", "not_exists", "not_exists.png"))
V
Veniamin Albaev 已提交
368 369 370 371
	if assert.Error(t, err) {
		assert.True(t, os.IsNotExist(err), "Expected os.IsNotExists(err) == true")
	}

372
	err = xlsx.SetSheetBackground("Sheet2", filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
373
	assert.EqualError(t, err, "unsupported image extension")
374
}
xurime's avatar
xurime 已提交
375

376
func TestMergeCell(t *testing.T) {
377
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
378 379
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
380
	}
V
Veniamin Albaev 已提交
381

382 383 384 385 386 387 388 389 390 391 392
	xlsx.MergeCell("Sheet1", "D9", "D9")
	xlsx.MergeCell("Sheet1", "D9", "E9")
	xlsx.MergeCell("Sheet1", "H14", "G13")
	xlsx.MergeCell("Sheet1", "C9", "D8")
	xlsx.MergeCell("Sheet1", "F11", "G13")
	xlsx.MergeCell("Sheet1", "H7", "B15")
	xlsx.MergeCell("Sheet1", "D11", "F13")
	xlsx.MergeCell("Sheet1", "G10", "K12")
	xlsx.SetCellValue("Sheet1", "G11", "set value in merged cell")
	xlsx.SetCellInt("Sheet1", "H11", 100)
	xlsx.SetCellValue("Sheet1", "I11", float64(0.5))
393
	xlsx.SetCellHyperLink("Sheet1", "J11", "https://github.com/360EntSecGroup-Skylar/excelize", "External")
394 395
	xlsx.SetCellFormula("Sheet1", "G12", "SUM(Sheet1!B19,Sheet1!C19)")
	xlsx.GetCellValue("Sheet1", "H11")
396
	xlsx.GetCellValue("Sheet2", "A6") // Merged cell ref is single coordinate.
397
	xlsx.GetCellFormula("Sheet1", "G12")
V
Veniamin Albaev 已提交
398

399
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestMergeCell.xlsx")))
xurime's avatar
xurime 已提交
400
}
N
Nikolas Silva 已提交
401

S
sairoutine 已提交
402 403 404 405 406
func TestGetMergeCells(t *testing.T) {
	wants := []struct {
		value string
		start string
		end   string
407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423
	}{{
		value: "A1",
		start: "A1",
		end:   "B1",
	}, {
		value: "A2",
		start: "A2",
		end:   "A3",
	}, {
		value: "A4",
		start: "A4",
		end:   "B5",
	}, {
		value: "A7",
		start: "A7",
		end:   "C10",
	}}
S
sairoutine 已提交
424

425
	xlsx, err := OpenFile(filepath.Join("test", "MergeCell.xlsx"))
V
Veniamin Albaev 已提交
426 427
	if !assert.NoError(t, err) {
		t.FailNow()
S
sairoutine 已提交
428
	}
429
	sheet1 := xlsx.GetSheetName(1)
S
sairoutine 已提交
430

xurime's avatar
xurime 已提交
431
	mergeCells, err := xlsx.GetMergeCells(sheet1)
432 433
	if !assert.Len(t, mergeCells, len(wants)) {
		t.FailNow()
S
sairoutine 已提交
434
	}
xurime's avatar
xurime 已提交
435
	assert.NoError(t, err)
S
sairoutine 已提交
436 437

	for i, m := range mergeCells {
438 439 440
		assert.Equal(t, wants[i].value, m.GetCellValue())
		assert.Equal(t, wants[i].start, m.GetStartAxis())
		assert.Equal(t, wants[i].end, m.GetEndAxis())
S
sairoutine 已提交
441
	}
xurime's avatar
xurime 已提交
442 443 444 445

	// Test get merged cells on not exists worksheet.
	_, err = xlsx.GetMergeCells("SheetN")
	assert.EqualError(t, err, "sheet SheetN is not exist")
S
sairoutine 已提交
446 447
}

448
func TestSetCellStyleAlignment(t *testing.T) {
V
Veniamin Albaev 已提交
449 450 451
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
452
	}
V
Veniamin Albaev 已提交
453

454 455
	var style int
	style, err = xlsx.NewStyle(`{"alignment":{"horizontal":"center","ident":1,"justify_last_line":true,"reading_order":0,"relative_indent":1,"shrink_to_fit":true,"text_rotation":45,"vertical":"top","wrap_text":true}}`)
V
Veniamin Albaev 已提交
456 457
	if !assert.NoError(t, err) {
		t.FailNow()
458
	}
V
Veniamin Albaev 已提交
459

460
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A22", "A22", style))
461

462
	// Test set cell style with given illegal rows number.
463 464
	assert.EqualError(t, xlsx.SetCellStyle("Sheet1", "A", "A22", style), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.EqualError(t, xlsx.SetCellStyle("Sheet1", "A22", "A", style), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
465

466
	// Test get cell style with given illegal rows number.
467 468 469
	index, err := xlsx.GetCellStyle("Sheet1", "A")
	assert.Equal(t, 0, index)
	assert.EqualError(t, err, `cannot convert cell "A" to coordinates: invalid cell name "A"`)
V
Veniamin Albaev 已提交
470

471
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleAlignment.xlsx")))
472 473 474
}

func TestSetCellStyleBorder(t *testing.T) {
V
Veniamin Albaev 已提交
475 476 477
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
478
	}
479

V
Veniamin Albaev 已提交
480
	var style int
481

482
	// Test set border on overlapping area with vertical variants shading styles gradient fill.
483
	style, err = xlsx.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 已提交
484 485
	if !assert.NoError(t, err) {
		t.FailNow()
486
	}
487
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "J21", "L25", style))
488 489

	style, err = xlsx.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 已提交
490 491
	if !assert.NoError(t, err) {
		t.FailNow()
492
	}
493
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "M28", "K24", style))
494 495

	style, err = xlsx.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 已提交
496 497
	if !assert.NoError(t, err) {
		t.FailNow()
498
	}
499
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "M28", "K24", style))
500

501
	// Test set border and solid style pattern fill for a single cell.
502
	style, err = xlsx.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 已提交
503 504
	if !assert.NoError(t, err) {
		t.FailNow()
505
	}
506

507
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "O22", "O22", style))
V
Veniamin Albaev 已提交
508

509
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleBorder.xlsx")))
V
Veniamin Albaev 已提交
510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527
}

func TestSetCellStyleBorderErrors(t *testing.T) {
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
	}

	// Set border with invalid style parameter.
	_, err = xlsx.NewStyle("")
	if !assert.EqualError(t, err, "unexpected end of JSON input") {
		t.FailNow()
	}

	// Set border with invalid style index number.
	_, err = xlsx.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}]}`)
	if !assert.NoError(t, err) {
		t.FailNow()
528 529 530
	}
}

531
func TestSetCellStyleNumberFormat(t *testing.T) {
V
Veniamin Albaev 已提交
532 533 534
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
535
	}
V
Veniamin Albaev 已提交
536

xurime's avatar
xurime 已提交
537
	// Test only set fill and number format for a cell.
538 539 540 541 542 543 544 545 546 547 548 549 550
	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 {
				xlsx.SetCellValue("Sheet2", c, v)
			} else {
				xlsx.SetCellValue("Sheet2", c, val)
			}
551
			style, err := xlsx.NewStyle(`{"fill":{"type":"gradient","color":["#FFFFFF","#E0EBF5"],"shading":5},"number_format": ` + strconv.Itoa(d) + `}`)
V
Veniamin Albaev 已提交
552 553
			if !assert.NoError(t, err) {
				t.FailNow()
554
			}
555
			assert.NoError(t, xlsx.SetCellStyle("Sheet2", c, c, style))
556 557 558
			t.Log(xlsx.GetCellValue("Sheet2", c))
		}
	}
559 560
	var style int
	style, err = xlsx.NewStyle(`{"number_format":-1}`)
V
Veniamin Albaev 已提交
561 562
	if !assert.NoError(t, err) {
		t.FailNow()
563
	}
564
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "L33", "L33", style))
V
Veniamin Albaev 已提交
565

566
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleNumberFormat.xlsx")))
567 568
}

569
func TestSetCellStyleCurrencyNumberFormat(t *testing.T) {
V
Veniamin Albaev 已提交
570 571 572 573 574
	t.Run("TestBook3", func(t *testing.T) {
		xlsx, err := prepareTestBook3()
		if !assert.NoError(t, err) {
			t.FailNow()
		}
575

V
Veniamin Albaev 已提交
576 577 578 579 580 581 582
		xlsx.SetCellValue("Sheet1", "A1", 56)
		xlsx.SetCellValue("Sheet1", "A2", -32.3)
		var style int
		style, err = xlsx.NewStyle(`{"number_format": 188, "decimal_places": -1}`)
		if !assert.NoError(t, err) {
			t.FailNow()
		}
583

584
		assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A1", "A1", style))
V
Veniamin Albaev 已提交
585 586 587 588
		style, err = xlsx.NewStyle(`{"number_format": 188, "decimal_places": 31, "negred": true}`)
		if !assert.NoError(t, err) {
			t.FailNow()
		}
589

590
		assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A2", "A2", style))
591

592
		assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleCurrencyNumberFormat.TestBook3.xlsx")))
V
Veniamin Albaev 已提交
593
	})
594

V
Veniamin Albaev 已提交
595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612
	t.Run("TestBook4", func(t *testing.T) {
		xlsx, err := prepareTestBook4()
		if !assert.NoError(t, err) {
			t.FailNow()
		}
		xlsx.SetCellValue("Sheet1", "A1", 42920.5)
		xlsx.SetCellValue("Sheet1", "A2", 42920.5)

		_, err = xlsx.NewStyle(`{"number_format": 26, "lang": "zh-tw"}`)
		if !assert.NoError(t, err) {
			t.FailNow()
		}

		style, err := xlsx.NewStyle(`{"number_format": 27}`)
		if !assert.NoError(t, err) {
			t.FailNow()
		}

613
		assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A1", "A1", style))
V
Veniamin Albaev 已提交
614 615 616 617 618
		style, err = xlsx.NewStyle(`{"number_format": 31, "lang": "ko-kr"}`)
		if !assert.NoError(t, err) {
			t.FailNow()
		}

619
		assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A2", "A2", style))
V
Veniamin Albaev 已提交
620 621 622 623 624

		style, err = xlsx.NewStyle(`{"number_format": 71, "lang": "th-th"}`)
		if !assert.NoError(t, err) {
			t.FailNow()
		}
625
		assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A2", "A2", style))
V
Veniamin Albaev 已提交
626

627
		assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleCurrencyNumberFormat.TestBook4.xlsx")))
V
Veniamin Albaev 已提交
628
	})
629 630
}

631 632 633 634 635 636 637 638
func TestSetCellStyleCustomNumberFormat(t *testing.T) {
	xlsx := NewFile()
	xlsx.SetCellValue("Sheet1", "A1", 42920.5)
	xlsx.SetCellValue("Sheet1", "A2", 42920.5)
	style, err := xlsx.NewStyle(`{"custom_number_format": "[$-380A]dddd\\,\\ dd\" de \"mmmm\" de \"yyyy;@"}`)
	if err != nil {
		t.Log(err)
	}
639
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A1", "A1", style))
640 641 642 643
	style, err = xlsx.NewStyle(`{"custom_number_format": "[$-380A]dddd\\,\\ dd\" de \"mmmm\" de \"yyyy;@"}`)
	if err != nil {
		t.Log(err)
	}
644
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "A2", "A2", style))
V
Veniamin Albaev 已提交
645

646
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleCustomNumberFormat.xlsx")))
647 648
}

649
func TestSetCellStyleFill(t *testing.T) {
V
Veniamin Albaev 已提交
650 651 652
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
653
	}
V
Veniamin Albaev 已提交
654

655
	var style int
656
	// Test set fill for cell with invalid parameter.
657
	style, err = xlsx.NewStyle(`{"fill":{"type":"gradient","color":["#FFFFFF","#E0EBF5"],"shading":6}}`)
V
Veniamin Albaev 已提交
658 659
	if !assert.NoError(t, err) {
		t.FailNow()
660
	}
661
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "O23", "O23", style))
662 663

	style, err = xlsx.NewStyle(`{"fill":{"type":"gradient","color":["#FFFFFF"],"shading":1}}`)
V
Veniamin Albaev 已提交
664 665
	if !assert.NoError(t, err) {
		t.FailNow()
666
	}
667
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "O23", "O23", style))
668 669

	style, err = xlsx.NewStyle(`{"fill":{"type":"pattern","color":[],"pattern":1}}`)
V
Veniamin Albaev 已提交
670 671
	if !assert.NoError(t, err) {
		t.FailNow()
672
	}
673
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "O23", "O23", style))
xurime's avatar
xurime 已提交
674

675
	style, err = xlsx.NewStyle(`{"fill":{"type":"pattern","color":["#E0EBF5"],"pattern":19}}`)
V
Veniamin Albaev 已提交
676 677
	if !assert.NoError(t, err) {
		t.FailNow()
678
	}
679
	assert.NoError(t, xlsx.SetCellStyle("Sheet1", "O23", "O23", style))
680

681
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleFill.xlsx")))
682 683 684
}

func TestSetCellStyleFont(t *testing.T) {
V
Veniamin Albaev 已提交
685 686 687
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
688
	}
V
Veniamin Albaev 已提交
689

690 691
	var style int
	style, err = xlsx.NewStyle(`{"font":{"bold":true,"italic":true,"family":"Berlin Sans FB Demi","size":36,"color":"#777777","underline":"single"}}`)
V
Veniamin Albaev 已提交
692 693
	if !assert.NoError(t, err) {
		t.FailNow()
694
	}
V
Veniamin Albaev 已提交
695

696
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "A1", "A1", style))
697 698

	style, err = xlsx.NewStyle(`{"font":{"italic":true,"underline":"double"}}`)
V
Veniamin Albaev 已提交
699 700
	if !assert.NoError(t, err) {
		t.FailNow()
701
	}
V
Veniamin Albaev 已提交
702

703
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "A2", "A2", style))
704 705

	style, err = xlsx.NewStyle(`{"font":{"bold":true}}`)
V
Veniamin Albaev 已提交
706 707
	if !assert.NoError(t, err) {
		t.FailNow()
708
	}
V
Veniamin Albaev 已提交
709

710
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "A3", "A3", style))
711 712

	style, err = xlsx.NewStyle(`{"font":{"bold":true,"family":"","size":0,"color":"","underline":""}}`)
V
Veniamin Albaev 已提交
713 714
	if !assert.NoError(t, err) {
		t.FailNow()
715
	}
V
Veniamin Albaev 已提交
716

717
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "A4", "A4", style))
718 719

	style, err = xlsx.NewStyle(`{"font":{"color":"#777777"}}`)
V
Veniamin Albaev 已提交
720 721
	if !assert.NoError(t, err) {
		t.FailNow()
722
	}
V
Veniamin Albaev 已提交
723

724
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "A5", "A5", style))
V
Veniamin Albaev 已提交
725

726
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleFont.xlsx")))
727
}
728

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

735 736
	var style int
	style, err = xlsx.NewStyle(`{"protection":{"hidden":true, "locked":true}}`)
V
Veniamin Albaev 已提交
737 738
	if !assert.NoError(t, err) {
		t.FailNow()
739
	}
V
Veniamin Albaev 已提交
740

741
	assert.NoError(t, xlsx.SetCellStyle("Sheet2", "A6", "A6", style))
742
	err = xlsx.SaveAs(filepath.Join("test", "TestSetCellStyleProtection.xlsx"))
V
Veniamin Albaev 已提交
743 744
	if !assert.NoError(t, err) {
		t.FailNow()
745 746 747
	}
}

V
Veniamin Albaev 已提交
748 749 750 751 752 753 754 755
func TestSetDeleteSheet(t *testing.T) {
	t.Run("TestBook3", func(t *testing.T) {
		xlsx, err := prepareTestBook3()
		if !assert.NoError(t, err) {
			t.FailNow()
		}

		xlsx.DeleteSheet("XLSXSheet3")
756
		assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetDeleteSheet.TestBook3.xlsx")))
V
Veniamin Albaev 已提交
757 758 759 760 761 762 763 764 765 766
	})

	t.Run("TestBook4", func(t *testing.T) {
		xlsx, err := prepareTestBook4()
		if !assert.NoError(t, err) {
			t.FailNow()
		}
		xlsx.DeleteSheet("Sheet1")
		xlsx.AddComment("Sheet1", "A1", "")
		xlsx.AddComment("Sheet1", "A1", `{"author":"Excelize: ","text":"This is a comment."}`)
767
		assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetDeleteSheet.TestBook4.xlsx")))
V
Veniamin Albaev 已提交
768
	})
769
}
770

771
func TestSheetVisibility(t *testing.T) {
V
Veniamin Albaev 已提交
772 773 774
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
775
	}
V
Veniamin Albaev 已提交
776

777 778 779 780
	xlsx.SetSheetVisible("Sheet2", false)
	xlsx.SetSheetVisible("Sheet1", false)
	xlsx.SetSheetVisible("Sheet1", true)
	xlsx.GetSheetVisible("Sheet1")
V
Veniamin Albaev 已提交
781

782
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSheetVisibility.xlsx")))
783 784
}

785
func TestColumnVisibility(t *testing.T) {
V
Veniamin Albaev 已提交
786 787 788 789 790 791
	t.Run("TestBook1", func(t *testing.T) {
		xlsx, err := prepareTestBook1()
		if !assert.NoError(t, err) {
			t.FailNow()
		}

792 793 794 795 796 797
		assert.NoError(t, xlsx.SetColVisible("Sheet1", "F", false))
		assert.NoError(t, xlsx.SetColVisible("Sheet1", "F", true))
		visible, err := xlsx.GetColVisible("Sheet1", "F")
		assert.Equal(t, true, visible)
		assert.NoError(t, err)

xurime's avatar
xurime 已提交
798 799 800 801
		// Test get column visiable on not exists worksheet.
		_, err = xlsx.GetColVisible("SheetN", "F")
		assert.EqualError(t, err, "sheet SheetN is not exist")

802 803 804 805 806
		// Test get column visiable with illegal cell coordinates.
		_, err = xlsx.GetColVisible("Sheet1", "*")
		assert.EqualError(t, err, `invalid column name "*"`)
		assert.EqualError(t, xlsx.SetColVisible("Sheet1", "*", false), `invalid column name "*"`)

xurime's avatar
xurime 已提交
807 808 809 810
		xlsx.NewSheet("Sheet3")
		assert.NoError(t, xlsx.SetColVisible("Sheet3", "E", false))

		assert.EqualError(t, xlsx.SetColVisible("SheetN", "E", false), "sheet SheetN is not exist")
811
		assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestColumnVisibility.xlsx")))
V
Veniamin Albaev 已提交
812 813 814 815 816 817 818 819 820
	})

	t.Run("TestBook3", func(t *testing.T) {
		xlsx, err := prepareTestBook3()
		if !assert.NoError(t, err) {
			t.FailNow()
		}
		xlsx.GetColVisible("Sheet1", "B")
	})
821 822
}

823
func TestCopySheet(t *testing.T) {
V
Veniamin Albaev 已提交
824 825 826
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
827
	}
V
Veniamin Albaev 已提交
828

829
	idx := xlsx.NewSheet("CopySheet")
xurime's avatar
xurime 已提交
830
	assert.EqualError(t, xlsx.CopySheet(1, idx), "sheet sheet1 is not exist")
V
Veniamin Albaev 已提交
831

832
	xlsx.SetCellValue("Sheet4", "F1", "Hello")
833 834 835
	val, err := xlsx.GetCellValue("Sheet1", "F1")
	assert.NoError(t, err)
	assert.NotEqual(t, "Hello", val)
V
Veniamin Albaev 已提交
836

837
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestCopySheet.xlsx")))
V
Veniamin Albaev 已提交
838 839 840 841 842 843
}

func TestCopySheetError(t *testing.T) {
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
844
	}
V
Veniamin Albaev 已提交
845 846 847 848

	err = xlsx.CopySheet(0, -1)
	if !assert.EqualError(t, err, "invalid worksheet index") {
		t.FailNow()
849
	}
V
Veniamin Albaev 已提交
850

851
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestCopySheetError.xlsx")))
852
}
853

xurime's avatar
xurime 已提交
854
func TestAddTable(t *testing.T) {
V
Veniamin Albaev 已提交
855 856 857
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
858
	}
V
Veniamin Albaev 已提交
859

860
	err = xlsx.AddTable("Sheet1", "B26", "A21", `{}`)
V
Veniamin Albaev 已提交
861 862
	if !assert.NoError(t, err) {
		t.FailNow()
863
	}
V
Veniamin Albaev 已提交
864

865
	err = xlsx.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 已提交
866 867
	if !assert.NoError(t, err) {
		t.FailNow()
868
	}
V
Veniamin Albaev 已提交
869

870
	err = xlsx.AddTable("Sheet2", "F1", "F1", `{"table_style":"TableStyleMedium8"}`)
V
Veniamin Albaev 已提交
871 872
	if !assert.NoError(t, err) {
		t.FailNow()
xurime's avatar
xurime 已提交
873
	}
V
Veniamin Albaev 已提交
874

875 876 877 878 879 880
	// Test add table with illegal formatset.
	assert.EqualError(t, xlsx.AddTable("Sheet1", "B26", "A21", `{x}`), "invalid character 'x' looking for beginning of object key string")
	// Test add table with illegal cell coordinates.
	assert.EqualError(t, xlsx.AddTable("Sheet1", "A", "B1", `{}`), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.EqualError(t, xlsx.AddTable("Sheet1", "A1", "B", `{}`), `cannot convert cell "B" to coordinates: invalid cell name "B"`)

881
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestAddTable.xlsx")))
882 883 884 885 886

	// Test addTable with illegal cell coordinates.
	f := NewFile()
	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 已提交
887 888
}

889
func TestAddShape(t *testing.T) {
V
Veniamin Albaev 已提交
890 891 892
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
893
	}
V
Veniamin Albaev 已提交
894

895 896 897 898
	xlsx.AddShape("Sheet1", "A30", `{"type":"rect","paragraph":[{"text":"Rectangle","font":{"color":"CD5C5C"}},{"text":"Shape","font":{"bold":true,"color":"2980B9"}}]}`)
	xlsx.AddShape("Sheet1", "B30", `{"type":"rect","paragraph":[{"text":"Rectangle"},{}]}`)
	xlsx.AddShape("Sheet1", "C30", `{"type":"rect","paragraph":[]}`)
	xlsx.AddShape("Sheet3", "H1", `{"type":"ellipseRibbon", "color":{"line":"#4286f4","fill":"#8eb9ff"}, "paragraph":[{"font":{"bold":true,"italic":true,"family":"Berlin Sans FB Demi","size":36,"color":"#777777","underline":"single"}}], "height": 90}`)
899
	xlsx.AddShape("Sheet3", "H1", "")
V
Veniamin Albaev 已提交
900

901
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestAddShape.xlsx")))
902 903
}

904
func TestAddComments(t *testing.T) {
V
Veniamin Albaev 已提交
905 906 907
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
908
	}
V
Veniamin Albaev 已提交
909

xurime's avatar
xurime 已提交
910
	s := strings.Repeat("c", 32768)
911 912
	xlsx.AddComment("Sheet1", "A30", `{"author":"`+s+`","text":"`+s+`"}`)
	xlsx.AddComment("Sheet2", "B7", `{"author":"Excelize: ","text":"This is a comment."}`)
R
Rad Cirskis 已提交
913

914
	if assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestAddComments.xlsx"))) {
V
Veniamin Albaev 已提交
915 916
		assert.Len(t, xlsx.GetComments(), 2)
	}
917 918
}

919 920 921 922 923
func TestGetSheetComments(t *testing.T) {
	f := NewFile()
	assert.Equal(t, "", f.getSheetComments(0))
}

924
func TestAutoFilter(t *testing.T) {
925 926
	outFile := filepath.Join("test", "TestAutoFilter%d.xlsx")

V
Veniamin Albaev 已提交
927 928 929
	xlsx, err := prepareTestBook1()
	if !assert.NoError(t, err) {
		t.FailNow()
930
	}
V
Veniamin Albaev 已提交
931 932 933

	formats := []string{
		``,
934 935 936 937 938 939 940
		`{"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 已提交
941 942 943 944
	}

	for i, format := range formats {
		t.Run(fmt.Sprintf("Expression%d", i+1), func(t *testing.T) {
xurime's avatar
xurime 已提交
945 946
			err = xlsx.AutoFilter("Sheet1", "D4", "B1", format)
			assert.NoError(t, err)
xurime's avatar
xurime 已提交
947
			assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, i+1)))
V
Veniamin Albaev 已提交
948 949 950
		})
	}

951 952 953
	// testing AutoFilter with illegal cell coordinates.
	assert.EqualError(t, xlsx.AutoFilter("Sheet1", "A", "B1", ""), `cannot convert cell "A" to coordinates: invalid cell name "A"`)
	assert.EqualError(t, xlsx.AutoFilter("Sheet1", "A1", "B", ""), `cannot convert cell "B" to coordinates: invalid cell name "B"`)
V
Veniamin Albaev 已提交
954 955 956
}

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

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

	formats := []string{
965 966 967 968 969 970 971
		`{"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 已提交
972 973 974 975
	for i, format := range formats {
		t.Run(fmt.Sprintf("Expression%d", i+1), func(t *testing.T) {
			err = xlsx.AutoFilter("Sheet3", "D4", "B1", format)
			if assert.Error(t, err) {
976
				assert.NoError(t, xlsx.SaveAs(fmt.Sprintf(outFile, i+1)))
V
Veniamin Albaev 已提交
977 978
			}
		})
979 980 981
	}
}

982
func TestAddChart(t *testing.T) {
983
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
984 985
	if !assert.NoError(t, err) {
		t.FailNow()
986
	}
V
Veniamin Albaev 已提交
987

xurime's avatar
xurime 已提交
988 989 990
	categories := map[string]string{"A30": "Small", "A31": "Normal", "A32": "Large", "B29": "Apple", "C29": "Orange", "D29": "Pear"}
	values := map[string]int{"B30": 2, "C30": 3, "D30": 3, "B31": 5, "C31": 2, "D31": 4, "B32": 6, "C32": 7, "D32": 8}
	for k, v := range categories {
xurime's avatar
xurime 已提交
991
		assert.NoError(t, xlsx.SetCellValue("Sheet1", k, v))
xurime's avatar
xurime 已提交
992 993
	}
	for k, v := range values {
xurime's avatar
xurime 已提交
994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019
		assert.NoError(t, xlsx.SetCellValue("Sheet1", k, v))
	}
	assert.EqualError(t, xlsx.AddChart("Sheet1", "P1", ""), "unexpected end of JSON input")

	// Test add chart on not exists worksheet.
	assert.EqualError(t, xlsx.AddChart("SheetN", "P1", "{}"), "sheet SheetN is not exist")

	assert.NoError(t, xlsx.AddChart("Sheet1", "P1", `{"type":"col","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Column Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet1", "X1", `{"type":"colStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Stacked Column Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet1", "P16", `{"type":"colPercentStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 100% Stacked Column Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet1", "X16", `{"type":"col3DClustered","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"bottom","show_legend_key":false},"title":{"name":"Fruit 3D Clustered Column Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet1", "P30", `{"type":"col3DStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D 100% Stacked Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet1", "X30", `{"type":"col3DPercentStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D 100% Stacked Column Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet1", "P45", `{"type":"col3D","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D Column Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "P1", `{"type":"radar","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"top_right","show_legend_key":false},"title":{"name":"Fruit Radar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"span"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "X1", `{"type":"scatter","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"bottom","show_legend_key":false},"title":{"name":"Fruit Scatter Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "P16", `{"type":"doughnut","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"right","show_legend_key":false},"title":{"name":"Fruit Doughnut Chart"},"plotarea":{"show_bubble_size":false,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":false,"show_val":false},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "X16", `{"type":"line","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"top","show_legend_key":false},"title":{"name":"Fruit Line Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "P32", `{"type":"pie3D","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"bottom","show_legend_key":false},"title":{"name":"Fruit 3D Pie Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":false,"show_val":false},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "X32", `{"type":"pie","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"bottom","show_legend_key":false},"title":{"name":"Fruit Pie Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":false,"show_val":false},"show_blanks_as":"gap"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "P48", `{"type":"bar","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Clustered Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "X48", `{"type":"barStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Stacked Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "P64", `{"type":"barPercentStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Stacked 100% Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "X64", `{"type":"bar3DClustered","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D Clustered Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "P80", `{"type":"bar3DStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D Stacked Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero","y_axis":{"maximum":7.5,"minimum":0.5}}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "X80", `{"type":"bar3DPercentStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D 100% Stacked Bar Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero","x_axis":{"reverse_order":true,"maximum":0,"minimum":0},"y_axis":{"reverse_order":true,"maximum":0,"minimum":0}}`))
1020
	// area series charts
xurime's avatar
xurime 已提交
1021 1022 1023 1024 1025 1026
	assert.NoError(t, xlsx.AddChart("Sheet2", "AF1", `{"type":"area","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Area Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "AN1", `{"type":"areaStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D Stacked Area Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "AF16", `{"type":"areaPercentStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 2D 100% Stacked Area Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "AN16", `{"type":"area3D","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D Area Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "AF32", `{"type":"area3DStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D Stacked Area Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
	assert.NoError(t, xlsx.AddChart("Sheet2", "AN32", `{"type":"area3DPercentStacked","series":[{"name":"Sheet1!$A$30","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$30:$D$30"},{"name":"Sheet1!$A$31","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$31:$D$31"},{"name":"Sheet1!$A$32","categories":"Sheet1!$B$29:$D$29","values":"Sheet1!$B$32:$D$32"}],"format":{"x_scale":1.0,"y_scale":1.0,"x_offset":15,"y_offset":10,"print_obj":true,"lock_aspect_ratio":false,"locked":false},"legend":{"position":"left","show_legend_key":false},"title":{"name":"Fruit 3D 100% Stacked Area Chart"},"plotarea":{"show_bubble_size":true,"show_cat_name":false,"show_leader_lines":false,"show_percent":true,"show_series_name":true,"show_val":true},"show_blanks_as":"zero"}`))
V
Veniamin Albaev 已提交
1027

1028
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestAddChart.xlsx")))
1029 1030 1031 1032
}

func TestInsertCol(t *testing.T) {
	xlsx := NewFile()
1033 1034 1035 1036 1037 1038 1039 1040
	sheet1 := xlsx.GetSheetName(1)

	fillCells(xlsx, sheet1, 10, 10)

	xlsx.SetCellHyperLink(sheet1, "A5", "https://github.com/360EntSecGroup-Skylar/excelize", "External")
	xlsx.MergeCell(sheet1, "A1", "C3")

	err := xlsx.AutoFilter(sheet1, "A2", "B2", `{"column":"B","expression":"x != blanks"}`)
V
Veniamin Albaev 已提交
1041 1042
	if !assert.NoError(t, err) {
		t.FailNow()
1043
	}
V
Veniamin Albaev 已提交
1044

1045
	assert.NoError(t, xlsx.InsertCol(sheet1, "A"))
V
Veniamin Albaev 已提交
1046

1047 1048 1049
	// Test insert column with illegal cell coordinates.
	assert.EqualError(t, xlsx.InsertCol("Sheet1", "*"), `invalid column name "*"`)

1050
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestInsertCol.xlsx")))
1051 1052 1053 1054
}

func TestRemoveCol(t *testing.T) {
	xlsx := NewFile()
1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
	sheet1 := xlsx.GetSheetName(1)

	fillCells(xlsx, sheet1, 10, 15)

	xlsx.SetCellHyperLink(sheet1, "A5", "https://github.com/360EntSecGroup-Skylar/excelize", "External")
	xlsx.SetCellHyperLink(sheet1, "C5", "https://github.com", "External")

	xlsx.MergeCell(sheet1, "A1", "B1")
	xlsx.MergeCell(sheet1, "A2", "B2")

1065 1066
	assert.NoError(t, xlsx.RemoveCol(sheet1, "A"))
	assert.NoError(t, xlsx.RemoveCol(sheet1, "A"))
V
Veniamin Albaev 已提交
1067

1068 1069 1070
	// Test remove column with illegal cell coordinates.
	assert.EqualError(t, xlsx.RemoveCol("Sheet1", "*"), `invalid column name "*"`)

xurime's avatar
xurime 已提交
1071 1072 1073
	// Test remove column on not exists worksheet.
	assert.EqualError(t, xlsx.RemoveCol("SheetN", "B"), "sheet SheetN is not exist")

1074
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestRemoveCol.xlsx")))
1075 1076
}

xurime's avatar
xurime 已提交
1077 1078 1079
func TestSetPane(t *testing.T) {
	xlsx := NewFile()
	xlsx.SetPanes("Sheet1", `{"freeze":false,"split":false}`)
1080 1081 1082 1083 1084 1085
	xlsx.NewSheet("Panes 2")
	xlsx.SetPanes("Panes 2", `{"freeze":true,"split":false,"x_split":1,"y_split":0,"top_left_cell":"B1","active_pane":"topRight","panes":[{"sqref":"K16","active_cell":"K16","pane":"topRight"}]}`)
	xlsx.NewSheet("Panes 3")
	xlsx.SetPanes("Panes 3", `{"freeze":false,"split":true,"x_split":3270,"y_split":1800,"top_left_cell":"N57","active_pane":"bottomLeft","panes":[{"sqref":"I36","active_cell":"I36"},{"sqref":"G33","active_cell":"G33","pane":"topRight"},{"sqref":"J60","active_cell":"J60","pane":"bottomLeft"},{"sqref":"O60","active_cell":"O60","pane":"bottomRight"}]}`)
	xlsx.NewSheet("Panes 4")
	xlsx.SetPanes("Panes 4", `{"freeze":true,"split":false,"x_split":0,"y_split":9,"top_left_cell":"A34","active_pane":"bottomLeft","panes":[{"sqref":"A11:XFD11","active_cell":"A11","pane":"bottomLeft"}]}`)
1086
	xlsx.SetPanes("Panes 4", "")
V
Veniamin Albaev 已提交
1087

1088
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetPane.xlsx")))
xurime's avatar
xurime 已提交
1089 1090
}

1091 1092
func TestConditionalFormat(t *testing.T) {
	xlsx := NewFile()
1093 1094 1095 1096
	sheet1 := xlsx.GetSheetName(1)

	fillCells(xlsx, sheet1, 10, 15)

1097 1098 1099 1100
	var format1, format2, format3 int
	var err error
	// Rose format for bad conditional.
	format1, err = xlsx.NewConditionalStyle(`{"font":{"color":"#9A0511"},"fill":{"type":"pattern","color":["#FEC7CE"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1101 1102 1103 1104
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1105 1106
	// Light yellow format for neutral conditional.
	format2, err = xlsx.NewConditionalStyle(`{"fill":{"type":"pattern","color":["#FEEAA0"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1107 1108 1109 1110
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1111 1112
	// Light green format for good conditional.
	format3, err = xlsx.NewConditionalStyle(`{"font":{"color":"#09600B"},"fill":{"type":"pattern","color":["#C7EECF"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1113 1114 1115 1116
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1117
	// Color scales: 2 color.
1118
	xlsx.SetConditionalFormat(sheet1, "A1:A10", `[{"type":"2_color_scale","criteria":"=","min_type":"min","max_type":"max","min_color":"#F8696B","max_color":"#63BE7B"}]`)
1119
	// Color scales: 3 color.
1120
	xlsx.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"}]`)
1121
	// Hightlight cells rules: between...
1122
	xlsx.SetConditionalFormat(sheet1, "C1:C10", fmt.Sprintf(`[{"type":"cell","criteria":"between","format":%d,"minimum":"6","maximum":"8"}]`, format1))
1123
	// Hightlight cells rules: Greater Than...
1124
	xlsx.SetConditionalFormat(sheet1, "D1:D10", fmt.Sprintf(`[{"type":"cell","criteria":">","format":%d,"value":"6"}]`, format3))
1125
	// Hightlight cells rules: Equal To...
1126
	xlsx.SetConditionalFormat(sheet1, "E1:E10", fmt.Sprintf(`[{"type":"top","criteria":"=","format":%d}]`, format3))
1127
	// Hightlight cells rules: Not Equal To...
1128
	xlsx.SetConditionalFormat(sheet1, "F1:F10", fmt.Sprintf(`[{"type":"unique","criteria":"=","format":%d}]`, format2))
1129
	// Hightlight cells rules: Duplicate Values...
1130
	xlsx.SetConditionalFormat(sheet1, "G1:G10", fmt.Sprintf(`[{"type":"duplicate","criteria":"=","format":%d}]`, format2))
1131
	// Top/Bottom rules: Top 10%.
1132
	xlsx.SetConditionalFormat(sheet1, "H1:H10", fmt.Sprintf(`[{"type":"top","criteria":"=","format":%d,"value":"6","percent":true}]`, format1))
1133
	// Top/Bottom rules: Above Average...
1134
	xlsx.SetConditionalFormat(sheet1, "I1:I10", fmt.Sprintf(`[{"type":"average","criteria":"=","format":%d, "above_average": true}]`, format3))
1135
	// Top/Bottom rules: Below Average...
1136
	xlsx.SetConditionalFormat(sheet1, "J1:J10", fmt.Sprintf(`[{"type":"average","criteria":"=","format":%d, "above_average": false}]`, format1))
1137
	// Data Bars: Gradient Fill.
1138
	xlsx.SetConditionalFormat(sheet1, "K1:K10", `[{"type":"data_bar", "criteria":"=", "min_type":"min","max_type":"max","bar_color":"#638EC6"}]`)
1139
	// Use a formula to determine which cells to format.
1140
	xlsx.SetConditionalFormat(sheet1, "L1:L10", fmt.Sprintf(`[{"type":"formula", "criteria":"L2<3", "format":%d}]`, format1))
1141
	// Test set invalid format set in conditional format
1142
	xlsx.SetConditionalFormat(sheet1, "L1:L10", "")
V
Veniamin Albaev 已提交
1143

1144
	err = xlsx.SaveAs(filepath.Join("test", "TestConditionalFormat.xlsx"))
V
Veniamin Albaev 已提交
1145 1146
	if !assert.NoError(t, err) {
		t.FailNow()
1147 1148 1149
	}

	// Set conditional format with illegal valid type.
1150
	xlsx.SetConditionalFormat(sheet1, "K1:K10", `[{"type":"", "criteria":"=", "min_type":"min","max_type":"max","bar_color":"#638EC6"}]`)
1151
	// Set conditional format with illegal criteria type.
1152
	xlsx.SetConditionalFormat(sheet1, "K1:K10", `[{"type":"data_bar", "criteria":"", "min_type":"min","max_type":"max","bar_color":"#638EC6"}]`)
V
Veniamin Albaev 已提交
1153 1154

	// Set conditional format with file without dxfs element shold not return error.
1155
	xlsx, err = OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1156 1157 1158 1159
	if !assert.NoError(t, err) {
		t.FailNow()
	}

1160
	_, err = xlsx.NewConditionalStyle(`{"font":{"color":"#9A0511"},"fill":{"type":"pattern","color":["#FEC7CE"],"pattern":1}}`)
V
Veniamin Albaev 已提交
1161 1162 1163
	if !assert.NoError(t, err) {
		t.FailNow()
	}
1164
}
xurime's avatar
xurime 已提交
1165

V
Veniamin Albaev 已提交
1166 1167
func TestConditionalFormatError(t *testing.T) {
	xlsx := NewFile()
1168 1169 1170
	sheet1 := xlsx.GetSheetName(1)

	fillCells(xlsx, sheet1, 10, 15)
V
Veniamin Albaev 已提交
1171 1172 1173 1174 1175

	// Set conditional format with illegal JSON string should return error
	_, err := xlsx.NewConditionalStyle("")
	if !assert.EqualError(t, err, "unexpected end of JSON input") {
		t.FailNow()
xurime's avatar
xurime 已提交
1176 1177
	}
}
1178 1179

func TestSharedStrings(t *testing.T) {
1180
	xlsx, err := OpenFile(filepath.Join("test", "SharedStrings.xlsx"))
V
Veniamin Albaev 已提交
1181 1182
	if !assert.NoError(t, err) {
		t.FailNow()
1183 1184 1185
	}
	xlsx.GetRows("Sheet1")
}
1186 1187

func TestSetSheetRow(t *testing.T) {
1188
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1189 1190
	if !assert.NoError(t, err) {
		t.FailNow()
1191
	}
V
Veniamin Albaev 已提交
1192

1193 1194
	xlsx.SetSheetRow("Sheet1", "B27", &[]interface{}{"cell", nil, int32(42), float64(42), time.Now().UTC()})

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

1198 1199
	assert.EqualError(t, xlsx.SetSheetRow("Sheet1", "B27", []interface{}{}), `pointer to slice expected`)
	assert.EqualError(t, xlsx.SetSheetRow("Sheet1", "B27", &xlsx), `pointer to slice expected`)
1200
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestSetSheetRow.xlsx")))
1201
}
1202

1203 1204 1205 1206 1207 1208 1209 1210
func TestOutlineLevel(t *testing.T) {
	xlsx := NewFile()
	xlsx.NewSheet("Sheet2")
	xlsx.SetColOutlineLevel("Sheet1", "D", 4)
	xlsx.GetColOutlineLevel("Sheet1", "D")
	xlsx.GetColOutlineLevel("Shee2", "A")
	xlsx.SetColWidth("Sheet2", "A", "D", 13)
	xlsx.SetColOutlineLevel("Sheet2", "B", 2)
1211 1212
	xlsx.SetRowOutlineLevel("Sheet1", 2, 250)

1213 1214
	// Test set and get column outline level with illegal cell coordinates.
	assert.EqualError(t, xlsx.SetColOutlineLevel("Sheet1", "*", 1), `invalid column name "*"`)
xurime's avatar
xurime 已提交
1215
	_, err := xlsx.GetColOutlineLevel("Sheet1", "*")
1216 1217
	assert.EqualError(t, err, `invalid column name "*"`)

xurime's avatar
xurime 已提交
1218 1219 1220
	// Test set column outline level on not exists worksheet.
	assert.EqualError(t, xlsx.SetColOutlineLevel("SheetN", "E", 2), "sheet SheetN is not exist")

1221
	assert.EqualError(t, xlsx.SetRowOutlineLevel("Sheet1", 0, 1), "invalid row number 0")
xurime's avatar
xurime 已提交
1222
	level, err := xlsx.GetRowOutlineLevel("Sheet1", 2)
1223 1224
	assert.NoError(t, err)
	assert.Equal(t, uint8(250), level)
1225

1226 1227
	_, err = xlsx.GetRowOutlineLevel("Sheet1", 0)
	assert.EqualError(t, err, `invalid row number 0`)
1228

1229 1230 1231 1232
	level, err = xlsx.GetRowOutlineLevel("Sheet1", 10)
	assert.NoError(t, err)
	assert.Equal(t, uint8(0), level)

1233
	err = xlsx.SaveAs(filepath.Join("test", "TestOutlineLevel.xlsx"))
V
Veniamin Albaev 已提交
1234 1235
	if !assert.NoError(t, err) {
		t.FailNow()
1236
	}
V
Veniamin Albaev 已提交
1237

1238
	xlsx, err = OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1239 1240
	if !assert.NoError(t, err) {
		t.FailNow()
1241 1242 1243 1244
	}
	xlsx.SetColOutlineLevel("Sheet2", "B", 2)
}

xurime's avatar
xurime 已提交
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
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))
}

1270
func TestSearchSheet(t *testing.T) {
1271
	xlsx, err := OpenFile(filepath.Join("test", "SharedStrings.xlsx"))
V
Veniamin Albaev 已提交
1272 1273
	if !assert.NoError(t, err) {
		t.FailNow()
1274
	}
V
Veniamin Albaev 已提交
1275

1276 1277 1278 1279 1280
	// Test search in a not exists worksheet.
	t.Log(xlsx.SearchSheet("Sheet4", ""))
	// Test search a not exists value.
	t.Log(xlsx.SearchSheet("Sheet1", "X"))
	t.Log(xlsx.SearchSheet("Sheet1", "A"))
1281 1282 1283
	// Test search the coordinates where the numerical value in the range of
	// "0-9" of Sheet1 is described by regular expression:
	t.Log(xlsx.SearchSheet("Sheet1", "[0-9]", true))
R
r-uchino 已提交
1284 1285
}

1286 1287 1288 1289 1290 1291 1292
func TestProtectSheet(t *testing.T) {
	xlsx := NewFile()
	xlsx.ProtectSheet("Sheet1", nil)
	xlsx.ProtectSheet("Sheet1", &FormatSheetProtection{
		Password:      "password",
		EditScenarios: false,
	})
V
Veniamin Albaev 已提交
1293

1294
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestProtectSheet.xlsx")))
xurime's avatar
xurime 已提交
1295 1296
	// Test protect not exists worksheet.
	assert.EqualError(t, xlsx.ProtectSheet("SheetN", nil), "sheet SheetN is not exist")
1297 1298
}

1299
func TestUnprotectSheet(t *testing.T) {
1300
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1301 1302
	if !assert.NoError(t, err) {
		t.FailNow()
1303
	}
xurime's avatar
xurime 已提交
1304 1305
	// Test unprotect not exists worksheet.
	assert.EqualError(t, xlsx.UnprotectSheet("SheetN"), "sheet SheetN is not exist")
V
Veniamin Albaev 已提交
1306

1307
	xlsx.UnprotectSheet("Sheet1")
1308
	assert.NoError(t, xlsx.SaveAs(filepath.Join("test", "TestUnprotectSheet.xlsx")))
1309
}
xurime's avatar
xurime 已提交
1310 1311 1312 1313 1314 1315

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")
}
1316

V
Veniamin Albaev 已提交
1317
func prepareTestBook1() (*File, error) {
1318
	xlsx, err := OpenFile(filepath.Join("test", "Book1.xlsx"))
V
Veniamin Albaev 已提交
1319 1320 1321 1322
	if err != nil {
		return nil, err
	}

1323
	err = xlsx.AddPicture("Sheet2", "I9", filepath.Join("test", "images", "excel.jpg"),
V
Veniamin Albaev 已提交
1324 1325 1326 1327 1328 1329
		`{"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.
1330
	err = xlsx.AddPicture("Sheet1", "F21", filepath.Join("test", "images", "excel.png"),
V
Veniamin Albaev 已提交
1331 1332 1333 1334 1335
		`{"x_offset": 10, "y_offset": 10, "hyperlink": "https://github.com/360EntSecGroup-Skylar/excelize", "hyperlink_type": "External", "positioning": "oneCell"}`)
	if err != nil {
		return nil, err
	}

1336
	file, err := ioutil.ReadFile(filepath.Join("test", "images", "excel.jpg"))
V
Veniamin Albaev 已提交
1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357
	if err != nil {
		return nil, err
	}

	err = xlsx.AddPictureFromBytes("Sheet1", "Q1", "", "Excel Logo", ".jpg", file)
	if err != nil {
		return nil, err
	}

	return xlsx, nil
}

func prepareTestBook3() (*File, error) {
	xlsx := NewFile()
	xlsx.NewSheet("Sheet1")
	xlsx.NewSheet("XLSXSheet2")
	xlsx.NewSheet("XLSXSheet3")
	xlsx.SetCellInt("XLSXSheet2", "A23", 56)
	xlsx.SetCellStr("Sheet1", "B20", "42")
	xlsx.SetActiveSheet(0)

1358 1359
	err := xlsx.AddPicture("Sheet1", "H2", filepath.Join("test", "images", "excel.gif"),
		`{"x_scale": 0.5, "y_scale": 0.5, "positioning": "absolute"}`)
V
Veniamin Albaev 已提交
1360 1361 1362 1363
	if err != nil {
		return nil, err
	}

1364
	err = xlsx.AddPicture("Sheet1", "C2", filepath.Join("test", "images", "excel.png"), "")
V
Veniamin Albaev 已提交
1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
	if err != nil {
		return nil, err
	}

	return xlsx, nil
}

func prepareTestBook4() (*File, error) {
	xlsx := NewFile()
	xlsx.SetColWidth("Sheet1", "B", "A", 12)
	xlsx.SetColWidth("Sheet1", "A", "B", 12)
	xlsx.GetColWidth("Sheet1", "A")
	xlsx.GetColWidth("Sheet1", "C")

	return xlsx, nil
}
1381 1382 1383 1384

func fillCells(xlsx *File, sheet string, colCount, rowCount int) {
	for col := 1; col <= colCount; col++ {
		for row := 1; row <= rowCount; row++ {
1385
			cell, _ := CoordinatesToCellName(col, row)
1386 1387 1388 1389
			xlsx.SetCellStr(sheet, cell, cell)
		}
	}
}