setup_test.go

package rod_test

import (
	"path/filepath"
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/suite"
	"github.com/ysmood/kit"
	"github.com/ysmood/rod"
)

var slash = filepath.FromSlash

// S test suite
type S struct {
	suite.Suite
	browser *rod.Browser
	page    *rod.Page
}

func Test(t *testing.T) {
	s := new(S)
	s.browser = rod.New().Trace(true).Client(nil).Connect()

	defer s.browser.Close()

	s.page = s.browser.Page("")
	s.page.Viewport(800, 600, 1, false)

	suite.Run(t, s)
}

// get abs file path from fixtures folder, return sample "file:///a/b/click.html"
func srcFile(path string) string {
	return "file://" + file(path)
}

// get abs file path from fixtures folder, return sample "/a/b/click.html"
func file(path string) string {
	f, err := filepath.Abs(slash(path))
	kit.E(err)
	return f
}

func ginHTML(body string) gin.HandlerFunc {
	return func(ctx kit.GinContext) {
		ctx.Header("Content-Type", "text/html;")
		kit.E(ctx.Writer.WriteString(body))
	}
}

func serve() (string, *gin.Engine, func()) {
	srv := kit.MustServer("127.0.0.1:0")
	go func() { kit.Noop(srv.Do()) }()

	url := "http://" + srv.Listener.Addr().String()

	return url, srv.Engine, func() { kit.E(srv.Listener.Close()) }
}