package host

import (
	"context"
	"regexp"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
	"hyapp/pkg/appcode"
	hostdomain "hyapp/services/user-service/internal/domain/host"
)

func TestGetHostEngagementStatsAggregatesOwnedFacts(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("create sqlmock: %v", err)
	}
	defer db.Close()

	mock.ExpectQuery("FROM user_mic_daily_stats").
		WithArgs(validMicDayThresholdMS, "fami", int64(42), "2026-07-06", "2026-07-12").
		WillReturnRows(sqlmock.NewRows([]string{"mic_ms", "valid_days"}).AddRow(int64(5400000), int64(1)))
	mock.ExpectQuery("FROM user_private_message_events").
		WithArgs("fami", int64(42), int64(1783296000000), int64(1783900800000)).
		WillReturnRows(sqlmock.NewRows([]string{"senders"}).AddRow(int64(6)))
	mock.ExpectQuery("FROM user_follow_events").
		WithArgs("fami", int64(42), int64(1783296000000), int64(1783900800000)).
		WillReturnRows(sqlmock.NewRows([]string{"followers"}).AddRow(int64(4)))
	mock.ExpectQuery(regexp.QuoteMeta(`
		SELECT created_at_ms, last_heartbeat_at_ms
		FROM auth_sessions
		WHERE app_code = ? AND user_id = ?
		  AND last_heartbeat_at_ms >= ? AND created_at_ms < ?`)).
		WithArgs("fami", int64(42), int64(1783296000000), int64(1783900800000)).
		WillReturnRows(sqlmock.NewRows([]string{"created_at_ms", "last_heartbeat_at_ms"}).
			AddRow(int64(1783299600000), int64(1783303200000)).
			AddRow(int64(1783301400000), int64(1783306800000)))

	repo := New(db)
	stats, err := repo.GetHostEngagementStats(appcode.WithContext(context.Background(), "fami"), hostdomain.EngagementStatsQuery{
		HostUserID: 42,
		StartAtMS:  1783296000000,
		EndAtMS:    1783900800000,
	})
	if err != nil {
		t.Fatalf("GetHostEngagementStats failed: %v", err)
	}
	// 两个重叠 session 按当天最早开始到最晚心跳合并，不能相加成重复在线时长。
	if stats.OnlineDurationMS != 7200000 || stats.ValidMicDurationMS != 5400000 || stats.ValidMicDays != 1 || stats.PrivateMessageSenders != 6 || stats.NewFollowers != 4 {
		t.Fatalf("unexpected engagement stats: %+v", stats)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("unmet SQL expectations: %v", err)
	}
}