adds a basic BroadcastReceiver test

broadcast_test.go

@@ -35,3 +35,57 @@ func testMessageMarshal(t *testing.T, m proto.Message) {
 		t.Fatalf("unexpected message marshalling: %s", unmarshalled)
 	}
 }
+
+// Ensure that BroadcastReceiver can register a BroadcastHandler.
+func TestBroadcast_BroadcastReceiver(t *testing.T) {
+
+	s := pilosa.NewServer()
+
+	sbr := NewSimpleBroadcastReceiver()
+	sbh := NewSimpleBroadcastHandler()
+
+	s.BroadcastReceiver = sbr
+	s.BroadcastReceiver.Start(sbh)
+
+	msg := &internal.DeleteDBMessage{
+		DB: "d",
+	}
+
+	s.BroadcastReceiver.(*SimpleBroadcastReceiver).Receive(msg)
+
+	// Make sure the message received is what was sentd
+	if !reflect.DeepEqual(sbh.receivedMessage, msg) {
+		t.Fatalf("unexpected message: %s", sbh.receivedMessage)
+	}
+}
+
+type SimpleBroadcastReceiver struct {
+	broadcastHandler pilosa.BroadcastHandler
+}
+
+func NewSimpleBroadcastReceiver() *SimpleBroadcastReceiver {
+	return &SimpleBroadcastReceiver{}
+}
+
+func (r *SimpleBroadcastReceiver) Start(h pilosa.BroadcastHandler) error {
+	r.broadcastHandler = h
+	return nil
+}
+
+func (r *SimpleBroadcastReceiver) Receive(pb proto.Message) error {
+	r.broadcastHandler.ReceiveMessage(pb)
+	return nil
+}
+
+type SimpleBroadcastHandler struct {
+	receivedMessage proto.Message
+}
+
+func NewSimpleBroadcastHandler() *SimpleBroadcastHandler {
+	return &SimpleBroadcastHandler{}
+}
+
+func (h *SimpleBroadcastHandler) ReceiveMessage(pb proto.Message) error {
+	h.receivedMessage = pb.(proto.Message)
+	return nil
+}