File size: 2,613 Bytes
530729e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
package web
import (
"fmt"
"runtime/debug"
"strings"
"testing"
"time"
"github.com/sclevine/agouti"
"github.com/stretchr/testify/assert"
)
type Page struct {
*agouti.Page
T *testing.T
Driver *agouti.WebDriver
Quit chan struct{}
}
func (page *Page) Destroy() {
if r := recover(); r != nil {
debug.PrintStack()
fmt.Println("Recovered in f", r)
_ = page.Page.Destroy()
_ = page.Driver.Stop()
page.T.Fail()
page.Quit <- struct{}{}
}
}
func (page *Page) Wait(t int) {
time.Sleep(time.Duration(t) * time.Second)
}
func (page *Page) Contain(s string) {
content, err := page.HTML()
assert.Equal(page.T, err, nil)
assert.Equal(page.T, strings.Contains(content, s), true)
}
func (page *Page) NoContain(s string) {
content, err := page.HTML()
assert.Equal(page.T, err, nil)
assert.Equal(page.T, strings.Contains(content, s), false)
}
func (page *Page) Css(xpath, css, res string) {
style, err := page.FindByXPath(xpath).CSS(css)
assert.Equal(page.T, err, nil)
assert.Equal(page.T, style, res)
}
func (page *Page) CssS(s *agouti.Selection, css, res string) {
style, err := s.CSS(css)
assert.Equal(page.T, err, nil)
assert.Equal(page.T, style, res)
}
func (page *Page) Text(xpath, text string) {
mli1, err := page.FindByXPath(xpath).Text()
assert.Equal(page.T, err, nil)
assert.Equal(page.T, mli1, text)
}
func (page *Page) MoveMouseBy(xOffset, yOffset int) {
assert.Equal(page.T, page.Page.MoveMouseBy(xOffset, yOffset), nil)
}
func (page *Page) Display(xpath string) {
page.Css(xpath, "display", "block")
}
func (page *Page) Nondisplay(xpath string) {
page.Css(xpath, "display", "none")
}
func (page *Page) Value(xpath, value string) {
val, err := page.FindByXPath(xpath).Attribute("value")
assert.Equal(page.T, err, nil)
assert.Equal(page.T, val, value)
}
func (page *Page) Click(xpath string, intervals ...int) {
assert.Equal(page.T, page.FindByXPath(xpath).Click(), nil)
interval := 1
if len(intervals) > 0 {
interval = intervals[0]
}
page.Wait(interval)
}
func (page *Page) ClickS(s *agouti.Selection, intervals ...int) {
assert.Equal(page.T, s.Click(), nil)
interval := 1
if len(intervals) > 0 {
interval = intervals[0]
}
page.Wait(interval)
}
func (page *Page) Attr(s *agouti.Selection, attr, res string) {
style, err := s.Attribute(attr)
assert.Equal(page.T, err, nil)
assert.Equal(page.T, style, res)
}
func (page *Page) Fill(xpath, content string) {
assert.Equal(page.T, page.FindByXPath(xpath).Fill(content), nil)
}
func (page *Page) NavigateTo(path string) {
assert.Equal(page.T, page.Navigate(path), nil)
page.Wait(2)
}
|