@@ -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