aboutsummaryrefslogtreecommitdiff
path: root/internal/mru/order_test.go
blob: 8b667e777e525c059493aa675e8040ff2b568e04 (about) (plain) (blame)
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package mru_test

import (
	"slices"
	"sync"
	"testing"

	"lindenii.org/go/furgit/internal/mru"
)

func set(keys ...string) map[string]struct{} {
	present := make(map[string]struct{}, len(keys))
	for _, key := range keys {
		present[key] = struct{}{}
	}

	return present
}

func TestTouchMovesToFront(t *testing.T) {
	t.Parallel()

	order := mru.New[string]()
	order.Sync(set("a", "b", "c"))

	order.Touch("a")

	if got := order.Keys(); len(got) == 0 || got[0] != "a" {
		t.Fatalf("after Touch(a), order = %v, want a first", got)
	}

	order.Touch("b")

	if got := order.Keys(); len(got) < 2 || got[0] != "b" || got[1] != "a" {
		t.Fatalf("after Touch(b), order = %v, want [b a ...]", got)
	}
}

func TestSyncDropsAbsentAndKeepsSurvivorOrder(t *testing.T) {
	t.Parallel()

	order := mru.New[string]()
	order.Sync(set("a", "b", "c"))

	// Establish a deterministic recency order: c, b, a.
	order.Touch("a")
	order.Touch("b")
	order.Touch("c")

	if got, want := order.Keys(), []string{"c", "b", "a"}; !slices.Equal(got, want) {
		t.Fatalf("order = %v, want %v", got, want)
	}

	order.Sync(set("a", "c"))

	if got, want := order.Keys(), []string{"c", "a"}; !slices.Equal(got, want) {
		t.Fatalf("after Sync, order = %v, want %v", got, want)
	}
}

func TestTouchAbsentIsNoOp(t *testing.T) {
	t.Parallel()

	order := mru.New[string]()
	order.Sync(set("a", "b"))
	order.Touch("a")
	order.Touch("z")

	if got := order.Keys(); order.Len() != 2 || got[0] != "a" {
		t.Fatalf("after Touch(z), order = %v, want front a and len 2", got)
	}
}

func TestKeysIsConsistentSnapshot(t *testing.T) {
	t.Parallel()

	order := mru.New[string]()
	order.Sync(set("a", "b"))

	snapshot := order.Keys()
	order.Sync(set("c"))

	got := slices.Clone(snapshot)
	slices.Sort(got)

	if want := []string{"a", "b"}; !slices.Equal(got, want) {
		t.Fatalf("captured snapshot = %v, want %v despite later Sync", got, want)
	}

	if cur, want := order.Keys(), []string{"c"}; !slices.Equal(cur, want) {
		t.Fatalf("current order = %v, want %v", cur, want)
	}
}

func TestConcurrentTouchAndKeys(t *testing.T) {
	t.Parallel()

	order := mru.New[string]()
	order.Sync(set("a", "b", "c", "d"))

	var wg sync.WaitGroup

	for range 8 {
		wg.Go(func() {
			for range 1000 {
				for _, key := range order.Keys() {
					order.Touch(key)
				}
			}
		})
	}

	wg.Wait()

	got := slices.Clone(order.Keys())
	slices.Sort(got)

	if want := []string{"a", "b", "c", "d"}; !slices.Equal(got, want) {
		t.Fatalf("membership corrupted: %v", got)
	}
}