Tea-Cli/tests/integration/task_issue_test.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package integration

import (
	"context"
	"fmt"
	"testing"
	"time"

	gitea "gitea.dev/sdk"
	"gitea.dev/tea/cmd"
	"github.com/stretchr/testify/require"
)

func TestEditIssue_ModifiesAssignees(t *testing.T) {
	// This test verifies that EditIssue correctly modifies assignees of an issue via set, add, and remove.
	// It sets up a test repository and organization with known users, then calls EditIssue and checks the results.
	login := createIntegrationLogin(t)
	client := login.Client()
	orgName := fmt.Sprintf("issue-org-%d", time.Now().UnixNano()%1_000_000)
	orgRepoName := fmt.Sprintf("issue-repo-%d", time.Now().UnixNano()%1_000_000)
	ctx := context.Background()

	// Clean up any existing test data that might interfere with the test.
	_, _ = client.Repositories.DeleteRepo(ctx, orgName, orgRepoName)
	_, _ = client.Organizations.DeleteOrg(ctx, orgName)
	_, _ = client.Admin.DeleteUser(ctx, "user1")
	_, _ = client.Admin.DeleteUser(ctx, "user2")

	_, _, err := client.Admin.CreateOrg(ctx, integrationUsername, gitea.CreateOrgOption{Name: orgName})
	require.NoError(t, err)
	t.Cleanup(func() {
		if _, delErr := client.Organizations.DeleteOrg(ctx, orgName); delErr != nil {
			t.Logf("failed to delete integration test org %q: %v", orgName, delErr)
		}
	})

	orgRepo, _, err := client.Repositories.CreateOrgRepo(ctx, orgName, gitea.CreateRepoOption{Name: orgRepoName})
	require.NoError(t, err)
	t.Cleanup(func() {
		if _, delErr := client.Repositories.DeleteRepo(ctx, orgName, orgRepoName); delErr != nil {
			t.Logf("failed to delete integration test repo %q: %v", orgRepoName, delErr)
		}
	})

	user1, _, err := client.Admin.CreateUser(ctx, gitea.CreateUserOption{Username: "user1", Password: "user1!1234", Email: "user1@test.com"})
	require.NoError(t, err)
	user2, _, err := client.Admin.CreateUser(ctx, gitea.CreateUserOption{Username: "user2", Password: "user2!1234", Email: "user2@test.com"})
	require.NoError(t, err)
	t.Cleanup(func() {
		_, _ = client.Admin.DeleteUser(ctx, "user1")
		_, _ = client.Admin.DeleteUser(ctx, "user2")
	})

	permission := gitea.AccessModeOwner
	team, _, err := client.Organizations.CreateTeam(ctx, orgName, gitea.CreateTeamOption{Name: "writers", Permission: permission})
	require.NoError(t, err)
	_, err = client.Organizations.AddTeamMember(ctx, team.ID, "user1")
	require.NoError(t, err)
	_, err = client.Organizations.AddTeamMember(ctx, team.ID, "user2")
	require.NoError(t, err)
	_, err = client.Organizations.AddTeamRepository(ctx, team.ID, orgName, orgRepoName)
	require.NoError(t, err)
	assigneeValid, _, err := client.Repositories.CheckRepoIssueAssignee(ctx, orgName, orgRepoName, user1.UserName)
	require.NoError(t, err)
	require.True(t, assigneeValid)
	assigneeValid, _, err = client.Repositories.CheckRepoIssueAssignee(ctx, orgName, orgRepoName, user2.UserName)
	require.NoError(t, err)
	require.True(t, assigneeValid)

	orgIssue, _, err := client.Issues.CreateIssue(ctx, orgName, orgRepoName, gitea.CreateIssueOption{Title: "issue_integration_test", Assignees: []string{integrationUsername}, Closed: false})
	require.NoError(t, err)
	require.Equal(t, integrationUsername, orgIssue.Assignees[0].UserName)
	curUser, _, err := client.Users.GetMyUserInfo(ctx)
	require.NoError(t, err)

	getNames := func(issue *gitea.Issue) []string {
		names := make([]string, len(issue.Assignees))
		for i, u := range issue.Assignees {
			names[i] = u.UserName
		}
		return names
	}

	checkAssignees := func(expected []string) {
		updatedIssue, _, err := client.Issues.GetIssue(ctx, orgName, orgRepoName, orgIssue.Index)
		require.NoError(t, err)
		require.ElementsMatch(t, getNames(updatedIssue), expected)
	}

	app := cmd.App()
	// test set overwrites add and remove
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--set-assignees", "user1", "--add-assignees", "user2", "--remove-assignees", integrationUsername, "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{user1.UserName})
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--set-assignees", "user2," + integrationUsername, "--add-assignees", "user1", "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{user2.UserName, curUser.UserName})
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--set-assignees", "user1", "--remove-assignees", integrationUsername, "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{user1.UserName})

	// test remove one assignee
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--remove-assignees", "user1", "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{})

	// test add multiple assignees and overwrites remove
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--add-assignees", "user1," + integrationUsername, "--remove-assignees", integrationUsername, "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{curUser.UserName, user1.UserName})

	// test add one assignee
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--add-assignees", "user2", "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{curUser.UserName, user1.UserName, user2.UserName})

	// test remove multiple assignees
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--remove-assignees", "user1,user2," + integrationUsername, "--repo", orgRepo.FullName})
	require.NoError(t, err)
	checkAssignees([]string{})

	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--set-assignees", "user_not_exists"})
	require.Error(t, err)
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--add-assignees", "user_not_exists"})
	require.Error(t, err)
	err = app.Run(ctx, []string{"tea", "i", "edit", "1", "--remove-assignees", "user_not_exists"})
	require.Error(t, err)
}