Proxy_test.go

package dns

import "tholian-endpoint/protocols/dns"
import "tholian-endpoint/protocols/http"
import "tholian-warps/protocols/test"
import net_url "net/url"
import "strconv"
import "testing"
import "time"

func TestProxy(t *testing.T) {

	t.Run("Proxy with DNS Type A Payload", func(t *testing.T) {

		cache := test.NewSpyProxyCache(false, nil, false)
		proxy := NewProxy("localhost", 13337, &cache)
		tunnel := NewTunnel("127.0.0.1", 13337)

		query := dns.NewPacket()
		query.SetType("query")
		query.AddQuestion(dns.NewQuestion("example.com", dns.TypeA))

		go func() {

			err1 := proxy.Listen()

			if err1 != nil {
				t.Errorf("Unexpected error '%s'", err1.Error())
			}

		}()

		go func() {

			time.Sleep(2 * time.Second)

			err2 := proxy.Destroy()

			if err2 != nil {
				t.Errorf("Unexpected error '%s'", err2.Error())
			}

		}()

		time.Sleep(100 * time.Millisecond)
		response := tunnel.ResolvePacket(query)

		if len(response.Questions) > 0 && len(response.Answers) > 0 {

			record := response.Answers[0]

			if record.Type != dns.TypeA {
				t.Errorf("Expected DNS response record type to be '%s' but got '%s'", dns.TypeA.String(), dns.Type(record.Type).String())
			}

		} else {
			t.Errorf("Expected DNS response questions to be '%d' but got '%d'", len(query.Questions), len(response.Questions))
		}

		// Let Proxy destroy the UDP Listener
		time.Sleep(2 * time.Second)

	})

	t.Run("Proxy with DNS Type AAAA Payload", func(t *testing.T) {

		cache := test.NewSpyProxyCache(false, nil, false)
		proxy := NewProxy("localhost", 13337, &cache)
		tunnel := NewTunnel("127.0.0.1", 13337)

		query := dns.NewPacket()
		query.SetType("query")
		query.AddQuestion(dns.NewQuestion("example.com", dns.TypeAAAA))

		go func() {

			err1 := proxy.Listen()

			if err1 != nil {
				t.Errorf("Unexpected error '%s'", err1.Error())
			}

		}()

		go func() {

			time.Sleep(2 * time.Second)

			err2 := proxy.Destroy()

			if err2 != nil {
				t.Errorf("Unexpected error '%s'", err2.Error())
			}

		}()

		time.Sleep(100 * time.Millisecond)
		response := tunnel.ResolvePacket(query)

		if len(response.Questions) > 0 && len(response.Answers) > 0 {

			record := response.Answers[0]

			if record.Type != dns.TypeAAAA {
				t.Errorf("Expected DNS response record type to be '%s' but got '%s'", dns.TypeAAAA.String(), dns.Type(record.Type).String())
			}

		} else {
			t.Errorf("Expected DNS response questions to be '%d' but got '%d'", len(query.Questions), len(response.Questions))
		}

		// Let Proxy destroy the UDP Listener
		time.Sleep(2 * time.Second)

	})

	t.Run("Proxy with ProxyCache and Small HTTP Payload", func(t *testing.T) {

		url, _ := net_url.Parse("http://example.com/index.html")
		expected := http.NewPacket()
		expected.SetURL(*url)
		expected.SetStatus(http.StatusOK)
		expected.SetHeader("Content-Type", "text/html")
		expected.SetHeader("X-Proxy", "SpyProxyCache")
		expected.SetPayload([]byte("Hello, world!"))

		cache := test.NewSpyProxyCache(true, &expected, false)
		proxy := NewProxy("localhost", 13337, &cache)
		tunnel := NewTunnel("127.0.0.1", 13337)

		request := http.NewPacket()
		request.SetMethod(http.MethodGet)
		request.SetURL(*url)

		go func() {

			err1 := proxy.Listen()

			if err1 != nil {
				t.Errorf("Unexpected error '%s'", err1.Error())
			}

		}()

		go func() {

			time.Sleep(2 * time.Second)

			err2 := proxy.Destroy()

			if err2 != nil {
				t.Errorf("Unexpected error '%s'", err2.Error())
			}

		}()

		response := tunnel.RequestPacket(request)
		response.Decode()

		if expected.Status != response.Status {
			t.Errorf("Expected HTTP response status '%s' but got '%s'", http.Status(expected.Status).String(), http.Status(response.Status).String())
		}

		if string(expected.Payload) != string(response.Payload) {
			t.Errorf("Expected HTTP response payload '%s' but got '%s'", string(expected.Payload), string(response.Payload))
		}

		// Let Proxy destroy the UDP Listener
		time.Sleep(2 * time.Second)

	})

	t.Run("Proxy with ProxyCache and Large HTTP Payload", func(t *testing.T) {

		url, _ := net_url.Parse("http://example.com/index.txt")
		expected := http.NewPacket()
		expected.SetURL(*url)
		expected.SetStatus(http.StatusOK)
		expected.SetHeader("Content-Type", "text/plain")
		expected.SetHeader("X-Proxy", "SpyProxyCache")

		payload := make([]byte, 0)

		for l := 0; l < 100; l++ {
			payload = append(payload, []byte("Hello, line "+strconv.Itoa(l)+"!\n")...)
		}

		expected.SetPayload(payload)

		cache := test.NewSpyProxyCache(true, &expected, false)
		proxy := NewProxy("localhost", 13337, &cache)
		tunnel := NewTunnel("127.0.0.1", 13337)

		request := http.NewPacket()
		request.SetMethod(http.MethodGet)
		request.SetURL(*url)

		go func() {

			err1 := proxy.Listen()

			if err1 != nil {
				t.Errorf("Unexpected error '%s'", err1.Error())
			}

		}()

		go func() {

			time.Sleep(2 * time.Second)

			err2 := proxy.Destroy()

			if err2 != nil {
				t.Errorf("Unexpected error '%s'", err2.Error())
			}

		}()

		response := tunnel.RequestPacket(request)
		response.Decode()

		if expected.Status != response.Status {
			t.Errorf("Expected HTTP response status '%s' but got '%s'", http.Status(expected.Status).String(), http.Status(response.Status).String())
		}

		if string(expected.Payload) != string(response.Payload) {
			t.Errorf("Expected HTTP different response payload")
		}

		// Let Proxy destroy the UDP Listener
		time.Sleep(2 * time.Second)

	})

}