100 lines
3.2 KiB
Go
100 lines
3.2 KiB
Go
package excelize
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAddShape(t *testing.T) {
|
|
f, err := prepareTestBook1()
|
|
if !assert.NoError(t, err) {
|
|
t.FailNow()
|
|
}
|
|
shape := &Shape{
|
|
Type: "rect",
|
|
Paragraph: []ShapeParagraph{
|
|
{Text: "Rectangle", Font: Font{Color: "CD5C5C"}},
|
|
{Text: "Shape", Font: Font{Bold: true, Color: "2980B9"}},
|
|
},
|
|
}
|
|
assert.NoError(t, f.AddShape("Sheet1", "A30", shape))
|
|
assert.NoError(t, f.AddShape("Sheet1", "B30", &Shape{Type: "rect", Paragraph: []ShapeParagraph{{Text: "Rectangle"}, {}}}))
|
|
assert.NoError(t, f.AddShape("Sheet1", "C30", &Shape{Type: "rect"}))
|
|
assert.EqualError(t, f.AddShape("Sheet3", "H1",
|
|
&Shape{
|
|
Type: "ellipseRibbon",
|
|
Color: ShapeColor{Line: "#4286f4", Fill: "#8eb9ff"},
|
|
Paragraph: []ShapeParagraph{
|
|
{
|
|
Font: Font{
|
|
Bold: true,
|
|
Italic: true,
|
|
Family: "Times New Roman",
|
|
Size: 36,
|
|
Color: "#777777",
|
|
Underline: "single",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
), "sheet Sheet3 does not exist")
|
|
assert.EqualError(t, f.AddShape("Sheet3", "H1", nil), ErrParameterInvalid.Error())
|
|
assert.EqualError(t, f.AddShape("Sheet1", "A", shape), newCellNameToCoordinatesError("A", newInvalidCellNameError("A")).Error())
|
|
assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddShape1.xlsx")))
|
|
|
|
// Test add first shape for given sheet
|
|
f = NewFile()
|
|
width, height := 1.2, 90
|
|
assert.NoError(t, f.AddShape("Sheet1", "A1",
|
|
&Shape{
|
|
Type: "ellipseRibbon",
|
|
Color: ShapeColor{Line: "#4286f4", Fill: "#8eb9ff"},
|
|
Paragraph: []ShapeParagraph{
|
|
{
|
|
Font: Font{
|
|
Bold: true,
|
|
Italic: true,
|
|
Family: "Times New Roman",
|
|
Size: 36,
|
|
Color: "#777777",
|
|
Underline: "single",
|
|
},
|
|
},
|
|
},
|
|
Height: &height,
|
|
Line: ShapeLine{Width: &width},
|
|
}))
|
|
assert.NoError(t, f.SaveAs(filepath.Join("test", "TestAddShape2.xlsx")))
|
|
// Test add shape with invalid sheet name
|
|
assert.EqualError(t, f.AddShape("Sheet:1", "A30", shape), ErrSheetNameInvalid.Error())
|
|
// Test add shape with unsupported charset style sheet
|
|
f.Styles = nil
|
|
f.Pkg.Store(defaultXMLPathStyles, MacintoshCyrillicCharset)
|
|
assert.EqualError(t, f.AddShape("Sheet1", "B30", &Shape{Type: "rect", Paragraph: []ShapeParagraph{{Text: "Rectangle"}, {}}}), "XML syntax error on line 1: invalid UTF-8")
|
|
// Test add shape with unsupported charset content types
|
|
f = NewFile()
|
|
f.ContentTypes = nil
|
|
f.Pkg.Store(defaultXMLPathContentTypes, MacintoshCyrillicCharset)
|
|
assert.EqualError(t, f.AddShape("Sheet1", "B30", &Shape{Type: "rect", Paragraph: []ShapeParagraph{{Text: "Rectangle"}, {}}}), "XML syntax error on line 1: invalid UTF-8")
|
|
}
|
|
|
|
func TestAddDrawingShape(t *testing.T) {
|
|
f := NewFile()
|
|
path := "xl/drawings/drawing1.xml"
|
|
f.Pkg.Store(path, MacintoshCyrillicCharset)
|
|
assert.EqualError(t, f.addDrawingShape("sheet1", path, "A1",
|
|
&Shape{
|
|
Width: intPtr(defaultShapeSize),
|
|
Height: intPtr(defaultShapeSize),
|
|
Format: PictureOptions{
|
|
PrintObject: boolPtr(true),
|
|
Locked: boolPtr(false),
|
|
XScale: float64Ptr(defaultPictureScale),
|
|
YScale: float64Ptr(defaultPictureScale),
|
|
},
|
|
},
|
|
), "XML syntax error on line 1: invalid UTF-8")
|
|
}
|