mirror of
https://github.com/tnypxl/rollup.git
synced 2025-12-13 06:23:18 +00:00
refactor: update browser initialization in scraper tests
This commit is contained in:
@@ -339,6 +339,16 @@ func ClosePlaywright() {
|
||||
}
|
||||
}
|
||||
|
||||
// InitBrowser initializes the browser
|
||||
func InitBrowser() error {
|
||||
return InitPlaywright()
|
||||
}
|
||||
|
||||
// CloseBrowser closes the browser
|
||||
func CloseBrowser() {
|
||||
ClosePlaywright()
|
||||
}
|
||||
|
||||
// FetchWebpageContent retrieves the content of a webpage using Playwright
|
||||
func FetchWebpageContent(urlStr string) (string, error) {
|
||||
logger.Printf("Fetching webpage content for URL: %s\n", urlStr)
|
||||
|
||||
@@ -144,11 +144,11 @@ This is a **test** paragraph.
|
||||
}
|
||||
|
||||
func TestExtractLinks(t *testing.T) {
|
||||
// Initialize the browser before running the test
|
||||
if err := InitBrowser(); err != nil {
|
||||
t.Fatalf("Failed to initialize browser: %v", err)
|
||||
// Initialize Playwright before running the test
|
||||
if err := InitPlaywright(); err != nil {
|
||||
t.Fatalf("Failed to initialize Playwright: %v", err)
|
||||
}
|
||||
defer CloseBrowser()
|
||||
defer ClosePlaywright()
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "text/html")
|
||||
|
||||
Reference in New Issue
Block a user