More benchmark tests

This commit is contained in:
James Mills 2022-04-03 19:43:38 +10:00
parent adc8e19aff
commit 229766036d
No known key found for this signature in database
GPG Key ID: AC4C014F1440EBD6

View File

@ -355,24 +355,45 @@ func TestServeHTTPSimple(t *testing.T) {
assert.Equal([]byte("hello world"), msg.Payload)
}
func BenchmarkServeHTTPPOST(b *testing.B) {
func BenchmarkServeHTTP_POST(b *testing.B) {
require := require.New(b)
testdir, err := ioutil.TempDir("", "msgbus-logs-*")
require.NoError(err)
defer os.RemoveAll(testdir)
b.Run("Sync", func(b *testing.B) {
testdir, err := ioutil.TempDir("", "msgbus-logs-*")
require.NoError(err)
defer os.RemoveAll(testdir)
mb, err := NewMessageBus(WithLogPath(testdir))
require.NoError(err)
mb, err := NewMessageBus(WithLogPath(testdir))
require.NoError(err)
b.ResetTimer()
for i := 0; i < b.N; i++ {
w := httptest.NewRecorder()
b := bytes.NewBufferString("hello world")
r, _ := http.NewRequest("POST", "/hello", b)
b.ResetTimer()
for i := 0; i < b.N; i++ {
w := httptest.NewRecorder()
b := bytes.NewBufferString("hello world")
r, _ := http.NewRequest("POST", "/hello", b)
mb.ServeHTTP(w, r)
}
})
b.Run("NoSync", func(b *testing.B) {
testdir, err := ioutil.TempDir("", "msgbus-logs-*")
require.NoError(err)
defer os.RemoveAll(testdir)
mb, err := NewMessageBus(WithLogPath(testdir), WithNoSync(true))
require.NoError(err)
b.ResetTimer()
for i := 0; i < b.N; i++ {
w := httptest.NewRecorder()
b := bytes.NewBufferString("hello world")
r, _ := http.NewRequest("POST", "/hello", b)
mb.ServeHTTP(w, r)
}
})
mb.ServeHTTP(w, r)
}
}
func TestServeHTTPSubscriber(t *testing.T) {