.\tests\tools.go

main
Zed A. Shaw 3 weeks ago
parent 74edb069d2
commit db9697130e
  1. 32
      tests/base_test.go

@ -2,40 +2,10 @@ package tests
import (
"testing"
"context"
"log"
"time"
"github.com/stretchr/testify/assert"
browser "github.com/chromedp/chromedp"
)
func Run(assert *assert.Assertions, ctx context.Context, actions ...browser.Action) {
err := browser.Run(ctx, actions...)
assert.NoError(err)
}
func Setup(timeout time.Duration) (context.Context, context.CancelFunc) {
opts := append(browser.DefaultExecAllocatorOptions[:],
browser.Flag("headless", false),)
ctx, cancel := browser.NewExecAllocator(context.Background(), opts...)
ctx, _ = browser.NewContext(ctx, browser.WithLogf(log.Printf))
ctx, _ = context.WithTimeout(ctx, timeout * time.Second)
return ctx, cancel
}
func ClickOn(assert *assert.Assertions, ctx context.Context, id string) {
err := browser.Run(ctx, browser.WaitVisible(`#streams`),)
assert.NoError(err)
resp, err := browser.RunResponse(ctx, browser.Click(id, browser.ByID))
assert.Equal(resp.Status, int64(200))
assert.NoError(err)
}
func TestLogin(t *testing.T) {
assert := assert.New(t)
@ -67,7 +37,7 @@ func TestStreamPage(t *testing.T) {
)
assert.NoError(err)
ClickOn(assert, ctx, `#streams`)
ClickOn(assert, ctx, `#streams`)
err = browser.Run(ctx, browser.WaitVisible(`#streams-title`))
assert.NoError(err)

Loading…
Cancel
Save