mirror of
https://github.com/restic/rest-server.git
synced 2025-12-07 09:36:13 -08:00
4cd82b6802
These are the vendor files needed for Prometheus support. I have not been able to figure out how to do this properly with gopkg.
24 lines
518 B
Go
24 lines
518 B
Go
package datacounter
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestResponseWriterCounter(t *testing.T) {
|
|
handler := func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write(data)
|
|
}
|
|
req, err := http.NewRequest("GET", "http://example.com/foo", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
w := httptest.NewRecorder()
|
|
counter := NewResponseWriterCounter(w)
|
|
handler(counter, req)
|
|
if counter.Count() != dataLen {
|
|
t.Fatalf("count mismatch len of test data: %d != %d", counter.Count(), len(data))
|
|
}
|
|
}
|