Commit 51cbd249 authored by santiaago's avatar santiaago
Browse files

add controller banner test

parent 918103ba
Loading
Loading
Loading
Loading
+30 −0
Original line number Diff line number Diff line
package squares

import (
	"net/http"
	"testing"

	"github.com/taironas/route"
	tgTesting "github.com/taironas/tinygraphs/testing"
)

func TestBannerRandom(t *testing.T) {

	r := new(route.Router)
	r.HandleFunc("/squares/banner/random", BannerRandom)

	test := tgTesting.GenerateHandlerFunc(t, BannerRandom)
	for _, p := range tgTesting.GoodParams {
		recorder := test("/squares/banner/random", "GET", p, r)
		if recorder.Code != http.StatusOK {
			t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusOK)
		}
	}

	for _, p := range tgTesting.BadParams {
		recorder := test("/squares/banner/random", "GET", p, r)
		if recorder.Code != http.StatusOK {
			t.Errorf("returned %v. Expected %v.", recorder.Code, http.StatusOK)
		}
	}
}