This commit is contained in:
@ -181,6 +181,16 @@ type Good struct {
|
||||
Price float64
|
||||
}
|
||||
|
||||
func (g *Good) GoodID() int {
|
||||
linkArr := strings.Split(g.GoodLink, "/")
|
||||
if len(linkArr) < 3 {
|
||||
return 0
|
||||
}
|
||||
|
||||
id, _ := strconv.Atoi(linkArr[3])
|
||||
return id
|
||||
}
|
||||
|
||||
func parseGoodsCategory(r io.Reader) ([]*Good, error) {
|
||||
doc, err := goquery.NewDocumentFromReader(r)
|
||||
if err != nil {
|
||||
|
@ -51,3 +51,30 @@ func Test_parseGoodsCategory(t *testing.T) {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestGood_GoodID(t *testing.T) {
|
||||
type fields struct {
|
||||
GoodLink string
|
||||
}
|
||||
tests := []struct {
|
||||
fields fields
|
||||
want int
|
||||
}{
|
||||
{
|
||||
fields: fields{
|
||||
GoodLink: "/itm/mortal-kombat-1-ua-kz-auto-24-7/4107232",
|
||||
},
|
||||
want: 4107232,
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.fields.GoodLink, func(t *testing.T) {
|
||||
g := &Good{
|
||||
GoodLink: tt.fields.GoodLink,
|
||||
}
|
||||
if got := g.GoodID(); got != tt.want {
|
||||
t.Errorf("Good.GoodID() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user