Skip to content

Commit 59f1c46

Browse files
kruskallRaduBerinde
authored andcommitted
sstable: update testutil readall to return an error
ReadAll is importing testutils which is pulling testify and other test dependencies in the final binary update function to return an error without calling testutils
1 parent f5d747d commit 59f1c46

File tree

2 files changed

+44
-11
lines changed

2 files changed

+44
-11
lines changed

internal/compact/spans_test.go

+2-1
Original file line numberDiff line numberDiff line change
@@ -132,10 +132,11 @@ func TestSplitAndEncodeSpan(t *testing.T) {
132132
tw := sstable.NewRawWriter(obj, wo)
133133
require.NoError(t, SplitAndEncodeSpan(base.DefaultComparer.Compare, &span, upToKey, tw))
134134
require.NoError(t, tw.Close())
135-
_, rangeDels, rangeKeys := sstable.ReadAll(obj, sstable.ReaderOptions{
135+
_, rangeDels, rangeKeys, err := sstable.ReadAll(obj, sstable.ReaderOptions{
136136
Comparer: wo.Comparer,
137137
KeySchemas: sstable.MakeKeySchemas(wo.KeySchema),
138138
})
139+
require.NoError(t, err)
139140
require.LessOrEqual(t, len(rangeDels)+len(rangeKeys), 1)
140141
s := "."
141142
if all := append(rangeDels, rangeKeys...); len(all) == 1 {

sstable/test_utils.go

+42-10
Original file line numberDiff line numberDiff line change
@@ -9,41 +9,73 @@ import (
99

1010
"github.com/cockroachdb/pebble/v2/internal/base"
1111
"github.com/cockroachdb/pebble/v2/internal/keyspan"
12-
"github.com/cockroachdb/pebble/v2/internal/testutils"
1312
"github.com/cockroachdb/pebble/v2/objstorage"
1413
)
1514

1615
// ReadAll returns all point keys, range del spans, and range key spans from an
1716
// sstable. Closes the Readable. Panics on errors.
1817
func ReadAll(
1918
r objstorage.Readable, ro ReaderOptions,
20-
) (points []base.InternalKV, rangeDels, rangeKeys []keyspan.Span) {
21-
reader := testutils.CheckErr(NewReader(context.Background(), r, ro))
19+
) (points []base.InternalKV, rangeDels, rangeKeys []keyspan.Span, err error) {
20+
reader, err := NewReader(context.Background(), r, ro)
21+
if err != nil {
22+
return nil, nil, nil, err
23+
}
2224
defer reader.Close()
23-
pointIter := testutils.CheckErr(reader.NewIter(NoTransforms, nil /* lower */, nil /* upper */))
25+
pointIter, err := reader.NewIter(NoTransforms, nil /* lower */, nil /* upper */)
26+
if err != nil {
27+
return nil, nil, nil, err
28+
}
2429
defer pointIter.Close()
2530

2631
for kv := pointIter.First(); kv != nil; kv = pointIter.Next() {
27-
val, _ := testutils.CheckErr2(kv.Value(nil))
32+
val, _, err := kv.Value(nil)
33+
if err != nil {
34+
return nil, nil, nil, err
35+
}
2836
points = append(points, base.InternalKV{
2937
K: kv.K.Clone(),
3038
V: base.MakeInPlaceValue(val),
3139
})
3240
}
3341

3442
ctx := context.Background()
35-
if rangeDelIter := testutils.CheckErr(reader.NewRawRangeDelIter(ctx, NoFragmentTransforms)); rangeDelIter != nil {
43+
rangeDelIter, err := reader.NewRawRangeDelIter(ctx, NoFragmentTransforms)
44+
if err != nil {
45+
return nil, nil, nil, err
46+
}
47+
if rangeDelIter != nil {
3648
defer rangeDelIter.Close()
37-
for s := testutils.CheckErr(rangeDelIter.First()); s != nil; s = testutils.CheckErr(rangeDelIter.Next()) {
49+
s, err := rangeDelIter.First()
50+
if err != nil {
51+
return nil, nil, nil, err
52+
}
53+
for s != nil {
3854
rangeDels = append(rangeDels, s.Clone())
55+
s, err = rangeDelIter.Next()
56+
if err != nil {
57+
return nil, nil, nil, err
58+
}
3959
}
4060
}
4161

42-
if rangeKeyIter := testutils.CheckErr(reader.NewRawRangeKeyIter(ctx, NoFragmentTransforms)); rangeKeyIter != nil {
62+
rangeKeyIter, err := reader.NewRawRangeKeyIter(ctx, NoFragmentTransforms)
63+
if err != nil {
64+
return nil, nil, nil, err
65+
}
66+
if rangeKeyIter != nil {
4367
defer rangeKeyIter.Close()
44-
for s := testutils.CheckErr(rangeKeyIter.First()); s != nil; s = testutils.CheckErr(rangeKeyIter.Next()) {
68+
s, err := rangeKeyIter.First()
69+
if err != nil {
70+
return nil, nil, nil, err
71+
}
72+
for s != nil {
4573
rangeKeys = append(rangeKeys, s.Clone())
74+
s, err = rangeKeyIter.Next()
75+
if err != nil {
76+
return nil, nil, nil, err
77+
}
4678
}
4779
}
48-
return points, rangeDels, rangeKeys
80+
return points, rangeDels, rangeKeys, nil
4981
}

0 commit comments

Comments
 (0)