diff options
Diffstat (limited to 'main_test.go')
| -rw-r--r-- | main_test.go | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/main_test.go b/main_test.go index b3fbde4..fab7129 100644 --- a/main_test.go +++ b/main_test.go @@ -36,7 +36,7 @@ var ( const pth = "testdata/brand/57083" -func helperLoadBytes(t *testing.T, name string) []byte { +func helperLoadBytes(t testing.TB, name string) []byte { t.Helper() path := filepath.Join("testdata", name) bytes, err := ioutil.ReadFile(path) @@ -92,7 +92,16 @@ func TestServedFeed(t *testing.T) { } } -func helperMockServer(t *testing.T) *httptest.Server { +func BenchmarkServedFeed(b *testing.B) { + server := helperMockServer(b) + defer helperCleanupServer(b) + + for n := 0; n < b.N; n++ { + processURL(fmt.Sprintf("%s/brand/57083/episodes", server.URL)) + } +} + +func helperMockServer(t testing.TB) *httptest.Server { t.Helper() fileserver := http.FileServer(http.Dir("testdata")) @@ -107,13 +116,13 @@ func helperMockServer(t *testing.T) *httptest.Server { return server } -func helperCleanupServer(t *testing.T) { +func helperCleanupServer(t testing.TB) { t.Helper() helperCleanupFile(t, "episodes") helperCleanupFile(t, "about") } -func helperCleanupFile(t *testing.T, name string) { +func helperCleanupFile(t testing.TB, name string) { t.Helper() if err := os.Remove(filepath.Join(pth, name)); err != nil { t.Fatal(err) |