Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.0 KiB
42 lines
1.0 KiB
// Copyright 2017 The Gitea Authors. All rights reserved. |
|
// Use of this source code is governed by a MIT-style |
|
// license that can be found in the LICENSE file. |
|
|
|
package integrations |
|
|
|
import ( |
|
"bytes" |
|
"testing" |
|
|
|
"github.com/PuerkitoBio/goquery" |
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
// HTMLDoc struct |
|
type HTMLDoc struct { |
|
doc *goquery.Document |
|
} |
|
|
|
// NewHTMLParser parse html file |
|
func NewHTMLParser(t testing.TB, content []byte) *HTMLDoc { |
|
doc, err := goquery.NewDocumentFromReader(bytes.NewReader(content)) |
|
assert.NoError(t, err) |
|
return &HTMLDoc{doc: doc} |
|
} |
|
|
|
// GetInputValueByID for get input value by id |
|
func (doc *HTMLDoc) GetInputValueByID(id string) string { |
|
text, _ := doc.doc.Find("#" + id).Attr("value") |
|
return text |
|
} |
|
|
|
// GetInputValueByName for get input value by name |
|
func (doc *HTMLDoc) GetInputValueByName(name string) string { |
|
text, _ := doc.doc.Find("input[name=\"" + name + "\"]").Attr("value") |
|
return text |
|
} |
|
|
|
// GetCSRF for get CSRC token value from input |
|
func (doc *HTMLDoc) GetCSRF() string { |
|
return doc.GetInputValueByName("_csrf") |
|
}
|
|
|