Fix test engine root component constraint

Author: awln-temporal

chasm/chasmtest/test_engine.go

@@ -18,9 +18,9 @@ import (
 )
 
 type (
-	Option[T chasm.Component] func(*Engine[T])
+	Option[T chasm.RootComponent] func(*Engine[T])
 
-	Engine[T chasm.Component] struct {
+	Engine[T chasm.RootComponent] struct {
 		t        *testing.T
 		registry *chasm.Registry
 		logger   log.Logger
@@ -46,7 +46,7 @@ type (
 	}
 )
 
-func NewEngine[T chasm.Component](
+func NewEngine[T chasm.RootComponent](
 	t *testing.T,
 	registry *chasm.Registry,
 	opts ...Option[T],
@@ -73,7 +73,7 @@ func NewEngine[T chasm.Component](
 	return e
 }
 
-func WithRoot[T chasm.Component](
+func WithRoot[T chasm.RootComponent](
 	factory func(chasm.MutableContext) T,
 ) Option[T] {
 	return func(e *Engine[T]) {
@@ -90,7 +90,7 @@ func WithRoot[T chasm.Component](
 	}
 }
 
-func WithExecutionKey[T chasm.Component](key chasm.ExecutionKey) Option[T] {
+func WithExecutionKey[T chasm.RootComponent](key chasm.ExecutionKey) Option[T] {
 	return func(e *Engine[T]) {
 		e.rootExecutionKey = key
 	}

chasm/lib/callback/tasks_test.go

@@ -55,6 +55,17 @@ func (m *mockNexusCompletionGetterComponent) LifecycleState(_ chasm.Context) cha
 	return chasm.LifecycleStateRunning
 }
 
+func (m *mockNexusCompletionGetterComponent) ContextMetadata(chasm.Context) map[string]string {
+	return nil
+}
+
+func (m *mockNexusCompletionGetterComponent) Terminate(
+	chasm.MutableContext,
+	chasm.TerminateComponentRequest,
+) (chasm.TerminateComponentResponse, error) {
+	return chasm.TerminateComponentResponse{}, nil
+}
+
 type mockNexusCompletionGetterLibrary struct {
 	chasm.UnimplementedLibrary
 }