Function TestMethod_Throttling

method_test.go:11–55

(t *testing.T)

)

func TestMethod_Throttling(t *testing.T) {
	k := New("testkite", "0.0.1")
	k.Config.DisableAuthentication = true
	k.Config.Port = 9996

	k.HandleFunc("foo", func(r *Request) (interface{}, error) {
		return "handle", nil
	}).Throttle(time.Second*2, 30)

	go k.Run()
	defer k.Close()
	<-k.ServerReadyNotify()

	c := New("exp", "0.0.1").NewClient("http://127.0.0.1:9996/kite")
	if err := c.Dial(); err != nil {
		t.Fatal(err)
	}

	// First let us exhaust the bucket
	for i := 0; i < 20; i++ {
		_, err := c.TellWithTimeout("foo", 4*time.Second)
		if err != nil {
			t.Fatal(err)
		}
	}

	// now, the request 21 should give a requestLimitError
	_, err := c.TellWithTimeout("foo", 4*time.Second)
	if err != nil {
		if kErr, ok := err.(*Error); ok {
			if kErr.Type != "requestLimitError" {
				t.Fatal(err)
			}
		}
	}

	// now wait until the bucket is filled again
	time.Sleep(time.Second * 2)

	// this shouldn't give any error at all
	_, err = c.TellWithTimeout("foo", 4*time.Second)
	if err != nil {
		t.Fatal(err)
	}
}

func TestMethod_Latest(t *testing.T) {
	k := New("testkite", "0.0.1")