aboutsummaryrefslogtreecommitdiff
path: root/internal/format/packidx/bloom/write_test.go
blob: 74173921d6f8b09ece05cf08069a44e5a7b473d1 (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
package bloom_test

import (
	"errors"
	"testing"

	"lindenii.org/go/furgit/internal/format/packidx/bloom"
	"lindenii.org/go/furgit/object/id"
)

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

	for _, format := range id.SupportedObjectFormats() {
		t.Run(format.String(), func(t *testing.T) {
			t.Parallel()

			for _, n := range []int{0, 1, 1000, 10000, 1000000} {
				bucketCount, k, err := bloom.RecommendParams(format, n)
				if err != nil {
					t.Fatalf("n=%d: %v", n, err)
				}

				if bucketCount == 0 || bucketCount&(bucketCount-1) != 0 {
					t.Errorf("n=%d: bucket count %d not a power of two", n, bucketCount)
				}

				_, err = bloom.NewBuilder(format, bucketCount, k, make([]byte, format.Size()))
				if err != nil {
					t.Errorf("n=%d: recommended parameters rejected: %v", n, err)
				}
			}
		})
	}
}

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

	cases := []struct {
		name        string
		bucketCount uint32
		k           uint16
	}{
		{"zero buckets", 0, 8},
		{"non power of two", 3, 8},
		{"zero probe count", 4, 0},
	}

	for _, format := range id.SupportedObjectFormats() {
		t.Run(format.String(), func(t *testing.T) {
			t.Parallel()

			for _, tc := range cases {
				t.Run(tc.name, func(t *testing.T) {
					t.Parallel()

					_, err := bloom.NewBuilder(format, tc.bucketCount, tc.k, make([]byte, format.Size()))
					if !errors.Is(err, bloom.ErrInvalidParameters) {
						t.Fatalf("error = %v, want ErrInvalidParameters", err)
					}
				})
			}
		})
	}
}

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

	defer func() {
		if recover() == nil {
			t.Fatal("NewBuilder did not panic on a short pack hash")
		}
	}()

	_, _ = bloom.NewBuilder(id.ObjectFormatSHA256, 4, 2, make([]byte, id.ObjectFormatSHA256.Size()-1))
}

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

	builder, err := bloom.NewBuilder(id.ObjectFormatSHA256, 4, 2, make([]byte, id.ObjectFormatSHA256.Size()))
	if err != nil {
		t.Fatal(err)
	}

	defer func() {
		if recover() == nil {
			t.Fatal("Add did not panic on a short object ID")
		}
	}()

	builder.Add(make([]byte, id.ObjectFormatSHA256.Size()-1))
}