wip: fix setup for tool mokapi_execute_code

Author: marle3003

mcp/run.go

@@ -72,7 +72,7 @@ Use this tool to:
 Prefer this tool over retrieving full API specifications, as it returns only the computed result.`,
 		InputSchema:  inputSchema,
 		OutputSchema: outputSchema,
-	}, s.GenerateHttpMockResponse)
+	}, s.GetRunResponse)
 
 	server.AddResource(&mcp.Resource{
 		URI:  "api://execute-types",

mcp/run_test.go

@@ -22,6 +22,22 @@ func TestService_Run(t *testing.T) {
 		app  *runtime.App
 		test func(t *testing.T, s *mcp.Service)
 	}{
+		{
+			name: "run JavaScript code",
+			app: runtimetest.NewHttpApp(
+				openapitest.NewConfig("3.1.0"),
+			),
+			test: func(t *testing.T, s *mcp.Service) {
+				r, err := s.GetRunResponse(
+					context.Background(),
+					mcp.RunInput{
+						Code: `1+1`,
+					},
+				)
+				require.NoError(t, err)
+				require.Equal(t, int64(2), r.Result)
+			},
+		},
 		{
 			name: "List APIs skip empty name",
 			app: runtimetest.NewHttpApp(

mcp/server_test.go

@@ -29,17 +29,69 @@ func TestServer(t *testing.T) {
 	require.NotNil(t, session)
 	defer func() { _ = session.Close() }()
 
-	list, err := session.ListTools(ctx, &gomcp.ListToolsParams{})
-	require.NoError(t, err)
-	require.Len(t, list.Tools, 9)
-	// alphabetical order
-	require.Equal(t, "mokapi_execute_code", list.Tools[0].Name)
-	require.Equal(t, "mokapi_generate_http_mock_response", list.Tools[1].Name)
-	require.Equal(t, "mokapi_get_api_spec", list.Tools[2].Name)
-	require.Equal(t, "mokapi_get_events", list.Tools[3].Name)
-	require.Equal(t, "mokapi_get_http_mock_template", list.Tools[4].Name)
-	require.Equal(t, "mokapi_get_scenarios", list.Tools[5].Name)
-	require.Equal(t, "mokapi_get_typescript_api", list.Tools[6].Name)
-	require.Equal(t, "mokapi_produce_kafka_message", list.Tools[7].Name)
-	require.Equal(t, "mokapi_send_http_request", list.Tools[8].Name)
+	testcases := []struct {
+		name string
+		test func(t *testing.T)
+	}{
+		{
+			name: "test tool list",
+			test: func(t *testing.T) {
+				list, err := session.ListTools(ctx, &gomcp.ListToolsParams{})
+				require.NoError(t, err)
+				require.Len(t, list.Tools, 9)
+				// alphabetical order
+				require.Equal(t, "mokapi_execute_code", list.Tools[0].Name)
+				require.Equal(t, "mokapi_generate_http_mock_response", list.Tools[1].Name)
+				require.Equal(t, "mokapi_get_api_spec", list.Tools[2].Name)
+				require.Equal(t, "mokapi_get_events", list.Tools[3].Name)
+				require.Equal(t, "mokapi_get_http_mock_template", list.Tools[4].Name)
+				require.Equal(t, "mokapi_get_scenarios", list.Tools[5].Name)
+				require.Equal(t, "mokapi_get_typescript_api", list.Tools[6].Name)
+				require.Equal(t, "mokapi_produce_kafka_message", list.Tools[7].Name)
+				require.Equal(t, "mokapi_send_http_request", list.Tools[8].Name)
+			},
+		},
+		{
+			name: "run code",
+			test: func(t *testing.T) {
+				r, err := session.CallTool(ctx, &gomcp.CallToolParams{
+					Name: "mokapi_execute_code",
+					Arguments: mcp.RunInput{
+						Code: "1+1",
+					},
+				})
+				require.NoError(t, err)
+				require.IsType(t, &gomcp.TextContent{}, r.Content[0])
+				tc := r.Content[0].(*gomcp.TextContent)
+				require.Equal(t, `{"result":2}`, tc.Text)
+			},
+		},
+		{
+			name: "get mokapi_execute_code tool information",
+			test: func(t *testing.T) {
+				list, err := session.ListTools(ctx, &gomcp.ListToolsParams{
+					Meta:   nil,
+					Cursor: "",
+				})
+				require.NoError(t, err)
+				require.Contains(t, list.Tools[0].Description, "api://execute-types")
+			},
+		},
+		{
+			name: "get resource execute-types",
+			test: func(t *testing.T) {
+				list, err := session.ListResources(ctx, &gomcp.ListResourcesParams{})
+				require.NoError(t, err)
+				require.Equal(t, "api-docs", list.Resources[0].Name)
+				require.Equal(t, "api://execute-types", list.Resources[0].URI)
+			},
+		},
+	}
+
+	for _, tc := range testcases {
+		t.Run(tc.name, func(t *testing.T) {
+			tc.test(t)
+		})
+	}
+
 }