Carlos Florêncio
34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package promptcontext
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/kandev/kandev/internal/sysprompt"
|
|
"github.com/kandev/kandev/internal/task/models"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBuildPullRequestTargets(t *testing.T) {
|
|
links := []*models.TaskRepository{
|
|
{RepositoryID: "repo-1", BaseBranch: "develop", CheckoutBranch: "feature/task", BranchPolicyID: "policy-1", BranchPolicyPullRequestTarget: "main"},
|
|
{RepositoryID: "repo-2", BaseBranch: "main", BranchPolicyID: "policy-2", BranchPolicyPullRequestTarget: "release"},
|
|
{RepositoryID: "repo-3", BranchPolicyID: "policy-3"},
|
|
{RepositoryID: "repo-4", BranchPolicyPullRequestTarget: "main"},
|
|
nil,
|
|
}
|
|
|
|
targets := BuildPullRequestTargets(context.Background(), links, func(_ context.Context, repositoryID string) (string, error) {
|
|
if repositoryID == "repo-2" {
|
|
return "", errors.New("repository unavailable")
|
|
}
|
|
return "repo-1-name", nil
|
|
})
|
|
|
|
require.Equal(t, []sysprompt.PullRequestTarget{
|
|
{RepositoryName: "repo-1-name", WorkingBranch: "feature/task", TargetBranch: "main"},
|
|
{RepositoryName: "repo-2", WorkingBranch: "main", TargetBranch: "release"},
|
|
}, targets)
|
|
}
|