mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-16 11:37:36 +02:00
6e15846d0f
* Added Context to HTML methods * Fixed unit tests * Updated timeout * Fixed WAIT_CLASS timeout
108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package html
|
|
|
|
import (
|
|
"context"
|
|
"github.com/MontFerret/ferret/pkg/drivers"
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values/types"
|
|
"time"
|
|
)
|
|
|
|
const defaultTimeout = 5000
|
|
|
|
func NewLib() map[string]core.Function {
|
|
return map[string]core.Function{
|
|
"CLICK": Click,
|
|
"CLICK_ALL": ClickAll,
|
|
"DOCUMENT": Document,
|
|
"DOWNLOAD": Download,
|
|
"ELEMENT": Element,
|
|
"ELEMENT_EXISTS": ElementExists,
|
|
"ELEMENTS": Elements,
|
|
"ELEMENTS_COUNT": ElementsCount,
|
|
"HOVER": Hover,
|
|
"INNER_HTML": InnerHTML,
|
|
"INNER_HTML_ALL": InnerHTMLAll,
|
|
"INNER_TEXT": InnerText,
|
|
"INNER_TEXT_ALL": InnerTextAll,
|
|
"INPUT": Input,
|
|
"NAVIGATE": Navigate,
|
|
"NAVIGATE_BACK": NavigateBack,
|
|
"NAVIGATE_FORWARD": NavigateForward,
|
|
"PAGINATION": Pagination,
|
|
"PDF": PDF,
|
|
"SCREENSHOT": Screenshot,
|
|
"SCROLL_BOTTOM": ScrollBottom,
|
|
"SCROLL_ELEMENT": ScrollInto,
|
|
"SCROLL_TOP": ScrollTop,
|
|
"SELECT": Select,
|
|
"WAIT_ELEMENT": WaitElement,
|
|
"WAIT_CLASS": WaitClass,
|
|
"WAIT_CLASS_ALL": WaitClassAll,
|
|
"WAIT_NAVIGATION": WaitNavigation,
|
|
}
|
|
}
|
|
|
|
func ValidateDocument(ctx context.Context, value core.Value) (core.Value, error) {
|
|
err := core.ValidateType(value, drivers.HTMLDocumentType, types.String)
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
var doc drivers.HTMLDocument
|
|
|
|
if value.Type() == types.String {
|
|
buf, err := Document(ctx, value, values.NewBoolean(true))
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
doc = buf.(drivers.HTMLDocument)
|
|
} else {
|
|
doc = value.(drivers.HTMLDocument)
|
|
}
|
|
|
|
return doc, nil
|
|
}
|
|
|
|
func waitTimeout(ctx context.Context, value values.Int) (context.Context, context.CancelFunc) {
|
|
return context.WithTimeout(
|
|
ctx,
|
|
time.Duration(value)*time.Millisecond,
|
|
)
|
|
}
|
|
|
|
func resolveElement(value core.Value) (drivers.HTMLElement, error) {
|
|
vt := value.Type()
|
|
|
|
if vt == drivers.HTMLDocumentType {
|
|
return value.(drivers.HTMLDocument).DocumentElement(), nil
|
|
} else if vt == drivers.HTMLElementType {
|
|
return value.(drivers.HTMLElement), nil
|
|
}
|
|
|
|
return nil, core.TypeError(value.Type(), drivers.HTMLDocumentType, drivers.HTMLElementType)
|
|
}
|
|
|
|
func toDocument(value core.Value) (drivers.HTMLDocument, error) {
|
|
err := core.ValidateType(value, drivers.HTMLDocumentType)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return value.(drivers.HTMLDocument), nil
|
|
}
|
|
|
|
func toElement(value core.Value) (drivers.HTMLElement, error) {
|
|
err := core.ValidateType(value, drivers.HTMLElementType)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return value.(drivers.HTMLElement), nil
|
|
}
|