xinmi/internal/application/service/knowledge_summary_refresh_test.go

package service

import (
	"context"
	"errors"
	"testing"

	"github.com/Tencent/WeKnora/internal/types"
	"github.com/Tencent/WeKnora/internal/types/interfaces"
)

type summaryRefreshTenantRepo struct {
	interfaces.TenantRepository
	tenant      *types.Tenant
	err         error
	requestedID uint64
}

func (r *summaryRefreshTenantRepo) GetTenantByID(_ context.Context, id uint64) (*types.Tenant, error) {
	r.requestedID = id
	return r.tenant, r.err
}

func TestRestoreSummaryRefreshTenantInfo(t *testing.T) {
	repo := &summaryRefreshTenantRepo{tenant: &types.Tenant{ID: 42}}
	ctx, err := restoreSummaryRefreshTenantInfo(context.Background(), repo, 42)
	if err != nil {
		t.Fatalf("restore tenant context: %v", err)
	}
	if repo.requestedID != 42 {
		t.Fatalf("requested tenant ID = %d, want 42", repo.requestedID)
	}
	tenant, ok := types.TenantInfoFromContext(ctx)
	if !ok || tenant.ID != 42 {
		t.Fatalf("tenant context = %#v, %v; want tenant 42", tenant, ok)
	}
}

func TestRestoreSummaryRefreshTenantInfoRejectsMissingTenant(t *testing.T) {
	tests := []struct {
		name string
		repo interfaces.TenantRepository
	}{
		{name: "repository unavailable"},
		{name: "tenant lookup failed", repo: &summaryRefreshTenantRepo{err: errors.New("lookup failed")}},
		{name: "tenant not found", repo: &summaryRefreshTenantRepo{}},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			ctx, err := restoreSummaryRefreshTenantInfo(context.Background(), tt.repo, 42)
			if err == nil {
				t.Fatal("expected an error")
			}
			if _, ok := types.TenantInfoFromContext(ctx); ok {
				t.Fatal("tenant info should not be added after a failed lookup")
			}
		})
	}
}

func TestErrSummaryRefreshStaleIsDistinct(t *testing.T) {
	if ErrSummaryRefreshStale == nil {
		t.Fatal("expected stale refresh sentinel")
	}
	if ErrSummaryRefreshStale.Error() == "" {
		t.Fatal("expected non-empty stale refresh message")
	}
}