request_test.go

package natsproxy

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
	"testing"

	"github.com/gogo/protobuf/proto"
)

func TestUnmarshallFrom(t *testing.T) {
	URL := "/api/method"
	original := &Request{
		URL:  URL,
		Body: []byte{0xFF, 0xFC},
	}
	payload, _ := proto.Marshal(original)

	copyObj := &Request{}
	if err := copyObj.UnmarshallFrom(payload); err != nil {
		t.Error(err)
	}

	if original.URL != copyObj.URL {
		fmt.Println()
		t.Error("URL not equals")
	}

	// TODO more precise equality test
	if len(original.Body) != len(copyObj.Body) {
		t.Error("Body not equals")
	}
}

func TestNewRequestFromHttp(t *testing.T) {
	url, _ := url.Parse("http://test.com/test")
	httpReq := &http.Request{
		Method: "GET",
		URL:    url,
		Body:   ioutil.NopCloser(bytes.NewReader([]byte{0xFF, 0xFC})),
	}
	req := NewRequest()
	err := req.FromHTTP(httpReq)

	if err != nil {
		t.Error(err)
	}
	if req.URL != "http://test.com/test" {
		t.Error("Url not equals")
	}
	// TODO better test for Body
	if len(req.Body) != 2 {
		t.Error("Body length not equals")
	}
}

func BenchmarkMarshallRequest(b *testing.B) {
	b.StopTimer()
	url, _ := url.Parse("http://test.com/test")
	httpReq := &http.Request{
		Method: "GET",
		URL:    url,
		Body:   ioutil.NopCloser(bytes.NewReader([]byte{0xFF, 0xFC})),
	}
	req := NewRequest()
	req.FromHTTP(httpReq)
	b.StartTimer()
	for i := 0; i < b.N; i++ {
		proto.Marshal(req)
	}

}