energylink/pkg/bus/bus_test.go

package bus

import (
	"errors"
	"testing"
)

type TestQuery struct {
	Id   int64
	Resp string
}

func TestHandlerReturnsError(t *testing.T) {
	bus := New()

	bus.AddQueryHandler(func(query *TestQuery) error {
		return errors.New("handler error")
	})

	err := bus.SendQuery(&TestQuery{})

	if err == nil {
		t.Fatal("Send query failed %v", err)
	} else {
		t.Log("Handler error received ok")
	}
}

func TestHandlerReturn(t *testing.T) {
	bus := New()

	bus.AddQueryHandler(func(q *TestQuery) error {
		q.Resp = "hello from handler"
		return nil
	})

	query := &TestQuery{}
	err := bus.SendQuery(query)

	if err != nil {
		t.Fatal("Send query failed %v", err)
	} else if query.Resp != "hello from handler" {
		t.Fatal("Failed to get response from handler")
	}
}