package uniswap

import (
	"math/big"
	"testing"

	"github.com/holiman/uint256"
	"github.com/stretchr/testify/assert"
)

func TestRoundTripConversions(t *testing.T) {
	// Test sqrtPriceX96 -> price -> sqrtPriceX96 round trip
	sqrtPriceX96 := new(big.Int)
	sqrtPriceX96.SetString("79228162514264337593543950336", 10) // 2^96 (price = 1.0)

	price := SqrtPriceX96ToPrice(sqrtPriceX96)
	resultSqrtPriceX96 := PriceToSqrtPriceX96(price)

	// Allow for small differences due to floating point precision
	diff := new(big.Int).Sub(sqrtPriceX96, resultSqrtPriceX96)
	assert.True(t, diff.Cmp(big.NewInt(1000000000000)) < 0, "Round trip conversion should be accurate")

	// Test tick -> sqrtPriceX96 -> tick round trip
	tick := 100000
	sqrtPrice := TickToSqrtPriceX96(tick)
	resultTick := SqrtPriceX96ToTick(sqrtPrice)

	// Allow for small differences due to floating point precision
	assert.InDelta(t, tick, resultTick, 1, "Round trip tick conversion should be accurate")
}

func TestGetTickAtSqrtPriceWithUint256(t *testing.T) {
	// Test with a known value
	bigInt := new(big.Int)
	bigInt.SetString("79228162514264337593543950336", 10) // 2^96
	sqrtPriceX96, _ := uint256.FromBig(bigInt)

	tick := GetTickAtSqrtPrice(sqrtPriceX96)
	expectedTick := 0 // sqrtPriceX96 = 2^96 corresponds to price = 1.0, which is tick 0

	assert.Equal(t, expectedTick, tick, "GetTickAtSqrtPrice should return correct tick")
}

func TestTickSpacingCalculations(t *testing.T) {
	currentTick := 100000

	// Test with medium tick spacing (60)
	nextTick := GetNextTick(currentTick, MediumTickSpacing)
	previousTick := GetPreviousTick(currentTick, MediumTickSpacing)

	assert.Equal(t, 100020, nextTick, "GetNextTick should return correct next tick")
	assert.Equal(t, 99960, previousTick, "GetPreviousTick should return correct previous tick")

	// Test with low tick spacing (10)
	nextTick = GetNextTick(currentTick, LowTickSpacing)
	previousTick = GetPreviousTick(currentTick, LowTickSpacing)

	assert.Equal(t, 100010, nextTick, "GetNextTick should return correct next tick")
	assert.Equal(t, 100000, previousTick, "GetPreviousTick should return correct previous tick")
}