Skip to content

Commit e76b4b0

Browse files
craig[bot]arulajmani
andcommitted
Merge #154402
154402: rangefeed: enable TestWithOnSSTable for secondary tenants r=wenyihu6 a=arulajmani The rangefeed and test setup needs to be tenant aware. Follows the pattern set in TestRangeFeedIntegration. Closes #109473 Release note: None Co-authored-by: Arul Ajmani <[email protected]>
2 parents c4b6ef8 + 8b38fc0 commit e76b4b0

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

pkg/kv/kvclient/rangefeed/rangefeed_external_test.go

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -560,13 +560,19 @@ func TestWithOnSSTable(t *testing.T) {
560560
// handle duplicated events.
561561
kvserver.RangefeedUseBufferedSender.Override(ctx, &settings.SV, rt.useBufferedSender)
562562
srv, _, db := serverutils.StartServer(t, base.TestServerArgs{
563-
Settings: settings,
564-
DefaultTestTenant: base.TestIsForStuffThatShouldWorkWithSecondaryTenantsButDoesntYet(109473),
563+
Settings: settings,
565564
})
566565
defer srv.Stopper().Stop(ctx)
567566
tsrv := srv.ApplicationLayer()
568567

569-
_, _, err := srv.SplitRange(roachpb.Key("a"))
568+
scratchKey := append(tsrv.Codec().TenantPrefix(), keys.ScratchRangeMin...)
569+
_, _, err := srv.SplitRange(scratchKey)
570+
require.NoError(t, err)
571+
scratchKey = scratchKey[:len(scratchKey):len(scratchKey)]
572+
mkKey := func(k string) roachpb.Key {
573+
return encoding.EncodeStringAscending(scratchKey, k)
574+
}
575+
_, _, err = srv.SplitRange(mkKey("a"))
570576
require.NoError(t, err)
571577

572578
for _, l := range []serverutils.ApplicationLayerInterface{tsrv, srv.SystemLayer()} {
@@ -583,7 +589,7 @@ func TestWithOnSSTable(t *testing.T) {
583589
var once sync.Once
584590
checkpointC := make(chan struct{})
585591
sstC := make(chan kvcoord.RangeFeedMessage)
586-
spans := []roachpb.Span{{Key: roachpb.Key("c"), EndKey: roachpb.Key("e")}}
592+
spans := []roachpb.Span{{Key: mkKey("c"), EndKey: mkKey("e")}}
587593
r, err := f.RangeFeed(ctx, "test", spans, db.Clock().Now(),
588594
func(ctx context.Context, value *kvpb.RangeFeedValue) {},
589595
rangefeed.WithOnCheckpoint(func(ctx context.Context, checkpoint *kvpb.RangeFeedCheckpoint) {
@@ -620,10 +626,10 @@ func TestWithOnSSTable(t *testing.T) {
620626
now.Logical = 0
621627
ts := int(now.WallTime)
622628
sstKVs := kvs{
623-
pointKV("a", ts, "1"),
624-
pointKV("b", ts, "2"),
625-
pointKV("c", ts, "3"),
626-
rangeKV("d", "e", ts, ""),
629+
pointKV(string(mkKey("a")), ts, "1"),
630+
pointKV(string(mkKey("b")), ts, "2"),
631+
pointKV(string(mkKey("c")), ts, "3"),
632+
rangeKV(string(mkKey("d")), string(mkKey("e")), ts, ""),
627633
}
628634
sst, sstStart, sstEnd := storageutils.MakeSST(t, tsrv.ClusterSettings(), sstKVs)
629635
_, _, _, pErr := db.AddSSTableAtBatchTimestamp(ctx, sstStart, sstEnd, sst,

0 commit comments

Comments
 (0)