/
page.go
110 lines (93 loc) · 2.43 KB
/
page.go
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
package web
import (
"fmt"
"github.com/sclevine/agouti"
"github.com/stretchr/testify/assert"
"runtime/debug"
"strings"
"testing"
"time"
)
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) 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)
}