123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273 |
- package excelize
- import (
- "encoding/xml"
- "fmt"
- "strconv"
- "strings"
- "testing"
- "github.com/stretchr/testify/assert"
- )
- var validColumns = []struct {
- Name string
- Num int
- }{
- {Name: "A", Num: 1},
- {Name: "Z", Num: 26},
- {Name: "AA", Num: 26 + 1},
- {Name: "AK", Num: 26 + 11},
- {Name: "ak", Num: 26 + 11},
- {Name: "Ak", Num: 26 + 11},
- {Name: "aK", Num: 26 + 11},
- {Name: "AZ", Num: 26 + 26},
- {Name: "ZZ", Num: 26 + 26*26},
- {Name: "AAA", Num: 26 + 26*26 + 1},
- }
- var invalidColumns = []struct {
- Name string
- Num int
- }{
- {Name: "", Num: -1},
- {Name: " ", Num: -1},
- {Name: "_", Num: -1},
- {Name: "__", Num: -1},
- {Name: "-1", Num: -1},
- {Name: "0", Num: -1},
- {Name: " A", Num: -1},
- {Name: "A ", Num: -1},
- {Name: "A1", Num: -1},
- {Name: "1A", Num: -1},
- {Name: " a", Num: -1},
- {Name: "a ", Num: -1},
- {Name: "a1", Num: -1},
- {Name: "1a", Num: -1},
- {Name: " _", Num: -1},
- {Name: "_ ", Num: -1},
- {Name: "_1", Num: -1},
- {Name: "1_", Num: -1},
- }
- var invalidCells = []string{"", "A", "AA", " A", "A ", "1A", "A1A", "A1 ", " A1", "1A1", "a-1", "A-1"}
- var invalidIndexes = []int{-100, -2, -1, 0}
- func TestColumnNameToNumber_OK(t *testing.T) {
- const msg = "Column %q"
- for _, col := range validColumns {
- out, err := ColumnNameToNumber(col.Name)
- if assert.NoErrorf(t, err, msg, col.Name) {
- assert.Equalf(t, col.Num, out, msg, col.Name)
- }
- }
- }
- func TestColumnNameToNumber_Error(t *testing.T) {
- const msg = "Column %q"
- for _, col := range invalidColumns {
- out, err := ColumnNameToNumber(col.Name)
- if assert.Errorf(t, err, msg, col.Name) {
- assert.Equalf(t, col.Num, out, msg, col.Name)
- }
- }
- _, err := ColumnNameToNumber("XFE")
- assert.EqualError(t, err, ErrColumnNumber.Error())
- }
- func TestColumnNumberToName_OK(t *testing.T) {
- const msg = "Column %q"
- for _, col := range validColumns {
- out, err := ColumnNumberToName(col.Num)
- if assert.NoErrorf(t, err, msg, col.Name) {
- assert.Equalf(t, strings.ToUpper(col.Name), out, msg, col.Name)
- }
- }
- }
- func TestColumnNumberToName_Error(t *testing.T) {
- out, err := ColumnNumberToName(-1)
- if assert.Error(t, err) {
- assert.Equal(t, "", out)
- }
- out, err = ColumnNumberToName(0)
- if assert.Error(t, err) {
- assert.Equal(t, "", out)
- }
- _, err = ColumnNumberToName(TotalColumns + 1)
- assert.EqualError(t, err, ErrColumnNumber.Error())
- }
- func TestSplitCellName_OK(t *testing.T) {
- const msg = "Cell \"%s%d\""
- for i, col := range validColumns {
- row := i + 1
- c, r, err := SplitCellName(col.Name + strconv.Itoa(row))
- if assert.NoErrorf(t, err, msg, col.Name, row) {
- assert.Equalf(t, col.Name, c, msg, col.Name, row)
- assert.Equalf(t, row, r, msg, col.Name, row)
- }
- }
- }
- func TestSplitCellName_Error(t *testing.T) {
- const msg = "Cell %q"
- for _, cell := range invalidCells {
- c, r, err := SplitCellName(cell)
- if assert.Errorf(t, err, msg, cell) {
- assert.Equalf(t, "", c, msg, cell)
- assert.Equalf(t, -1, r, msg, cell)
- }
- }
- }
- func TestJoinCellName_OK(t *testing.T) {
- const msg = "Cell \"%s%d\""
- for i, col := range validColumns {
- row := i + 1
- cell, err := JoinCellName(col.Name, row)
- if assert.NoErrorf(t, err, msg, col.Name, row) {
- assert.Equalf(t, strings.ToUpper(fmt.Sprintf("%s%d", col.Name, row)), cell, msg, row)
- }
- }
- }
- func TestJoinCellName_Error(t *testing.T) {
- const msg = "Cell \"%s%d\""
- test := func(col string, row int) {
- cell, err := JoinCellName(col, row)
- if assert.Errorf(t, err, msg, col, row) {
- assert.Equalf(t, "", cell, msg, col, row)
- }
- }
- for _, col := range invalidColumns {
- test(col.Name, 1)
- for _, row := range invalidIndexes {
- test("A", row)
- test(col.Name, row)
- }
- }
- }
- func TestCellNameToCoordinates_OK(t *testing.T) {
- const msg = "Cell \"%s%d\""
- for i, col := range validColumns {
- row := i + 1
- c, r, err := CellNameToCoordinates(col.Name + strconv.Itoa(row))
- if assert.NoErrorf(t, err, msg, col.Name, row) {
- assert.Equalf(t, col.Num, c, msg, col.Name, row)
- assert.Equalf(t, i+1, r, msg, col.Name, row)
- }
- }
- }
- func TestCellNameToCoordinates_Error(t *testing.T) {
- const msg = "Cell %q"
- for _, cell := range invalidCells {
- c, r, err := CellNameToCoordinates(cell)
- if assert.Errorf(t, err, msg, cell) {
- assert.Equalf(t, -1, c, msg, cell)
- assert.Equalf(t, -1, r, msg, cell)
- }
- }
- _, _, err := CellNameToCoordinates("A1048577")
- assert.EqualError(t, err, "row number exceeds maximum limit")
- }
- func TestCoordinatesToCellName_OK(t *testing.T) {
- const msg = "Coordinates [%d, %d]"
- for i, col := range validColumns {
- row := i + 1
- cell, err := CoordinatesToCellName(col.Num, row)
- if assert.NoErrorf(t, err, msg, col.Num, row) {
- assert.Equalf(t, strings.ToUpper(col.Name+strconv.Itoa(row)), cell, msg, col.Num, row)
- }
- }
- }
- func TestCoordinatesToCellName_Error(t *testing.T) {
- const msg = "Coordinates [%d, %d]"
- test := func(col, row int) {
- cell, err := CoordinatesToCellName(col, row)
- if assert.Errorf(t, err, msg, col, row) {
- assert.Equalf(t, "", cell, msg, col, row)
- }
- }
- for _, col := range invalidIndexes {
- test(col, 1)
- for _, row := range invalidIndexes {
- test(1, row)
- test(col, row)
- }
- }
- }
- func TestBytesReplace(t *testing.T) {
- s := []byte{0x01}
- assert.EqualValues(t, s, bytesReplace(s, []byte{}, []byte{}, 0))
- }
- func TestSetIgnorableNameSpace(t *testing.T) {
- f := NewFile()
- f.xmlAttr["xml_path"] = []xml.Attr{{}}
- f.setIgnorableNameSpace("xml_path", 0, xml.Attr{Name: xml.Name{Local: "c14"}})
- assert.EqualValues(t, "c14", f.xmlAttr["xml_path"][0].Value)
- }
- func TestStack(t *testing.T) {
- s := NewStack()
- assert.Equal(t, s.Peek(), nil)
- assert.Equal(t, s.Pop(), nil)
- }
- func TestGenXMLNamespace(t *testing.T) {
- assert.Equal(t, genXMLNamespace([]xml.Attr{
- {Name: xml.Name{Space: NameSpaceXML, Local: "space"}, Value: "preserve"},
- }), `xml:space="preserve">`)
- }
- func TestBstrUnmarshal(t *testing.T) {
- bstrs := map[string]string{
- "*": "*",
- "*_x0000_": "*",
- "*_x0008_": "*",
- "_x0008_*": "*",
- "*_x0008_*": "**",
- "*_x4F60__x597D_": "*你好",
- "*_xG000_": "*_xG000_",
- "*_xG05F_x0001_*": "*_xG05F*",
- "*_x005F__x0008_*": "*_x005F_*",
- "*_x005F_x0001_*": "*_x0001_*",
- "*_x005f_x005F__x0008_*": "*_x005F_*",
- "*_x005F_x005F_xG05F_x0006_*": "*_x005F_xG05F*",
- "*_x005F_x005F_x005F_x0006_*": "*_x005F_x0006_*",
- "_x005F__x0008_******": "_x005F_******",
- "******_x005F__x0008_": "******_x005F_",
- "******_x005F__x0008_******": "******_x005F_******",
- }
- for bstr, expected := range bstrs {
- assert.Equal(t, expected, bstrUnmarshal(bstr))
- }
- }
- func TestBstrMarshal(t *testing.T) {
- bstrs := map[string]string{
- "*_xG05F_*": "*_xG05F_*",
- "*_x0008_*": "*_x005F_x0008_*",
- "*_x005F_*": "*_x005F_x005F_*",
- "*_x005F_xG006_*": "*_x005F_x005F_xG006_*",
- "*_x005F_x0006_*": "*_x005F_x005F_x005F_x0006_*",
- }
- for bstr, expected := range bstrs {
- assert.Equal(t, expected, bstrMarshal(bstr))
- }
- }
|