chore: add context tests

Author: ssncferreira

context/context_test.go

@@ -0,0 +1,88 @@
+package context
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/aibridge/recorder"
+)
+
+func TestAsActor(t *testing.T) {
+	t.Parallel()
+
+	// Given: a metadata map
+	metadata := recorder.Metadata{"key": "value"}
+
+	// When: storing an actor in the context
+	ctx := AsActor(context.Background(), "actor-123", metadata)
+
+	// Then: the actor should be retrievable with correct ID and metadata
+	actor := ActorFromContext(ctx)
+	require.NotNil(t, actor)
+	assert.Equal(t, "actor-123", actor.ID)
+	assert.Equal(t, "value", actor.Metadata["key"])
+}
+
+func TestActorFromContext(t *testing.T) {
+	t.Parallel()
+
+	t.Run("returns actor when present", func(t *testing.T) {
+		t.Parallel()
+
+		// Given: a context with an actor
+		ctx := AsActor(context.Background(), "test-id", recorder.Metadata{})
+
+		// When: extracting the actor from context
+		actor := ActorFromContext(ctx)
+
+		// Then: the actor should be returned with correct ID
+		require.NotNil(t, actor)
+		assert.Equal(t, "test-id", actor.ID)
+	})
+
+	t.Run("returns nil when no actor", func(t *testing.T) {
+		t.Parallel()
+
+		// Given: a context without an actor
+		ctx := context.Background()
+
+		// When: extracting the actor from context
+		actor := ActorFromContext(ctx)
+
+		// Then: nil should be returned
+		assert.Nil(t, actor)
+	})
+}
+
+func TestActorIDFromContext(t *testing.T) {
+	t.Parallel()
+
+	t.Run("returns actor ID when present", func(t *testing.T) {
+		t.Parallel()
+
+		// Given: a context with an actor
+		ctx := AsActor(context.Background(), "test-actor-id", recorder.Metadata{})
+
+		// When: extracting the actor ID from context
+		got := ActorIDFromContext(ctx)
+
+		// Then: the actor ID should be returned
+		assert.Equal(t, "test-actor-id", got)
+	})
+
+	t.Run("returns empty string when no actor", func(t *testing.T) {
+		t.Parallel()
+
+		// Given: a context without an actor
+		ctx := context.Background()
+
+		// When: extracting the actor ID from context
+		got := ActorIDFromContext(ctx)
+
+		// Then: an empty string should be returned
+		assert.Empty(t, got)
+	})
+}