wee/browser_test.go at main · coghost/wee · GitHub

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
package wee

import (
	"net/http/httptest"
	"testing"

	"github.com/coghost/wee/fixtures"
	"github.com/stretchr/testify/suite"
)

type BrowserSuite struct {
	suite.Suite

	ts *httptest.Server
}

func TestBrowser(t *testing.T) {
	suite.Run(t, new(BrowserSuite))
}

func (s *BrowserSuite) SetupSuite() {
	s.ts = fixtures.NewTestServer()
}

func (s *BrowserSuite) TearDownSuite() {
}

func (s *BrowserSuite) TestNewLauncher() {
	client := NewLauncher(BrowserHeadless(true))
	s.Contains(client.MustLaunch(), "ws://127.0.0.1:")
}

func (s *BrowserSuite) TestNewBrowser() {
	exts := "fixtures/chrome-extension"
	dataDir := "/tmp/001"

	l, brw := NewBrowser(
		BrowserExtensions(exts),
		BrowserPaintRects(true),
		BrowserIncognito(true),
		BrowserUserDataDir(dataDir),
	)
	s.NotNil(l)
	s.NotNil(brw)

	defer l.Cleanup()
	defer brw.Close()

	ff, b := l.GetFlags("load-extension")
	s.True(b)
	s.Equal(exts, ff)

	v, b := l.GetFlags("user-data-dir")
	s.True(b)
	s.Contains(v, dataDir)

	brw.MustPage(s.ts.URL)
	// uncomment Blocked() to check the extension manually.
	// Blocked()
	SleepN(2)
}