add tests

builder/beacon_client.go

@@ -18,8 +18,7 @@ type IBeaconClient interface {
 
 type NilBeaconClient struct{}
 
-func (b *NilBeaconClient) SubscribeToPayloadAttributesEvents(payloadAttrC chan BuilderPayloadAttributes) {
-}
+func (b *NilBeaconClient) SubscribeToPayloadAttributesEvents(payloadAttrC chan BuilderPayloadAttributes) {}
 
 func (b *NilBeaconClient) Start() error { return nil }
 

builder/beacon_client_test.go

@@ -1,33 +1,42 @@
 package builder
 
 import (
-	"net/http/httptest"
+	"testing"
+	"time"
 
-	"github.com/gorilla/mux"
+	"gotest.tools/assert"
 )
 
-type testBeaconClient struct {
-	slot uint64
+type mockBeaconNode struct {
+	payloadAttributes BuilderPayloadAttributes
 }
 
-func (b *testBeaconClient) Stop() {}
+func (b *mockBeaconNode) Stop() {}
 
-func (b *testBeaconClient) SubscribeToPayloadAttributesEvents(payloadAttrC chan BuilderPayloadAttributes) {
+func (b *mockBeaconNode) SubscribeToPayloadAttributesEvents(payloadAttrC chan BuilderPayloadAttributes) {
+	go func() {
+		payloadAttrC <- b.payloadAttributes
+	}()
 }
 
-func (b *testBeaconClient) Start() error { return nil }
+func (b *mockBeaconNode) Start() error { return nil }
 
-type mockBeaconNode struct {
-	srv *httptest.Server
+func newMockBeaconNode(payloadAttributes BuilderPayloadAttributes) *mockBeaconNode {
+	return &mockBeaconNode{
+		payloadAttributes: payloadAttributes,
+	}
 }
 
-func newMockBeaconNode() *mockBeaconNode {
-	r := mux.NewRouter()
-	srv := httptest.NewServer(r)
+func TestMockBeaconNode(t *testing.T) {
+	mockBeaconNode := newMockBeaconNode(BuilderPayloadAttributes{Slot: 123})
+	payloadAttrC := make(chan BuilderPayloadAttributes, 1) // Use buffered channel
 
-	mbn := &mockBeaconNode{
-		srv: srv,
-	}
+	mockBeaconNode.SubscribeToPayloadAttributesEvents(payloadAttrC)
 
-	return mbn
+	select {
+	case payloadAttributes := <-payloadAttrC:
+		assert.Equal(t, uint64(123), payloadAttributes.Slot)
+	case <-time.After(2 * time.Second):
+		t.Fatal("Test timed out waiting for payload attributes")
+	}
 }