1 files changed, 64 insertions, 0 deletions
diff --git a/vendor/gopkg.in/throttled/throttled.v1/memstats_test.go b/vendor/gopkg.in/throttled/throttled.v1/memstats_test.go
new file mode 100644
index 000000000..2b8faa721
--- /dev/null
+++ b/vendor/gopkg.in/throttled/throttled.v1/memstats_test.go
@@ -0,0 +1,64 @@
+package throttled
+
+import (
+	"net/http"
+	"runtime"
+	"testing"
+	"time"
+
+	"github.com/PuerkitoBio/boom/commands"
+)
+
+func TestMemStats(t *testing.T) {
+	if testing.Short() {
+		t.Skip()
+	}
+	cases := []struct {
+		n     int
+		c     int
+		gc    uint32
+		total uint64
+		rate  time.Duration
+	}{
+		0: {1000, 10, 3, 0, 0},
+		1: {200, 10, 0, 600000, 0},
+		2: {500, 10, 2, 555555, 10 * time.Millisecond},
+	}
+	for i, c := range cases {
+		// Setup the stats handler
+		st := &stats{}
+		// Create the throttler
+		limit := MemThresholds(&runtime.MemStats{NumGC: c.gc, TotalAlloc: c.total})
+		th := MemStats(limit, c.rate)
+		th.DeniedHandler = http.HandlerFunc(st.DeniedHTTP)
+		// Run the test
+		b := commands.Boom{
+			Req:    &commands.ReqOpts{},
+			N:      c.n,
+			C:      c.c,
+			Output: "quiet",
+		}
+		rpts := runTest(th.Throttle(st), b)
+		// Assert results
+		assertStats(t, i, st, rpts)
+		assertMem(t, i, limit)
+	}
+}
+
+func assertMem(t *testing.T, ix int, limit *runtime.MemStats) {
+	var mem runtime.MemStats
+	runtime.ReadMemStats(&mem)
+	if mem.NumGC < limit.NumGC {
+		t.Errorf("%d: expected gc to be at least %d, got %d", ix, limit.NumGC, mem.NumGC)
+	}
+	if mem.TotalAlloc < limit.TotalAlloc {
+		t.Errorf("%d: expected total alloc to be at least %dKb, got %dKb", ix, limit.TotalAlloc/1024, mem.TotalAlloc/1024)
+	}
+}
+
+func BenchmarkReadMemStats(b *testing.B) {
+	var mem runtime.MemStats
+	for i := 0; i < b.N; i++ {
+		runtime.ReadMemStats(&mem)
+	}
+}