summaryrefslogtreecommitdiff
path: root/datavalidation_test.go
blob: 82bc42f23f9be3a011c24705ad9814b8ecabb5d7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// Copyright 2016 - 2019 The excelize Authors. All rights reserved. Use of
// this source code is governed by a BSD-style license that can be found in
// the LICENSE file.
//
// Package excelize providing a set of functions that allow you to write to
// and read from XLSX files. Support reads and writes XLSX file generated by
// Microsoft Excelâ„¢ 2007 and later. Support save file without losing original
// charts of XLSX. This library needs Go version 1.8 or later.

package excelize

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestDataValidation(t *testing.T) {
	const resultFile = "./test/TestDataValidation.xlsx"

	xlsx := NewFile()

	dvRange := NewDataValidation(true)
	dvRange.Sqref = "A1:B2"
	dvRange.SetRange(10, 20, DataValidationTypeWhole, DataValidationOperatorBetween)
	dvRange.SetError(DataValidationErrorStyleStop, "error title", "error body")
	dvRange.SetError(DataValidationErrorStyleWarning, "error title", "error body")
	dvRange.SetError(DataValidationErrorStyleInformation, "error title", "error body")
	xlsx.AddDataValidation("Sheet1", dvRange)
	if !assert.NoError(t, xlsx.SaveAs(resultFile)) {
		t.FailNow()
	}

	dvRange = NewDataValidation(true)
	dvRange.Sqref = "A3:B4"
	dvRange.SetRange(10, 20, DataValidationTypeWhole, DataValidationOperatorGreaterThan)
	dvRange.SetInput("input title", "input body")
	xlsx.AddDataValidation("Sheet1", dvRange)
	if !assert.NoError(t, xlsx.SaveAs(resultFile)) {
		t.FailNow()
	}

	dvRange = NewDataValidation(true)
	dvRange.Sqref = "A5:B6"
	dvRange.SetDropList([]string{"1", "2", "3"})
	xlsx.AddDataValidation("Sheet1", dvRange)
	if !assert.NoError(t, xlsx.SaveAs(resultFile)) {
		t.FailNow()
	}
}

func TestDataValidationError(t *testing.T) {
	const resultFile = "./test/TestDataValidationError.xlsx"

	xlsx := NewFile()
	xlsx.SetCellStr("Sheet1", "E1", "E1")
	xlsx.SetCellStr("Sheet1", "E2", "E2")
	xlsx.SetCellStr("Sheet1", "E3", "E3")

	dvRange := NewDataValidation(true)
	dvRange.SetSqref("A7:B8")
	dvRange.SetSqref("A7:B8")
	dvRange.SetSqrefDropList("$E$1:$E$3", true)

	err := dvRange.SetSqrefDropList("$E$1:$E$3", false)
	assert.EqualError(t, err, "cross-sheet sqref cell are not supported")

	xlsx.AddDataValidation("Sheet1", dvRange)
	if !assert.NoError(t, xlsx.SaveAs(resultFile)) {
		t.FailNow()
	}

	dvRange = NewDataValidation(true)
	dvRange.SetDropList(make([]string, 258))

	err = dvRange.SetRange(10, 20, DataValidationTypeWhole, DataValidationOperatorGreaterThan)
	assert.EqualError(t, err, "data validation must be 0-255 characters")

	xlsx.AddDataValidation("Sheet1", dvRange)
	if !assert.NoError(t, xlsx.SaveAs(resultFile)) {
		t.FailNow()
	}
}