File: ring_test.go

package info (click to toggle)
kitty 0.42.1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 28,564 kB
  • sloc: ansic: 82,787; python: 55,191; objc: 5,122; sh: 1,295; xml: 364; makefile: 143; javascript: 78
file content (46 lines) | stat: -rw-r--r-- 1,057 bytes parent folder | download | duplicates (2)
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
// License: GPLv3 Copyright: 2022, Kovid Goyal, <kovid at kovidgoyal.net>

package utils

import (
	"fmt"
	"reflect"
	"testing"
)

var _ = fmt.Print

func TestRingBuffer(t *testing.T) {
	r := NewRingBuffer[int](8)

	test_contents := func(expected ...int) {
		actual := make([]int, len(expected))
		num_read := r.ReadTillEmpty(actual)
		if num_read != uint64(len(actual)) {
			t.Fatalf("Did not read expected num of items: %d != %d", num_read, len(expected))
		}
		if !reflect.DeepEqual(expected, actual) {
			t.Fatalf("Did not read expected items:\n%#v != %#v", actual, expected)
		}
		if r.Len() != 0 {
			t.Fatalf("Reading contents did not empty the buffer")
		}
	}
	r.WriteTillFull(1, 2, 3, 4)
	test_contents(1, 2, 3, 4)
	r.WriteTillFull(1, 2, 3, 4)
	test_contents(1, 2, 3, 4)

	r.Clear()
	r.WriteTillFull(1, 2, 3, 4)
	r.ReadTillEmpty([]int{0, 1})
	test_contents(3, 4)
	r.WriteTillFull(1, 2, 3, 4, 5)
	test_contents(1, 2, 3, 4, 5)

	r.Clear()
	r.WriteTillFull(1, 2, 3, 4)
	r.WriteAllAndDiscardOld(5, 6, 7, 8, 9)
	test_contents(2, 3, 4, 5, 6, 7, 8, 9)

}