dispatch_test.go

package dispatch

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"reflect"
	"testing"

	"github.com/lunny/log"
	"github.com/lunny/tango"
	"github.com/unknwon/macaron"
)

func TestDispatch(t *testing.T) {
	buff := bytes.NewBufferString("")
	recorder := httptest.NewRecorder()
	recorder.Body = buff

	l := log.Std
	l.SetOutputLevel(log.Ldebug)

	t1 := tango.NewWithLog(l, tango.Logging())
	t1.Get("/", func(res http.ResponseWriter, req *http.Request) {
		res.Write([]byte("tango 1"))
	})

	t2 := tango.NewWithLog(l, tango.Logging())
	t2.Get("/", func(res http.ResponseWriter, req *http.Request) {
		res.Write([]byte("tango 2"))
	})

	dispatch := New(map[string]Handler{
		"/": t1,
	})
	dispatch.Add("/api/", t2)

	t3 := tango.NewWithLog(l)
	t3.Use(dispatch)

	req, err := http.NewRequest("GET", "http://localhost:8000/", nil)
	if err != nil {
		t.Error(err)
	}

	t3.ServeHTTP(recorder, req)
	expect(t, recorder.Code, http.StatusOK)
	refute(t, len(buff.String()), 0)
	expect(t, buff.String(), "tango 1")

	req, err = http.NewRequest("GET", "http://localhost:8000/api/", nil)
	if err != nil {
		t.Error(err)
	}

	buff.Reset()

	t3.ServeHTTP(recorder, req)
	expect(t, recorder.Code, http.StatusOK)
	refute(t, len(buff.String()), 0)
	expect(t, buff.String(), "tango 2")
}

func TestDispatch2(t *testing.T) {
	buff := bytes.NewBufferString("")
	recorder := httptest.NewRecorder()
	recorder.Body = buff

	l := log.Std
	l.SetOutputLevel(log.Ldebug)

	t1 := tango.NewWithLog(l, tango.Logging())
	t1.Get("/", func(res http.ResponseWriter, req *http.Request) {
		res.Write([]byte("tango 1"))
	})

	m := macaron.Classic()
	m.Get("/", func(ctx *macaron.Context) string {
		return "macaron"
	})

	dispatch := New(map[string]Handler{
		"/": t1,
	})
	dispatch.Add("/api/", m)

	t3 := tango.NewWithLog(l)
	t3.Use(dispatch)

	req, err := http.NewRequest("GET", "http://localhost:8000/", nil)
	if err != nil {
		t.Error(err)
	}

	t3.ServeHTTP(recorder, req)
	expect(t, recorder.Code, http.StatusOK)
	refute(t, len(buff.String()), 0)
	expect(t, buff.String(), "tango 1")

	req, err = http.NewRequest("GET", "http://localhost:8000/api/", nil)
	if err != nil {
		t.Error(err)
	}

	buff.Reset()

	t3.ServeHTTP(recorder, req)
	expect(t, recorder.Code, http.StatusOK)
	refute(t, len(buff.String()), 0)
	expect(t, buff.String(), "macaron")
}

/* Test Helpers */
func expect(t *testing.T, a interface{}, b interface{}) {
	if a != b {
		t.Errorf("Expected %v (type %v) - Got %v (type %v)", b, reflect.TypeOf(b), a, reflect.TypeOf(a))
	}
}

func refute(t *testing.T, a interface{}, b interface{}) {
	if a == b {
		t.Errorf("Did not expect %v (type %v) - Got %v (type %v)", b, reflect.TypeOf(b), a, reflect.TypeOf(a))
	}
}