chromedp/nav_test.go

package chromedp

import (
	"strings"
	"testing"
	"time"

	"github.com/knq/chromedp/cdp/page"
)

func TestNavigate(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	expurl, exptitle := testdataDir+"/image.html", "this is title"

	err = c.Run(defaultContext, Navigate(expurl))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, WaitVisible(`#icon-brankas`, ByID))
	if err != nil {
		t.Fatal(err)
	}

	var urlstr string
	err = c.Run(defaultContext, Location(&urlstr))
	if err != nil {
		t.Fatal(err)
	}
	if !strings.HasPrefix(urlstr, expurl) {
		t.Errorf("expected to be on image.html, at: %s", urlstr)
	}

	var title string
	err = c.Run(defaultContext, Title(&title))
	if err != nil {
		t.Fatal(err)
	}
	if title != exptitle {
		t.Errorf("expected title to contain google, instead title is: %s", title)
	}
}

func TestNavigationEntries(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	tests := []string{
		"form.html",
		"image.html",
	}

	var entries []*page.NavigationEntry
	var index int64

	err = c.Run(defaultContext, NavigationEntries(&index, &entries))
	if err != nil {
		t.Fatal(err)
	}

	if len(entries) != 1 {
		t.Errorf("expected to have 1 navigation entry: got %d", len(entries))
	}
	if index != 0 {
		t.Errorf("expected navigation index is 0, got: %d", index)
	}

	expIdx, expEntries := 1, 2
	for i, url := range tests {
		err = c.Run(defaultContext, Navigate(testdataDir+"/"+url))
		if err != nil {
			t.Fatal(err)
		}

		time.Sleep(50 * time.Millisecond)

		err = c.Run(defaultContext, NavigationEntries(&index, &entries))
		if err != nil {
			t.Fatal(err)
		}

		if len(entries) != expEntries {
			t.Errorf("test %d expected to have %d navigation entry: got %d", i, expEntries, len(entries))
		}
		if index != int64(i+1) {
			t.Errorf("test %d expected navigation index is %d, got: %d", i, i, index)
		}

		expIdx++
		expEntries++
	}
}

func TestNavigateToHistoryEntry(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	var entries []*page.NavigationEntry
	var index int64
	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"image.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	err = c.Run(defaultContext, NavigationEntries(&index, &entries))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	err = c.Run(defaultContext, NavigateToHistoryEntry(entries[index].ID))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var title string
	err = c.Run(defaultContext, Title(&title))
	if err != nil {
		t.Fatal(err)
	}
	if title != entries[index].Title {
		t.Errorf("expected title to be %s, instead title is: %s", entries[index].Title, title)
	}
}

func TestNavigateBack(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var exptitle string
	err = c.Run(defaultContext, Title(&exptitle))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"image.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	err = c.Run(defaultContext, NavigateBack())
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var title string
	err = c.Run(defaultContext, Title(&title))
	if err != nil {
		t.Fatal(err)
	}
	if title != exptitle {
		t.Errorf("expected title to be %s, instead title is: %s", exptitle, title)
	}
}

func TestNavigateForward(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"image.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var exptitle string
	err = c.Run(defaultContext, Title(&exptitle))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, NavigateBack())
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	err = c.Run(defaultContext, NavigateForward())
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var title string
	err = c.Run(defaultContext, Title(&title))
	if err != nil {
		t.Fatal(err)
	}
	if title != exptitle {
		t.Errorf("expected title to be %s, instead title is: %s", exptitle, title)
	}
}

func TestStop(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, Stop())
	if err != nil {
		t.Fatal(err)
	}
}

func TestReload(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var exptitle string
	err = c.Run(defaultContext, Title(&exptitle))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, Reload())
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var title string
	err = c.Run(defaultContext, Title(&title))
	if err != nil {
		t.Fatal(err)
	}
	if title != exptitle {
		t.Errorf("expected title to be %s, instead title is: %s", exptitle, title)
	}
}

func TestCaptureScreenshot(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"image.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var buf []byte
	err = c.Run(defaultContext, CaptureScreenshot(&buf))
	if err != nil {
		t.Fatal(err)
	}

	if len(buf) == 0 {
		t.Fatal("failed to capture screenshot")
	}
	//TODO: test image
}

func TestAddOnLoadScript(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	var scriptID page.ScriptIdentifier
	err = c.Run(defaultContext, AddOnLoadScript(`window.alert("TEST")`, &scriptID))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	if scriptID == "" {
		t.Fatal("got empty script ID")
	}
	// TODO: Handle javascript dialog.
}

func TestRemoveOnLoadScript(t *testing.T) {
	t.Parallel()

	var err error

	c := testAllocate(t, "")
	defer c.Release()

	var scriptID page.ScriptIdentifier
	err = c.Run(defaultContext, AddOnLoadScript(`window.alert("TEST")`, &scriptID))
	if err != nil {
		t.Fatal(err)
	}

	if scriptID == "" {
		t.Fatal("got empty script ID")
	}

	err = c.Run(defaultContext, RemoveOnLoadScript(scriptID))
	if err != nil {
		t.Fatal(err)
	}

	err = c.Run(defaultContext, Navigate(testdataDir+"/"+"form.html"))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)
}

func TestLocation(t *testing.T) {
	t.Parallel()

	var err error
	expurl := testdataDir + "/" + "form.html"

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(expurl))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var urlstr string
	err = c.Run(defaultContext, Location(&urlstr))
	if err != nil {
		t.Fatal(err)
	}

	if urlstr != expurl {
		t.Fatalf("expected to be on form.html,got: %s", urlstr)
	}
}

func TestTitle(t *testing.T) {
	t.Parallel()

	var err error
	expurl, exptitle := testdataDir+"/image.html", "this is title"

	c := testAllocate(t, "")
	defer c.Release()

	err = c.Run(defaultContext, Navigate(expurl))
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(50 * time.Millisecond)

	var title string
	err = c.Run(defaultContext, Title(&title))
	if err != nil {
		t.Fatal(err)
	}

	if title != exptitle {
		t.Fatalf("expected title to be %s, got: %s", exptitle, title)
	}
}