EPMRPP-109176 || MCP. Tools. Add tool, update defect types by item ids

Author: AleksandrParamonoff

cmd/reportportal-mcp-server/main.go

@@ -59,6 +59,7 @@ func main() {
 			Name:     "project",
 			Required: false,
 			Sources:  cli.EnvVars("RP_PROJECT"),
+			Value:    "",
 			Usage:    "ReportPortal project name",
 		},
 		&cli.StringFlag{
@@ -238,6 +239,7 @@ func newMCPServer(cmd *cli.Command) (*server.MCPServer, *mcpreportportal.Analyti
 	token := cmd.String("token")                           // API token
 	host := cmd.String("rp-host")                          // ReportPortal host URL
 	userID := cmd.String("user-id")                        // Unified user ID for analytics
+	project := cmd.String("project")                       // ReportPortal project name
 	analyticsAPISecret := mcpreportportal.GetAnalyticArg() // Analytics API secret
 	analyticsOff := cmd.Bool("analytics-off")              // Disable analytics flag
 
@@ -252,6 +254,7 @@ func newMCPServer(cmd *cli.Command) (*server.MCPServer, *mcpreportportal.Analyti
 		hostUrl,
 		token,
 		userID,
+		project,
 		analyticsAPISecret,
 		!analyticsOff, // Convert analyticsOff to analyticsOn
 	)

internal/reportportal/http_server.go

@@ -140,7 +140,7 @@ func (hs *HTTPServer) initializeTools() error {
 	rpClient.APIClient.GetConfig().Middleware = QueryParamsMiddleware
 
 	// Add launch management tools with analytics
-	launches := NewLaunchResources(rpClient, hs.analytics)
+	launches := NewLaunchResources(rpClient, hs.analytics, "")
 
 	hs.mcpServer.AddTool(launches.toolGetLaunches())
 	hs.mcpServer.AddTool(launches.toolGetLastLaunchByName())
@@ -153,7 +153,7 @@ func (hs *HTTPServer) initializeTools() error {
 	hs.mcpServer.AddResourceTemplate(launches.resourceLaunch())
 
 	// Add test item tools
-	testItems := NewTestItemResources(rpClient, hs.analytics)
+	testItems := NewTestItemResources(rpClient, hs.analytics, "")
 
 	hs.mcpServer.AddTool(testItems.toolGetTestItemById())
 	hs.mcpServer.AddTool(testItems.toolGetTestItemsByFilter())

internal/reportportal/items.go

@@ -12,6 +12,7 @@ import (
 	"github.com/mark3labs/mcp-go/mcp"
 	"github.com/mark3labs/mcp-go/server"
 	"github.com/reportportal/goRP/v5/pkg/gorp"
+	"github.com/reportportal/goRP/v5/pkg/openapi"
 	"github.com/yosida95/uritemplate/v3"
 )
 
@@ -25,11 +26,13 @@ type TestItemResources struct {
 func NewTestItemResources(
 	client *gorp.Client,
 	analytics *Analytics,
+	project string,
 ) *TestItemResources {
 	return &TestItemResources{
 		client: client,
 		projectParameter: mcp.WithString("project", // Parameter for specifying the project name)
 			mcp.Description("Project name"),
+			mcp.DefaultString(project),
 		),
 		analytics: analytics,
 	}
@@ -626,7 +629,8 @@ func (lr *TestItemResources) toolGetProjectDefectTypes() (mcp.Tool, server.ToolH
 			"get_project_defect_types",
 			// Tool metadata
 			mcp.WithDescription(
-				"Get all defect types for a specific project, returns a JSON which contains a list of defect types in the 'configuration/subtypes' array",
+				"Get all defect types for a specific project, returns a JSON which contains a list of defect types in the 'configuration/subtypes' array and represents the defect type ID. "+
+					"Example: { \"NO_DEFECT\": { \"locator\": \"nd001\" } NO_DEFECT - defect type name, nd001 - defect type unique id }, ",
 			),
 			lr.projectParameter,
 		), lr.analytics.WithAnalytics("get_project_defect_types", func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
@@ -646,3 +650,69 @@ func (lr *TestItemResources) toolGetProjectDefectTypes() (mcp.Tool, server.ToolH
 			return readResponseBody(response)
 		})
 }
+
+// toolUpdateTestItemDefectType creates a tool to update the defect type for a specific test items.
+func (lr *TestItemResources) toolUpdateTestItemDefectType() (mcp.Tool, server.ToolHandlerFunc) {
+	return mcp.NewTool("update_test_item_defect_type",
+			// Tool metadata
+			mcp.WithDescription("Update test item defect type"),
+			lr.projectParameter,
+			mcp.WithArray("test_item_ids", // Parameter for specifying the array of test item IDs
+				mcp.Description("Array of test item IDs"),
+				mcp.Required(),
+			),
+			mcp.WithString(
+				"defect_type_id", // Parameter for specifying the defect type ID
+				mcp.Description(
+					"Defect Type ID, all possible values can be received from the tool 'get_project_defect_types', \"locator\" is a unique identifier of the defect type. "+
+						"Example: \"NO_DEFECT\": { \"locator\": \"nd001\" } NO_DEFECT - defect type name, nd001 - defect type unique id",
+				),
+				mcp.Required(),
+			),
+		), lr.analytics.WithAnalytics("update_test_item_defect_type", func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+			project, err := extractProject(ctx, request)
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			// Extract the "defect_type_id" parameter from the request
+			defectTypeId, err := request.RequireString("defect_type_id")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			// Extract the "test_item_ids" parameter from the request
+			testItemIdStrs, err := request.RequireStringSlice("test_item_ids")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			// Build the list of issues
+			issues := make([]openapi.IssueDefinition, 0, len(testItemIdStrs))
+			for _, testItemIdStr := range testItemIdStrs {
+				testItemId, err := strconv.ParseInt(testItemIdStr, 10, 64)
+				if err != nil {
+					return mcp.NewToolResultError(err.Error()), nil
+				}
+				issues = append(issues, openapi.IssueDefinition{
+					TestItemId: testItemId,
+					Issue: openapi.Issue{
+						IssueType: defectTypeId,
+					},
+				})
+			}
+
+			apiRequest := lr.client.TestItemAPI.DefineTestItemIssueType(ctx, project).
+				DefineIssueRQ(openapi.DefineIssueRQ{
+					Issues: issues,
+				})
+
+			// Execute the request
+			_, response, err := apiRequest.Execute()
+			if err != nil {
+				return mcp.NewToolResultError(extractResponseError(err, response)), nil
+			}
+
+			// Return the serialized testItem as a text result
+			return readResponseBody(response)
+		})
+}

internal/reportportal/launches.go

@@ -25,11 +25,13 @@ type LaunchResources struct {
 func NewLaunchResources(
 	client *gorp.Client,
 	analytics *Analytics,
+	project string,
 ) *LaunchResources {
 	return &LaunchResources{
 		client: client,
 		projectParameter: mcp.WithString("project", // Parameter for specifying the project name)
 			mcp.Description("Project name"),
+			mcp.DefaultString(project),
 		),
 		analytics: analytics,
 	}

internal/reportportal/launches_test.go

@@ -43,7 +43,7 @@ func TestListLaunchesTool(t *testing.T) {
 	srv := mcptest.NewUnstartedServer(t)
 
 	serverURL, _ := url.Parse(mockServer.URL)
-	launchTools := NewLaunchResources(gorp.NewClient(serverURL, ""), nil)
+	launchTools := NewLaunchResources(gorp.NewClient(serverURL, ""), nil, "")
 	srv.AddTool(launchTools.toolGetLaunches())
 
 	err := srv.Start(ctx)

internal/reportportal/server.go

@@ -20,7 +20,7 @@ func NewServer(
 	version string,
 	hostUrl *url.URL,
 	token,
-	userID, analyticsAPISecret string,
+	userID, project, analyticsAPISecret string,
 	analyticsOn bool,
 ) (*server.MCPServer, *Analytics, error) {
 	s := server.NewMCPServer(
@@ -47,7 +47,7 @@ func NewServer(
 		}
 	}
 
-	launches := NewLaunchResources(rpClient, analytics)
+	launches := NewLaunchResources(rpClient, analytics, project)
 	s.AddTool(launches.toolGetLaunches())
 	s.AddTool(launches.toolGetLastLaunchByName())
 	s.AddTool(launches.toolForceFinishLaunch())
@@ -57,13 +57,14 @@ func NewServer(
 	s.AddTool(launches.toolRunQualityGate())
 	s.AddResourceTemplate(launches.resourceLaunch())
 
-	testItems := NewTestItemResources(rpClient, analytics)
+	testItems := NewTestItemResources(rpClient, analytics, project)
 	s.AddTool(testItems.toolGetTestItemById())
 	s.AddTool(testItems.toolGetTestItemsByFilter())
 	s.AddTool(testItems.toolGetTestItemLogsByFilter())
 	s.AddTool(testItems.toolGetTestItemAttachment())
 	s.AddTool(testItems.toolGetTestSuitesByFilter())
 	s.AddTool(testItems.toolGetProjectDefectTypes())
+	s.AddTool(testItems.toolUpdateTestItemDefectType())
 	s.AddResourceTemplate(testItems.resourceTestItem())
 
 	prompts, err := readPrompts(promptFiles, "prompts")