package postgres

import (
	"context"
	"errors"
	"testing"

	"cryptoHermes/internal/entity"

	"github.com/pashagolub/pgxmock/v4"
	"github.com/stretchr/testify/require"
)

func TestFundingRepo_UpsertMany_HappyPath(t *testing.T) {
	mock, err := pgxmock.NewPool()
	require.NoError(t, err)
	defer mock.Close()

	mock.ExpectBegin()
	mock.ExpectExec("INSERT INTO funding_rates").
		WithArgs("binance", "BTCUSDT", int64(1700000000000), "0.0001", "50000").
		WillReturnResult(pgxmock.NewResult("INSERT", 1))
	mock.ExpectCommit()
	mock.ExpectRollback()

	r := &FundingRepo{pool: mock}
	err = r.UpsertMany(context.Background(), []entity.FundingRate{{
		Source: "binance", Symbol: "BTCUSDT",
		FundingTime: 1700000000000, FundingRate: "0.0001", MarkPrice: "50000",
	}})
	require.NoError(t, err)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestFundingRepo_UpsertMany_SkipsEmptyRate(t *testing.T) {
	mock, err := pgxmock.NewPool()
	require.NoError(t, err)
	defer mock.Close()

	mock.ExpectBegin()
	mock.ExpectCommit()
	mock.ExpectRollback()

	r := &FundingRepo{pool: mock}
	err = r.UpsertMany(context.Background(), []entity.FundingRate{
		{Symbol: "BTCUSDT", FundingTime: 1, FundingRate: ""}, // 空 rate 跳过
		{Symbol: "BTCUSDT", FundingTime: 0, FundingRate: "1"}, // 空时间戳跳过
	})
	require.NoError(t, err)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestFundingRepo_UpsertMany_ExecError(t *testing.T) {
	mock, err := pgxmock.NewPool()
	require.NoError(t, err)
	defer mock.Close()

	mock.ExpectBegin()
	mock.ExpectExec("INSERT INTO funding_rates").
		WithArgs(pgxmock.AnyArg(), pgxmock.AnyArg(), pgxmock.AnyArg(), pgxmock.AnyArg(), pgxmock.AnyArg()).
		WillReturnError(errors.New("dup key"))
	mock.ExpectRollback()

	r := &FundingRepo{pool: mock}
	err = r.UpsertMany(context.Background(), []entity.FundingRate{{
		Symbol: "BTCUSDT", FundingTime: 1, FundingRate: "0.0001",
	}})
	require.Error(t, err)
	require.Contains(t, err.Error(), "upsert funding")
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestFundingRepo_FindRecent_HappyPath(t *testing.T) {
	mock, err := pgxmock.NewPool()
	require.NoError(t, err)
	defer mock.Close()

	rows := mock.NewRows([]string{"source", "symbol", "funding_time", "funding_rate", "mark_price"}).
		AddRow("binance", "BTCUSDT", int64(2), "0.0002", "50100").
		AddRow("binance", "BTCUSDT", int64(1), "0.0001", "50000")

	mock.ExpectQuery("SELECT source, symbol, funding_time").
		WithArgs("BTCUSDT", 100).
		WillReturnRows(rows)

	r := &FundingRepo{pool: mock}
	got, err := r.FindRecent(context.Background(), "BTCUSDT", 0) // 0 → 100
	require.NoError(t, err)
	require.Len(t, got, 2)
	// 反序后升序：[1,2]
	require.Equal(t, int64(1), got[0].FundingTime)
	require.Equal(t, "50100", got[1].MarkPrice)
	require.NoError(t, mock.ExpectationsWereMet())
}

func TestFundingRepo_FindRecent_QueryError(t *testing.T) {
	mock, err := pgxmock.NewPool()
	require.NoError(t, err)
	defer mock.Close()

	mock.ExpectQuery("SELECT source, symbol, funding_time").
		WithArgs("BTCUSDT", 50).
		WillReturnError(errors.New("conn closed"))

	r := &FundingRepo{pool: mock}
	_, err = r.FindRecent(context.Background(), "BTCUSDT", 50)
	require.Error(t, err)
	require.NoError(t, mock.ExpectationsWereMet())
}