Skip to content

Commit ad6528c

Browse files
committed
kvserver: use require package for brevity
Epic: none Release note: none
1 parent cd17f4e commit ad6528c

File tree

1 file changed

+24
-53
lines changed

1 file changed

+24
-53
lines changed

pkg/kv/kvserver/client_merge_test.go

Lines changed: 24 additions & 53 deletions
Original file line numberDiff line numberDiff line change
@@ -3852,11 +3852,7 @@ func TestStoreRangeMergeRaftSnapshot(t *testing.T) {
38523852
// NOTE: There are no range-local keys or lock table keys, in [d,
38533853
// /Max) in the store we're sending a snapshot to, so we aren't
38543854
// expecting SSTs to clear those keys.
3855-
expectedSSTCount := 9
3856-
if len(sstNames) != expectedSSTCount {
3857-
return errors.Errorf("expected to ingest %d SSTs, got %d SSTs",
3858-
expectedSSTCount, len(sstNames))
3859-
}
3855+
require.Len(t, sstNames, 9)
38603856

38613857
// Only try to predict SSTs for:
38623858
// - The user keys in the snapshot
@@ -3900,11 +3896,11 @@ func TestStoreRangeMergeRaftSnapshot(t *testing.T) {
39003896
file := &storage.MemObject{}
39013897
writer := storage.MakeIngestionSSTWriter(ctx, st, file)
39023898
if i < len(keySpans)-1 {
3903-
// The last span is the MVCC span, and is always cleared via
3904-
// Excise. See MultiSSTWriter.
3905-
if err := writer.ClearRawRange(span.Key, span.EndKey, true /* pointKeys */, true /* rangeKeys */); err != nil {
3906-
return err
3907-
}
3899+
// The last span is the MVCC span, and is always cleared via Excise.
3900+
// See MultiSSTWriter.
3901+
require.NoError(t, writer.ClearRawRange(
3902+
span.Key, span.EndKey, true /* pointKeys */, true, /* rangeKeys */
3903+
))
39083904
}
39093905
sstFileWriters[string(span.Key)] = sstFileWriter{
39103906
span: span,
@@ -3913,7 +3909,7 @@ func TestStoreRangeMergeRaftSnapshot(t *testing.T) {
39133909
}
39143910
}
39153911

3916-
if err := rditer.IterateReplicaKeySpans(ctx, inSnap.Desc, snapReader, rditer.SelectOpts{
3912+
require.NoError(t, rditer.IterateReplicaKeySpans(ctx, inSnap.Desc, snapReader, rditer.SelectOpts{
39173913
Ranged: rditer.SelectRangedOptions{
39183914
SystemKeys: true,
39193915
LockTable: true,
@@ -3923,43 +3919,27 @@ func TestStoreRangeMergeRaftSnapshot(t *testing.T) {
39233919
UnreplicatedByRangeID: false,
39243920
}, func(iter storage.EngineIterator, span roachpb.Span) error {
39253921
fw, ok := sstFileWriters[string(span.Key)]
3926-
if !ok || !fw.span.Equal(span) {
3927-
return errors.Errorf("unexpected span %s", span)
3928-
}
3922+
require.True(t, ok)
3923+
require.Truef(t, fw.span.Equal(span), "unexpected span %s", span)
39293924
var err error
39303925
for ok := true; ok && err == nil; ok, err = iter.NextEngineKey() {
3931-
var key storage.EngineKey
3932-
if key, err = iter.UnsafeEngineKey(); err != nil {
3933-
return err
3934-
}
3926+
key, err := iter.UnsafeEngineKey()
3927+
require.NoError(t, err)
39353928
v, err := iter.UnsafeValue()
3936-
if err != nil {
3937-
return err
3938-
}
3939-
if err := fw.writer.PutEngineKey(key, v); err != nil {
3940-
return err
3941-
}
3942-
}
3943-
if err != nil {
3944-
return err
3929+
require.NoError(t, err)
3930+
require.NoError(t, fw.writer.PutEngineKey(key, v))
39453931
}
3932+
require.NoError(t, err)
39463933
return nil
3947-
}); err != nil {
3948-
return err
3949-
}
3934+
}))
39503935

39513936
for _, span := range keySpans {
39523937
fw := sstFileWriters[string(span.Key)]
3953-
if err := fw.writer.Finish(); err != nil {
3954-
return err
3955-
}
3938+
require.NoError(t, fw.writer.Finish())
39563939
expectedSSTs = append(expectedSSTs, fw.file.Data())
39573940
}
39583941

3959-
if len(expectedSSTs) != 5 {
3960-
return errors.Errorf("len of expectedSSTs should expected to be %d, but got %d",
3961-
5, len(expectedSSTs))
3962-
}
3942+
require.Len(t, expectedSSTs, 5)
39633943
// Keep the last one which contains the user keys.
39643944
expectedSSTs = expectedSSTs[len(expectedSSTs)-1:]
39653945

@@ -3998,15 +3978,11 @@ func TestStoreRangeMergeRaftSnapshot(t *testing.T) {
39983978
require.NoError(t, sst.ClearRawRange(keys.RaftHardStateKey(rangeID), s.EndKey, true, false))
39993979
}
40003980

4001-
if err := kvstorage.MakeStateLoader(rangeID).SetRangeTombstone(
3981+
require.NoError(t, kvstorage.MakeStateLoader(rangeID).SetRangeTombstone(
40023982
context.Background(), &sst,
40033983
kvserverpb.RangeTombstone{NextReplicaID: math.MaxInt32},
4004-
); err != nil {
4005-
return err
4006-
}
4007-
if err := sst.Finish(); err != nil {
4008-
return err
4009-
}
3984+
))
3985+
require.NoError(t, sst.Finish())
40103986
expectedSSTs = append(expectedSSTs, sstFile.Data())
40113987
}
40123988

@@ -4018,23 +3994,18 @@ func TestStoreRangeMergeRaftSnapshot(t *testing.T) {
40183994
StartKey: roachpb.RKey(keyD),
40193995
EndKey: roachpb.RKey(keyEnd),
40203996
}
4021-
if err := storage.ClearRangeWithHeuristic(
3997+
require.NoError(t, storage.ClearRangeWithHeuristic(
40223998
ctx, receivingEng, &sst, desc.StartKey.AsRawKey(), desc.EndKey.AsRawKey(), 64,
4023-
); err != nil {
4024-
return err
4025-
} else if err := sst.Finish(); err != nil {
4026-
return err
4027-
}
3999+
))
4000+
require.NoError(t, sst.Finish())
40284001
expectedSSTs = append(expectedSSTs, sstFile.Data())
40294002

40304003
// Iterate over all the tested SSTs and check that they're
40314004
// byte-by-byte equal.
40324005
var dumpDir string
40334006
for i := range sstNamesSubset {
40344007
actualSST, err := fs.ReadFile(receivingEng.Env(), sstNamesSubset[i])
4035-
if err != nil {
4036-
return err
4037-
}
4008+
require.NoError(t, err)
40384009
if !bytes.Equal(expectedSSTs[i], actualSST) { // intentionally not printing
40394010
t.Logf("%d=%s", i, sstNamesSubset[i])
40404011
if dumpDir == "" {

0 commit comments

Comments
 (0)