Exemplo n.º 1
0
func TestMiddlewares(t *testing.T) {
	m := mux.New()
	m.Use(testMiddleware)

	m.Get("/teas").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("sencha\n"))
	})

	m.Get("/cars").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("audi\n"))
	})

	w := httptest.NewRecorder()
	r, _ := http.NewRequest("GET", "/teas", nil)
	m.ServeHTTP(w, r)
	if w.Body.String() != "test\nsencha\n" {
		t.Fatalf("response body expected: %#v, got: %#v", "test\nsencha\n", w.Body.String())
	}

	w = httptest.NewRecorder()
	r, _ = http.NewRequest("GET", "/cars", nil)
	m.ServeHTTP(w, r)
	if w.Body.String() != "test\naudi\n" {
		t.Fatalf("response body expected: %#v, got: %#v", "test\naudi\n", w.Body.String())
	}
}
Exemplo n.º 2
0
func TestRoute(t *testing.T) {
	m := mux.New()
	m.Get("/teas").Use(testMiddleware).Then(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("sencha\n"))
	}))

	w := httptest.NewRecorder()
	r, err := http.NewRequest("GET", "/teas", nil)
	if err != nil {
		t.Fatal(err)
	}

	m.ServeHTTP(w, r)

	if w.Body.String() != "test\nsencha\n" {
		t.Fatalf("response body expected: %#v, got: %#v", "test\nsencha\n", w.Body.String())
	}
}
Exemplo n.º 3
0
func TestParams(t *testing.T) {
	m := mux.New()
	m.Get("/teas/:id").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		p := mux.Params(r)
		fmt.Fprintf(w, "%s\n", p.ByName("id"))
	})

	w := httptest.NewRecorder()
	r, err := http.NewRequest("GET", "/teas/hojicha", nil)
	if err != nil {
		t.Fatal(err)
	}

	m.ServeHTTP(w, r)

	if w.Body.String() != "hojicha\n" {
		t.Fatalf("response body expected: %#v, got: %#v", "hojicha\n", w.Body.String())
	}
}
Exemplo n.º 4
0
// New() returns a new mux with handlers useful to create JSON APIs.
func New() *mux.Mux {
	m := mux.New()
	m.Chain = alice.New(RecoverHandler, LoggingHandler, AcceptHandler)
	return m
}
Exemplo n.º 5
0
func TestRoutes(t *testing.T) {
	var get, head, post, patch, put, delete bool

	m := mux.New()
	m.Get("/get").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		get = true
	})

	m.Head("/head").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		head = true
	})

	m.Post("/post").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		post = true
	})

	m.Patch("/patch").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		patch = true
	})

	m.Put("/put").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		put = true
	})

	m.Delete("/delete").ThenFunc(func(w http.ResponseWriter, r *http.Request) {
		delete = true
	})

	w := httptest.NewRecorder()

	r, _ := http.NewRequest("GET", "/get", nil)
	m.ServeHTTP(w, r)
	if !get {
		t.Fatalf("routing GET failed")
	}

	r, _ = http.NewRequest("HEAD", "/head", nil)
	m.ServeHTTP(w, r)
	if !head {
		t.Fatalf("routing HEAD failed")
	}

	r, _ = http.NewRequest("POST", "/post", nil)
	m.ServeHTTP(w, r)
	if !post {
		t.Fatalf("routing POST failed")
	}

	r, _ = http.NewRequest("PATCH", "/patch", nil)
	m.ServeHTTP(w, r)
	if !patch {
		t.Fatalf("routing PATCH failed")
	}

	r, _ = http.NewRequest("PUT", "/put", nil)
	m.ServeHTTP(w, r)
	if !put {
		t.Fatalf("routing PUT failed")
	}

	r, _ = http.NewRequest("DELETE", "/delete", nil)
	m.ServeHTTP(w, r)
	if !delete {
		t.Fatalf("routing DELETE failed")
	}
}