package postgres

import (
	"context"
	"errors"
	"testing"

	"cryptoHermes/internal/entity"

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

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

	mock.ExpectBegin()
	mock.ExpectExec("INSERT INTO taker_buy_sell_volume").
		WithArgs("binance", "BTCUSDT", "1h", int64(1), "1.2", "60", "50").
		WillReturnResult(pgxmock.NewResult("INSERT", 1))
	mock.ExpectCommit()
	mock.ExpectRollback()

	r := &TakerVolumeRepo{pool: mock}
	err = r.UpsertMany(context.Background(), []entity.TakerBuySellVolume{{
		Source: "binance", Symbol: "BTCUSDT", Period: "1h",
		Timestamp: 1, BuySellRatio: "1.2", BuyVolume: "60", SellVolume: "50",
	}})
	require.NoError(t, err)
	require.NoError(t, mock.ExpectationsWereMet())
}

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

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

	r := &TakerVolumeRepo{pool: mock}
	err = r.UpsertMany(context.Background(), []entity.TakerBuySellVolume{
		{Timestamp: 0, BuyVolume: "60"},
	})
	require.NoError(t, err)
	require.NoError(t, mock.ExpectationsWereMet())
}

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

	mock.ExpectBegin()
	mock.ExpectExec("INSERT INTO taker_buy_sell_volume").
		WithArgs(pgxmock.AnyArg(), pgxmock.AnyArg(), pgxmock.AnyArg(), pgxmock.AnyArg(),
			pgxmock.AnyArg(), pgxmock.AnyArg(), pgxmock.AnyArg()).
		WillReturnError(errors.New("readonly"))
	mock.ExpectRollback()

	r := &TakerVolumeRepo{pool: mock}
	err = r.UpsertMany(context.Background(), []entity.TakerBuySellVolume{{
		Symbol: "BTCUSDT", Period: "1h", Timestamp: 1,
	}})
	require.Error(t, err)
	require.Contains(t, err.Error(), "upsert taker")
	require.NoError(t, mock.ExpectationsWereMet())
}

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

	rows := mock.NewRows([]string{"source", "symbol", "period", "timestamp",
		"buy_sell_ratio", "buy_volume", "sell_volume"}).
		AddRow("binance", "BTCUSDT", "1h", int64(2), "1.3", "65", "50").
		AddRow("binance", "BTCUSDT", "1h", int64(1), "1.2", "60", "50")

	mock.ExpectQuery("SELECT source, symbol, period").
		WithArgs("BTCUSDT", "1h", 200).
		WillReturnRows(rows)

	r := &TakerVolumeRepo{pool: mock}
	got, err := r.FindRecent(context.Background(), "BTCUSDT", "1h", 200)
	require.NoError(t, err)
	require.Len(t, got, 2)
	require.Equal(t, int64(1), got[0].Timestamp)
	require.Equal(t, "65", got[1].BuyVolume)
	require.NoError(t, mock.ExpectationsWereMet())
}

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

	mock.ExpectQuery("SELECT source, symbol, period").
		WithArgs("BTCUSDT", "1h", 100).
		WillReturnError(errors.New("deadlock"))

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