Add more tests and bench target

Makefile

@@ -37,6 +37,9 @@ image:
 	@docker build --build-arg TAG=$(TAG) --build-arg BUILD=$(BUILD) -t $(REPO):$(TAG) .
 	@echo "Image created: $(REPO):$(TAG)"
 
+bench:
+	@go test -v -bench=. $(TEST_ARGS)
+
 test:
 	@go test -v -cover -race $(TEST_ARGS)
 

msgbus_test.go

@@ -34,6 +34,17 @@ func TestMessageGetEmpty(t *testing.T) {
 	assert.Equal(t, msg, Message{})
 }
 
+func TestMessageBusPutGet(t *testing.T) {
+	mb := NewMessageBus(nil)
+	topic := mb.NewTopic("foo")
+	expected := Message{Topic: topic, Payload: []byte("foo")}
+	mb.Put(expected)
+
+	actual, ok := mb.Get(topic)
+	assert.True(t, ok)
+	assert.Equal(t, actual, expected)
+}
+
 func BenchmarkMessageBusPut(b *testing.B) {
 	mb := NewMessageBus(nil)
 	topic := mb.NewTopic("foo")