124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
package suites
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-rod/rod"
|
|
"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
|
|
}
|
|
|
|
// StartRodWithProxy create a rod/chromedp session.
|
|
func StartRodWithProxy(proxy string) (*RodSession, error) {
|
|
browserPath := os.Getenv("BROWSER_PATH")
|
|
if browserPath == "" {
|
|
browserPath = "/usr/bin/chromium-browser"
|
|
}
|
|
|
|
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(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,
|
|
}, nil
|
|
}
|
|
|
|
// StartRod create a rod/chromedp session.
|
|
func StartRod() (*RodSession, error) {
|
|
return StartRodWithProxy("")
|
|
}
|
|
|
|
// 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)
|
|
}
|