Merge pull request #3 from groove-x/feature/windowsize

add method of changing windows size
groove-x · Jul 31, 2019 · 8a6aff4 · 8a6aff4
2 parents a5ff347 + 1ccf3b0
commit 8a6aff4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/browser.go b/browser.go
@@ -240,3 +240,17 @@ func (b *Browser) ExpectTransitTo(rawurl string) *Browser {
 	}
 	return b
 }
+
+
+
+func (b *Browser) SetWindowSize(width,height int) {
+	b.session.GetCurrentWindowHandle().SetSize(webdriver.Size{width,height})
+}
+
+func (b *Browser) GetWindowSize()(width,height int, err error) {
+	size,err := b.session.GetCurrentWindowHandle().GetSize()
+	if err != nil {
+		return
+	}
+	return size.Width,size.Height, nil
+}
diff --git a/example/example_test.go b/example/example_test.go
@@ -8,8 +8,8 @@ import (
 	"testing"
 	"time"
 
-	"github.com/atotto/webtester"
-	"github.com/atotto/webtester/chrome"
+	"github.com/groove-x/webtester"
+	"github.com/groove-x/webtester/chrome"
 )
 
 func TestMain(m *testing.M) {
@@ -60,6 +60,7 @@ func TestPlayground(t *testing.T) {
 	defer ts.TearDown()
 
 	b := ts.OpenBrowser()
+	b.SetWindowSize(1600,1200)
 	b.SetPageLoadTimeout(2 * time.Second)
 
 	b.VisitTo("https://play.golang.org/")