summaryrefslogtreecommitdiff
path: root/lib/containers/arcache_string_test.go
blob: b72b21b0b7db0866dfa55877000abeabdddb86b0 (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
72
73
74
75
76
77
78
// Copyright (C) 2023  Luke Shumaker <lukeshu@lukeshu.com>
//
// SPDX-License-Identifier: GPL-2.0-or-later

package containers

import (
	"fmt"
	"strings"
	"testing"

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

func (c *ARCache[K, V]) String() string {
	c.mu.RLock()
	defer c.mu.RUnlock()

	keys := make([]string, 0, c.fullLen())
	for entry := c.recentGhost.byAge.oldest; entry != nil; entry = entry.newer {
		keys = append(keys, fmt.Sprint(entry.Value.key))
	}
	for entry := c.recentLive.byAge.oldest; entry != nil; entry = entry.newer {
		keys = append(keys, fmt.Sprint(entry.Value.key))
	}
	for entry := c.frequentLive.byAge.newest; entry != nil; entry = entry.older {
		keys = append(keys, fmt.Sprint(entry.Value.key))
	}
	for entry := c.frequentGhost.byAge.newest; entry != nil; entry = entry.older {
		keys = append(keys, fmt.Sprint(entry.Value.key))
	}

	keyLen := 3
	for _, key := range keys {
		keyLen = max(keyLen, len(key))
	}

	var out strings.Builder
	blankLeft := c.MaxLen - c.recentLen()
	for i := 0; i <= 2*c.MaxLen; i++ {
		sep := []byte("    ")
		if i == blankLeft+c.recentGhost.Len() {
			sep[0] = '['
		}
		if i == blankLeft+c.recentGhost.Len()+c.recentLive.Len() {
			sep[1] = '!'
		}
		if i == blankLeft+c.recentGhost.Len()+c.recentLive.Len()-c.recentLiveTarget {
			sep[2] = '^'
		}
		if i == blankLeft+c.recentGhost.Len()+c.recentLive.Len()+c.frequentLive.Len() {
			sep[3] = ']'
		}
		out.Write(sep)

		if i < 2*c.MaxLen {
			key := ""
			if i >= blankLeft && i < blankLeft+c.fullLen() {
				key = keys[i-blankLeft]
			}
			spaceLeft := (keyLen - len(key)) / 2
			out.WriteString(strings.Repeat("_", spaceLeft))
			out.WriteString(key)
			out.WriteString(strings.Repeat("_", keyLen-len(key)-spaceLeft))
		}
	}

	return out.String()
}

func TestARCacheString(t *testing.T) {
	t.Parallel()
	cache := &ARCache[int, int]{
		MaxLen: 4,
	}

	assert.Equal(t, `    ___    ___    ___    ___[!^]___    ___    ___    ___    `, cache.String())
}