1  // Copyright 2025 Alibaba Group Holding Ltd.
 2  //
 3  // Licensed under the Apache License, Version 2.0 (the "License");
 4  // you may not use this file except in compliance with the License.
 5  // You may obtain a copy of the License at
 6  //
 7  //     http://www.apache.org/licenses/LICENSE-2.0
 8  //
 9  // Unless required by applicable law or agreed to in writing, software
10  // distributed under the License is distributed on an "AS IS" BASIS,
11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  // See the License for the specific language governing permissions and
13  // limitations under the License.
14  
15  package runtime
16  
17  import (
18  	"reflect"
19  	"testing"
20  
21  	"github.com/stretchr/testify/require"
22  )
23  
24  func TestExecuteCodeRequest_SetDefaultHooks(t *testing.T) {
25  	customResult := func(map[string]any, int) {}
26  
27  	req := &ExecuteCodeRequest{
28  		Hooks: ExecuteResultHook{
29  			OnExecuteResult: customResult,
30  		},
31  	}
32  
33  	req.SetDefaultHooks()
34  
35  	require.NotNil(t, req.Hooks.OnExecuteStdout)
36  	require.NotNil(t, req.Hooks.OnExecuteStderr)
37  	require.NotNil(t, req.Hooks.OnExecuteError)
38  	require.NotNil(t, req.Hooks.OnExecuteResult, "expected OnExecuteResult to remain set")
39  	require.Equal(t, reflect.ValueOf(customResult).Pointer(), reflect.ValueOf(req.Hooks.OnExecuteResult).Pointer(),
40  		"default hooks should not override existing ones")
41  }