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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
package suites
import (
"fmt"
"os"
"strings"
"testing"
"time"
"github.com/go-rod/rod"
"github.com/go-rod/rod/lib/input"
"github.com/go-rod/rod/lib/launcher"
"github.com/stretchr/testify/require"
)
// RodSession binding a chrome session with devtool protocol.
type RodSession struct {
Launcher *launcher.Launcher
WebDriver *rod.Browser
RodSuiteCredentialsProvider
}
type RodSessionCredentials struct {
TOTP *OptionsTOTP
}
type RodSessionOpts struct {
proxy string
provider RodSuiteCredentialsProvider
}
type RodSessionOpt func(opts *RodSessionOpts) (err error)
func RodSessionWithProxy(proxy string) RodSessionOpt {
return func(opts *RodSessionOpts) (err error) {
opts.proxy = proxy
return nil
}
}
func RodSessionWithCredentials(provider RodSuiteCredentialsProvider) RodSessionOpt {
return func(opts *RodSessionOpts) (err error) {
opts.provider = provider
return nil
}
}
func NewRodSession(options ...RodSessionOpt) (session *RodSession, err error) {
opts := &RodSessionOpts{}
for _, option := range options {
if err = option(opts); err != nil {
return nil, fmt.Errorf("failed to apply option: %w", err)
}
}
if opts.provider == nil {
opts.provider = NewRodSuiteCredentials()
}
var browserPath string
if browserPath, err = GetBrowserPath(); err != nil {
return nil, err
}
headless := false
trace := true
motion := 0 * time.Second
if os.Getenv("HEADLESS") != "" {
headless = true
trace = false
motion = 0 * time.Second
}
l := launcher.New().
Bin(browserPath).
Proxy(opts.proxy).
Headless(headless).
Devtools(true)
url := l.MustLaunch()
browser := rod.New().
ControlURL(url).
Trace(trace).
SlowMotion(motion).
MustConnect()
browser.MustIgnoreCertErrors(true)
return &RodSession{
Launcher: l,
WebDriver: browser,
RodSuiteCredentialsProvider: opts.provider,
}, nil
}
// StartRod create a rod/chromedp session.
func StartRod() (*RodSession, error) {
return NewRodSession()
}
// Stop stop the rod/chromedp session.
func (rs *RodSession) Stop() error {
err := rs.WebDriver.Close()
if err != nil {
return err
}
rs.Launcher.Cleanup()
return err
}
// CheckElementExistsLocatedByID checks the existence of an element located by an id.
func (rs *RodSession) CheckElementExistsLocatedByID(t *testing.T, page *rod.Page, cssSelector string) bool {
b, _, err := page.Has("#" + cssSelector)
require.NoError(t, err)
return b
}
// WaitElementLocatedByClassName wait an element is located by class name.
func (rs *RodSession) WaitElementLocatedByClassName(t *testing.T, page *rod.Page, className string) *rod.Element {
e, err := page.Element("." + className)
require.NoError(t, err)
require.NotNil(t, e)
return e
}
// WaitElementLocatedByID waits for an element located by an id.
func (rs *RodSession) WaitElementLocatedByID(t *testing.T, page *rod.Page, cssSelector string) *rod.Element {
e, err := page.Element("#" + cssSelector)
require.NoError(t, err)
require.NotNil(t, e)
return e
}
// WaitElementsLocatedByID waits for an elements located by an id.
func (rs *RodSession) WaitElementsLocatedByID(t *testing.T, page *rod.Page, cssSelector string) rod.Elements {
e, err := page.Elements("#" + cssSelector)
require.NoError(t, err)
require.NotNil(t, e)
return e
}
func (rs *RodSession) waitBodyContains(t *testing.T, page *rod.Page, pattern string) {
text, err := page.MustElementR("body", pattern).Text()
require.NoError(t, err)
require.NotNil(t, text)
if strings.Contains(text, pattern) {
err = nil
} else {
err = fmt.Errorf("body does not contain pattern: %s", pattern)
}
require.NoError(t, err)
}
func (rs *RodSession) toInputs(in string) (out []input.Key) {
out = make([]input.Key, len(in))
for i, c := range in {
out[i] = input.Key(c)
}
return out
}
|
