about summary refs log tree commit diff
path: root/pkg/libs/rangemap/rangemap_test.go
blob: b120efafa8746b86685f0f4288ce11d898846cdc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package rangemap_test

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

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

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

	assert.True(t, rm.AppendToLast(0, "a"))
	assert.True(t, rm.AppendToLast(1, "b"))
	assert.True(t, rm.AppendToLast(4, "c"))
}

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

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

	a, ok := rm.Get(0)
	assert.True(t, ok)
	assert.Equal(t, "a", *a)

	b, ok := rm.Get(1)
	assert.True(t, ok)
	assert.Equal(t, "b", *b)

	cfrom, ok := rm.Get(2)
	assert.True(t, ok)
	assert.Equal(t, "c", *cfrom)

	cmid, ok := rm.Get(3)
	assert.True(t, ok)
	assert.Equal(t, "c", *cmid)

	cto, ok := rm.Get(4)
	assert.True(t, ok)
	assert.Equal(t, "c", *cto)
}

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

	assert.True(t, rm.AppendToLast(0, "a"))
	assert.False(t, rm.AppendToLast(0, "b"))
}

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

	assert.True(t, rm.AppendToLast(6, "a"))

	a, ok := rm.Get(6)
	assert.True(t, ok)
	assert.Equal(t, "a", *a)

	b, ok := rm.Get(7)
	assert.False(t, ok)
	assert.Nil(t, b)
}

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

	assert.False(t, rm.AppendToLast(-1, "a"))
}