diff options
Diffstat (limited to 'datavalidation_test.go')
| -rw-r--r-- | datavalidation_test.go | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/datavalidation_test.go b/datavalidation_test.go index d9e060a..88625d1 100644 --- a/datavalidation_test.go +++ b/datavalidation_test.go @@ -32,6 +32,11 @@ func TestDataValidation(t *testing.T) { dvRange.SetError(DataValidationErrorStyleWarning, "error title", "error body") dvRange.SetError(DataValidationErrorStyleInformation, "error title", "error body") assert.NoError(t, f.AddDataValidation("Sheet1", dvRange)) + + dataValidations, err := f.GetDataValidations("Sheet1") + assert.NoError(t, err) + assert.Equal(t, len(dataValidations), 1) + assert.NoError(t, f.SaveAs(resultFile)) dvRange = NewDataValidation(true) @@ -39,6 +44,11 @@ func TestDataValidation(t *testing.T) { assert.NoError(t, dvRange.SetRange(10, 20, DataValidationTypeWhole, DataValidationOperatorGreaterThan)) dvRange.SetInput("input title", "input body") assert.NoError(t, f.AddDataValidation("Sheet1", dvRange)) + + dataValidations, err = f.GetDataValidations("Sheet1") + assert.NoError(t, err) + assert.Equal(t, len(dataValidations), 2) + assert.NoError(t, f.SaveAs(resultFile)) f.NewSheet("Sheet2") @@ -49,6 +59,12 @@ func TestDataValidation(t *testing.T) { assert.NoError(t, dvRange.SetRange("INDIRECT($A$2)", "INDIRECT($A$3)", DataValidationTypeWhole, DataValidationOperatorBetween)) dvRange.SetError(DataValidationErrorStyleStop, "error title", "error body") assert.NoError(t, f.AddDataValidation("Sheet2", dvRange)) + dataValidations, err = f.GetDataValidations("Sheet1") + assert.NoError(t, err) + assert.Equal(t, len(dataValidations), 2) + dataValidations, err = f.GetDataValidations("Sheet2") + assert.NoError(t, err) + assert.Equal(t, len(dataValidations), 1) dvRange = NewDataValidation(true) dvRange.Sqref = "A5:B6" @@ -67,7 +83,22 @@ func TestDataValidation(t *testing.T) { } assert.Equal(t, `<formula1>"A<,B>,C"",D ,E',F"</formula1>`, dvRange.Formula1) assert.NoError(t, f.AddDataValidation("Sheet1", dvRange)) + + dataValidations, err = f.GetDataValidations("Sheet1") + assert.NoError(t, err) + assert.Equal(t, len(dataValidations), 3) + + // Test get data validation on no exists worksheet + _, err = f.GetDataValidations("SheetN") + assert.EqualError(t, err, "sheet SheetN is not exist") + assert.NoError(t, f.SaveAs(resultFile)) + + // Test get data validation on a worksheet without data validation settings + f = NewFile() + dataValidations, err = f.GetDataValidations("Sheet1") + assert.NoError(t, err) + assert.Equal(t, []*DataValidation(nil), dataValidations) } func TestDataValidationError(t *testing.T) { |