74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package e2eutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/playwright-community/playwright-go"
|
|
)
|
|
|
|
// NewPlaywright initializes Playwright and returns a browser instance.
|
|
// It automatically registers cleanup handlers for test teardown.
|
|
func NewPlaywright(t *testing.T) (*playwright.Playwright, playwright.Browser, error) {
|
|
t.Helper()
|
|
|
|
// Run Playwright
|
|
pw, err := playwright.Run()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
// Register cleanup for Playwright
|
|
t.Cleanup(func() {
|
|
if err := pw.Stop(); err != nil {
|
|
t.Logf("Failed to stop Playwright: %v", err)
|
|
}
|
|
})
|
|
|
|
// Launch Chromium browser with headless options
|
|
browser, err := pw.Chromium.Launch(playwright.BrowserTypeLaunchOptions{
|
|
Headless: playwright.Bool(true),
|
|
Args: []string{
|
|
"--disable-gpu",
|
|
"--disable-dev-shm-usage",
|
|
"--no-sandbox",
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
// Register cleanup for browser
|
|
t.Cleanup(func() {
|
|
if err := browser.Close(); err != nil {
|
|
t.Logf("Failed to close browser: %v", err)
|
|
}
|
|
})
|
|
|
|
return pw, browser, nil
|
|
}
|
|
|
|
// NewBrowserContext creates a new browser context with a standard viewport.
|
|
// It automatically registers cleanup handlers for test teardown.
|
|
func NewBrowserContext(browser playwright.Browser, t *testing.T) (playwright.BrowserContext, error) {
|
|
t.Helper()
|
|
|
|
// Create browser context with standard viewport
|
|
context, err := browser.NewContext(playwright.BrowserNewContextOptions{
|
|
Viewport: &playwright.Size{
|
|
Width: 1280,
|
|
Height: 720,
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Register cleanup
|
|
t.Cleanup(func() {
|
|
if err := context.Close(); err != nil {
|
|
t.Logf("Failed to close browser context: %v", err)
|
|
}
|
|
})
|
|
|
|
return context, nil
|
|
}
|