Fix client tests

client/client_test.go

@@ -1,7 +1,9 @@
 package client
 
 import (
+	"io/ioutil"
 	"net/http/httptest"
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -14,8 +16,13 @@ func TestClientPublish(t *testing.T) {
 	assert := assert.New(t)
 	require := require.New(t)
 
-	mb, err := msgbus.NewMessageBus()
+	testdir, err := ioutil.TempDir("", "msgbus-logs-*")
 	require.NoError(err)
+	defer os.RemoveAll(testdir)
+
+	mb, err := msgbus.NewMessageBus(msgbus.WithLogPath(testdir))
+	require.NoError(err)
+	defer os.RemoveAll(testdir)
 
 	server := httptest.NewServer(mb)
 	defer server.Close()
@@ -23,14 +30,15 @@ func TestClientPublish(t *testing.T) {
 	client := NewClient(server.URL, nil)
 
 	err = client.Publish("hello", "hello world")
-	require.NoError(err)
+
+	assert.NoError(err)
 
 	topic := mb.NewTopic("hello")
 	expected := msgbus.Message{Topic: topic, Payload: []byte("hello world")}
 
 	actual, ok := mb.Get(topic)
-	assert.True(ok)
-	assert.Equal(actual.ID, expected.ID)
-	assert.Equal(actual.Topic, expected.Topic)
-	assert.Equal(actual.Payload, expected.Payload)
+	require.True(ok)
+	assert.Equal(expected.ID, actual.ID)
+	assert.Equal(expected.Topic, actual.Topic)
+	assert.Equal(expected.Payload, actual.Payload)
 }