Skip to content

Commit 38f0d48

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 6921ec5 commit 38f0d48

File tree

2 files changed

+44
-11
lines changed

2 files changed

+44
-11
lines changed

internal/compact/spans_test.go

Lines changed: 2 additions & 1 deletion
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

Lines changed: 42 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@ import (
99

1010
"github.com/cockroachdb/pebble/internal/base"
1111
"github.com/cockroachdb/pebble/internal/keyspan"
12-
"github.com/cockroachdb/pebble/internal/testutils"
1312
"github.com/cockroachdb/pebble/objstorage"
1413
"github.com/cockroachdb/pebble/sstable/block"
1514
)
@@ -18,33 +17,66 @@ import (
1817
// sstable. Closes the Readable. Panics on errors.
1918
func ReadAll(
2019
r objstorage.Readable, ro ReaderOptions,
21-
) (points []base.InternalKV, rangeDels, rangeKeys []keyspan.Span) {
22-
reader := testutils.CheckErr(NewReader(context.Background(), r, ro))
20+
) (points []base.InternalKV, rangeDels, rangeKeys []keyspan.Span, err error) {
21+
reader, err := NewReader(context.Background(), r, ro)
22+
if err != nil {
23+
return nil, nil, nil, err
24+
}
2325
defer reader.Close()
24-
pointIter := testutils.CheckErr(reader.NewIter(NoTransforms, nil /* lower */, nil /* upper */))
26+
pointIter, err := reader.NewIter(NoTransforms, nil /* lower */, nil /* upper */)
27+
if err != nil {
28+
return nil, nil, nil, err
29+
}
2530
defer pointIter.Close()
2631

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

3543
ctx := context.Background()
36-
if rangeDelIter := testutils.CheckErr(reader.NewRawRangeDelIter(ctx, NoFragmentTransforms, block.NoReadEnv)); rangeDelIter != nil {
44+
rangeDelIter, err := reader.NewRawRangeDelIter(ctx, NoFragmentTransforms, block.NoReadEnv)
45+
if err != nil {
46+
return nil, nil, nil, err
47+
}
48+
if rangeDelIter != nil {
3749
defer rangeDelIter.Close()
38-
for s := testutils.CheckErr(rangeDelIter.First()); s != nil; s = testutils.CheckErr(rangeDelIter.Next()) {
50+
s, err := rangeDelIter.First()
51+
if err != nil {
52+
return nil, nil, nil, err
53+
}
54+
for s != nil {
3955
rangeDels = append(rangeDels, s.Clone())
56+
s, err = rangeDelIter.Next()
57+
if err != nil {
58+
return nil, nil, nil, err
59+
}
4060
}
4161
}
4262

43-
if rangeKeyIter := testutils.CheckErr(reader.NewRawRangeKeyIter(ctx, NoFragmentTransforms, block.NoReadEnv)); rangeKeyIter != nil {
63+
rangeKeyIter, err := reader.NewRawRangeKeyIter(ctx, NoFragmentTransforms, block.NoReadEnv)
64+
if err != nil {
65+
return nil, nil, nil, err
66+
}
67+
if rangeKeyIter != nil {
4468
defer rangeKeyIter.Close()
45-
for s := testutils.CheckErr(rangeKeyIter.First()); s != nil; s = testutils.CheckErr(rangeKeyIter.Next()) {
69+
s, err := rangeKeyIter.First()
70+
if err != nil {
71+
return nil, nil, nil, err
72+
}
73+
for s != nil {
4674
rangeKeys = append(rangeKeys, s.Clone())
75+
s, err = rangeKeyIter.Next()
76+
if err != nil {
77+
return nil, nil, nil, err
78+
}
4779
}
4880
}
49-
return points, rangeDels, rangeKeys
81+
return points, rangeDels, rangeKeys, nil
5082
}

0 commit comments

Comments
 (0)