81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package core
|
||
|
||
import (
|
||
"reflect"
|
||
"testing"
|
||
|
||
"github.com/celestix/telegraph-go/v2"
|
||
)
|
||
|
||
func TestGetImgSrcs(t *testing.T) {
|
||
complexStructure := telegraph.NodeElement{
|
||
Tag: "div",
|
||
Children: []telegraph.Node{
|
||
telegraph.NodeElement{
|
||
Tag: "figure",
|
||
Children: []telegraph.Node{
|
||
telegraph.NodeElement{
|
||
Tag: "img",
|
||
Attrs: map[string]string{
|
||
"src": "https://example.com/image1.png",
|
||
},
|
||
},
|
||
telegraph.NodeElement{
|
||
Tag: "p",
|
||
Children: []telegraph.Node{
|
||
"A text node",
|
||
},
|
||
},
|
||
telegraph.NodeElement{
|
||
Tag: "figure",
|
||
Children: []telegraph.Node{
|
||
telegraph.NodeElement{
|
||
Tag: "img",
|
||
Attrs: map[string]string{
|
||
"src": "https://example.com/image2.png",
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
telegraph.NodeElement{
|
||
Tag: "img",
|
||
Attrs: map[string]string{
|
||
"src": "https://example.com/image3.png",
|
||
},
|
||
},
|
||
"text node",
|
||
telegraph.NodeElement{
|
||
Tag: "div",
|
||
Children: []telegraph.Node{
|
||
telegraph.NodeElement{
|
||
Tag: "span",
|
||
Children: []telegraph.Node{
|
||
telegraph.NodeElement{
|
||
Tag: "img",
|
||
Attrs: map[string]string{
|
||
"src": "https://example.com/image4.png",
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
},
|
||
}
|
||
|
||
expected := []string{
|
||
"https://example.com/image1.png",
|
||
"https://example.com/image2.png",
|
||
"https://example.com/image3.png",
|
||
"https://example.com/image4.png",
|
||
}
|
||
|
||
got := getNodeImages(complexStructure)
|
||
|
||
if !reflect.DeepEqual(expected, got) {
|
||
t.Errorf("expected %v，got %v", expected, got)
|
||
}
|
||
}
|