package converter

import (
	"testing"
	"time"

	"github.com/stretchr/testify/require"
	"github.com/tidbcloud/AgentTeam/internal/store/kb"
)

func TestToKBSourceIncludesRepositoryAccessProjection(t *testing.T) {
	now := time.Now().UTC()
	converted := ToKBSource(kb.Source{
		ID:                              "src_1",
		KnowledgeBaseID:                 "kb_1",
		SourceType:                      "codebase",
		Name:                            "TiDB",
		SourceRepositoryAccessProfile:   "write",
		SourceRepositoryAccessStatus:    "current",
		InsightsRepositoryAccessProfile: "read",
		InsightsRepositoryAccessStatus:  "current",
		CreatedAt:                       now,
		UpdatedAt:                       now,
	})

	require.Equal(t, "write", converted.SourceRepositoryAccessProfile)
	require.Equal(t, "current", converted.SourceRepositoryAccessStatus)
	require.Equal(t, "read", converted.InsightsRepositoryAccessProfile)
	require.Equal(t, "current", converted.InsightsRepositoryAccessStatus)
}