Merge pull request #26 from thiagokokada/add-test-subscribe-test

event/event_test: add TestSubscribe

event/event_test.go

@@ -2,6 +2,7 @@ package event
 
 import (
 	"context"
+	"errors"
 	"os"
 	"testing"
 	"time"
@@ -44,6 +45,27 @@ func TestReceive(t *testing.T) {
 	}
 }
 
+func TestSubscribe(t *testing.T) {
+	if os.Getenv("HYPRLAND_INSTANCE_SIGNATURE") == "" {
+		t.Skip("HYPRLAND_INSTANCE_SIGNATURE not set, skipping test")
+	}
+
+	c := MustClient()
+	defer c.Close()
+	ctx, cancel := context.WithTimeout(
+		context.Background(),
+		100*time.Millisecond,
+	)
+	defer cancel()
+
+	// Make sure that we can exit a Subscribe loop by cancelling the
+	// context
+	err := c.Subscribe(ctx, &DefaultEventHandler{}, AllEvents...)
+
+	assert.Error(t, err)
+	assert.True(t, errors.Is(err, context.DeadlineExceeded))
+}
+
 func TestProcessEvent(t *testing.T) {
 	h := &FakeEventHandler{t: t}
 	c := &FakeEventClient{}