package rangemap_test

import (
	"jinx/pkg/libs/rangemap"
	"testing"

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

func TestAppendAndGet(t *testing.T) {
	rm := rangemap.New[string]()

	rm.AppendToLast(0, "a")
	rm.AppendToLast(1, "b")
	rm.AppendToLast(4, "c")

	assert.Equal(t, "a", *rm.Get(0))
	assert.Equal(t, "b", *rm.Get(1))
	assert.Equal(t, "c", *rm.Get(2))
	assert.Equal(t, "c", *rm.Get(3))
	assert.Equal(t, "c", *rm.Get(4))
}

func TestEmpty(t *testing.T) {
	rm := rangemap.New[int]()

	assert.Nil(t, rm.Get(0))
	assert.Nil(t, rm.Get(1))
	assert.Nil(t, rm.Get(2))
	assert.Nil(t, rm.Get(-1))
}

func TestNoOverlap(t *testing.T) {
	rm := rangemap.New[string]()

	rm.Set(0, 2, "a")
	rm.Set(3, 3, "b")
	rm.Set(4, 7, "c")

	assert.Equal(t, "a", *rm.Get(0))
	assert.Equal(t, "a", *rm.Get(1))
	assert.Equal(t, "a", *rm.Get(2))

	assert.Equal(t, "b", *rm.Get(3))

	assert.Equal(t, "c", *rm.Get(4))
	assert.Equal(t, "c", *rm.Get(5))
	assert.Equal(t, "c", *rm.Get(6))
	assert.Equal(t, "c", *rm.Get(7))

	assert.Nil(t, rm.Get(8))

	assert.Equal(t, "[{0..2: a}, {3..3: b}, {4..7: c}, {8..-1}]", rm.String())
}

func TestOverlap(t *testing.T) {
	rm := rangemap.New[string]()

	rm.Set(0, 2, "a")
	rm.Set(3, 5, "b")
	rm.Set(6, 10, "c")

	rm.Set(4, 8, "x")

	assert.Equal(t, "[{0..2: a}, {3..3: b}, {4..8: x}, {9..10: c}, {11..-1}]", rm.String())
}

func TestOverwrite(t *testing.T) {
	rm := rangemap.New[string]()

	rm.Set(0, 2, "a")
	rm.Set(3, 5, "b")
	rm.Set(6, 10, "c")

	rm.Set(3, 9, "x")

	assert.Equal(t, "[{0..2: a}, {3..9: x}, {10..10: c}, {11..-1}]", rm.String())
}

func TestOverwriteInBetween(t *testing.T) {
	rm := rangemap.New[string]()

	rm.Set(0, 2, "a")
	rm.Set(3, 4, "b")
	rm.Set(5, 5, "c")
	rm.Set(6, 10, "d")

	rm.Set(3, 9, "x")

	assert.Equal(t, "[{0..2: a}, {3..9: x}, {10..10: d}, {11..-1}]", rm.String())
}

func TestPerfectFit(t *testing.T) {
	rm := rangemap.New[string]()

	rm.Set(0, 2, "a")
	rm.Set(5, 6, "b")

	rm.Set(3, 4, "x")

	assert.Equal(t, "[{0..2: a}, {3..4: x}, {5..6: b}, {7..-1}]", rm.String())
}

func TestSpaceBetween(t *testing.T) {
	rm := rangemap.New[string]()

	rm.Set(0, 2, "a")
	rm.Set(9, 9, "b")

	rm.Set(4, 6, "x")

	// Since our search can never fail, we pad the space with a sentinel value.
	assert.Equal(t, "[{0..2: a}, {3..3}, {4..6: x}, {7..8}, {9..9: b}, {10..-1}]", rm.String())
}