package taskboard

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/tidbcloud/AgentTeam/internal/store/message"
)

func TestTaskSourceContributionSourceUsesCreatorProfile(t *testing.T) {
	thread := &TaskSourceThread{
		ParentChannelID: "chn_source",
		ParentMessageID: "msg_source_parent",
	}

	source := taskSourceContributionSource(thread, &message.AuthorProfile{
		ActorType:   message.AuthorActorTypeAgent,
		PrincipalID: "agt_creator",
		DisplayName: "Creator Agent",
	})

	require.Equal(t, message.ContributionSourceKindAgentRuntime, source.Kind)
	require.Equal(t, message.ContributionSourceRefTypeMessage, source.RefType)
	require.Equal(t, "msg_source_parent", source.RefID)
}