parser_test.go

package gpac_test

import (
	"fmt"
	"io/ioutil"
	"os"
	"testing"

	"github.com/darren/gpac"
)

func Example() {
	pacf, _ := os.Open("testdata/wpad.dat")
	defer pacf.Close()

	data, _ := ioutil.ReadAll(pacf)
	pac, _ := gpac.New(string(data))

	r, _ := pac.FindProxyForURL("http://www.example.com/")

	fmt.Println(r)
	// Output:
	// PROXY 4.5.6.7:8080; PROXY 7.8.9.10:8080
}

func TestProxyGet(t *testing.T) {
	pacf, _ := os.Open("testdata/wpad.dat")
	defer pacf.Close()

	data, _ := ioutil.ReadAll(pacf)

	pac, err := gpac.New(string(data))
	if err != nil {
		t.Fatal(err)
	}

	proxies, err := pac.FindProxy("http://www.example.com/")
	if err != nil {
		t.Fatal(err)
	}

	if len(proxies) != 2 {
		t.Fatal("Find proxy failed")
	}

	if proxies[1].URL() != "7.8.9.10:8080" {
		t.Error("Get URL from proxy failed")
	}

}

func TestProxyGetDirect(t *testing.T) {
	dsts := []string{
		"http://localhost/",
		"https://intranet.domain.com",
		"http://abcdomain.com",
		"http://www.abcdomain.com",
		"ftp://example.com.com",
	}

	pacf, _ := os.Open("testdata/wpad.dat")
	defer pacf.Close()

	data, _ := ioutil.ReadAll(pacf)

	pac, err := gpac.New(string(data))
	if err != nil {
		t.Fatal(err)
	}

	for _, dst := range dsts {
		proxies, err := pac.FindProxy(dst)
		if err != nil {
			t.Fatal(err)
		}

		if len(proxies) != 1 {
			t.Fatalf("Find proxy failed for %s", dst)
		}

		if proxies[0].URL() != "" {
			t.Errorf("Get URL from proxy failed: %s, proxies: %+v", dst, proxies)
		}
	}
}

// test server started in proxy_test.go
func TestPacGet(t *testing.T) {
	pac, err := gpac.New(`
        function FindProxyForURL(url, host) {
            return "PROXY 127.0.0.1:9991; PROXY 127.0.0.1:9992; PROXY 127.0.0.1:8080; DIRECT"
        }
    `)

	if err != nil {
		t.Fatal(err)
	}

	resp, err := pac.Get("http://localhost:8081")
	if err != nil {
		t.Fatal(err)
	}

	body := readBodyAndClose(resp)
	if body != "Example" {
		t.Errorf("Response not expected: %s", body)
	}
}

func BenchmarkFind(b *testing.B) {
	pacf, _ := os.Open("testdata/wpad.dat")
	defer pacf.Close()

	data, _ := ioutil.ReadAll(pacf)
	pac, _ := gpac.New(string(data))

	for n := 0; n < b.N; n++ {
		pac.FindProxyForURL("http://www.example.com/")
		pac.FindProxyForURL("http://localhost/")
		pac.FindProxyForURL("http://192.168.1.1/")
	}
}