Skip to content

Commit 4139769

Browse files
committed
kvserver: update truncator datadriven tests
Epic: none Release note: none
1 parent 407e68a commit 4139769

File tree

2 files changed

+21
-26
lines changed

2 files changed

+21
-26
lines changed

pkg/kv/kvserver/raft_log_truncator_test.go

+5-8
Original file line numberDiff line numberDiff line change
@@ -156,18 +156,15 @@ func (r *replicaTruncatorTest) getStateLoader() stateloader.StateLoader {
156156
}
157157

158158
func (r *replicaTruncatorTest) stagePendingTruncation(_ context.Context, pt pendingTruncation) {
159-
expectedFirstIndexWasAccurate := r.truncState.Index+1 == pt.expectedFirstIndex
160-
r.truncState = pt.RaftTruncatedState
161-
// TODO(pav-kv): this is printing the legacy func names. Fix.
159+
trusted := pt.isDeltaTrusted && r.truncState.Index+1 == pt.expectedFirstIndex
162160
fmt.Fprintf(r.buf,
163-
"r%d.setTruncatedStateAndSideEffects(..., expectedFirstIndex:%d) => trusted:%t\n",
164-
r.rangeID, pt.expectedFirstIndex, expectedFirstIndexWasAccurate)
165-
deltaTrusted := pt.isDeltaTrusted && expectedFirstIndexWasAccurate
166-
fmt.Fprintf(r.buf, "r%d.setTruncationDeltaAndTrusted(delta:%d, trusted:%t)\n",
167-
r.rangeID, pt.logDeltaBytes, deltaTrusted)
161+
"r%d.stagePendingTruncation(..., expFirstIndex:%d, delta:%v, trusted:%t) => trusted:%t\n",
162+
r.rangeID, pt.expectedFirstIndex, pt.logDeltaBytes, pt.isDeltaTrusted, trusted)
163+
r.truncState = pt.RaftTruncatedState
168164
}
169165

170166
func (r *replicaTruncatorTest) finalizeTruncation(_ context.Context) {
167+
fmt.Fprintf(r.buf, "r%d.finalizeTruncation\n", r.rangeID)
171168
}
172169

173170
func (r *replicaTruncatorTest) writeRaftStateToEngine(

pkg/kv/kvserver/testdata/raft_log_truncator

+16-18
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,8 @@ acquireReplica(1)
9191
r1.getTruncatedState
9292
r1.getPendingTruncs
9393
r1.getStateLoader
94-
r1.setTruncatedStateAndSideEffects(..., expectedFirstIndex:16) => trusted:false
95-
r1.setTruncationDeltaAndTrusted(delta:-30, trusted:false)
94+
r1.stagePendingTruncation(..., expFirstIndex:16, delta:-30, trusted:true) => trusted:false
95+
r1.finalizeTruncation
9696
releaseReplica(1)
9797
acquireReplica(2)
9898
r2.getTruncatedState
@@ -167,8 +167,8 @@ acquireReplica(2)
167167
r2.getTruncatedState
168168
r2.getPendingTruncs
169169
r2.getStateLoader
170-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:21) => trusted:true
171-
r2.setTruncationDeltaAndTrusted(delta:-30, trusted:true)
170+
r2.stagePendingTruncation(..., expFirstIndex:21, delta:-30, trusted:true) => trusted:true
171+
r2.finalizeTruncation
172172
releaseReplica(2)
173173
truncator ranges:
174174

@@ -205,8 +205,8 @@ acquireReplica(2)
205205
r2.getTruncatedState
206206
r2.getPendingTruncs
207207
r2.getStateLoader
208-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:21) => trusted:false
209-
r2.setTruncationDeltaAndTrusted(delta:-30, trusted:false)
208+
r2.stagePendingTruncation(..., expFirstIndex:21, delta:-30, trusted:true) => trusted:false
209+
r2.finalizeTruncation
210210
releaseReplica(2)
211211
truncator ranges:
212212

@@ -269,8 +269,8 @@ acquireReplica(2)
269269
r2.getTruncatedState
270270
r2.getPendingTruncs
271271
r2.getStateLoader
272-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:25) => trusted:true
273-
r2.setTruncationDeltaAndTrusted(delta:-30, trusted:true)
272+
r2.stagePendingTruncation(..., expFirstIndex:25, delta:-30, trusted:true) => trusted:true
273+
r2.finalizeTruncation
274274
releaseReplica(2)
275275
truncator ranges: 2
276276

@@ -316,10 +316,9 @@ acquireReplica(2)
316316
r2.getTruncatedState
317317
r2.getPendingTruncs
318318
r2.getStateLoader
319-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:27) => trusted:true
320-
r2.setTruncationDeltaAndTrusted(delta:-60, trusted:false)
321-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:30) => trusted:true
322-
r2.setTruncationDeltaAndTrusted(delta:-30, trusted:true)
319+
r2.stagePendingTruncation(..., expFirstIndex:27, delta:-60, trusted:false) => trusted:false
320+
r2.stagePendingTruncation(..., expFirstIndex:30, delta:-30, trusted:true) => trusted:true
321+
r2.finalizeTruncation
323322
releaseReplica(2)
324323
truncator ranges:
325324

@@ -382,10 +381,9 @@ acquireReplica(2)
382381
r2.getTruncatedState
383382
r2.getPendingTruncs
384383
r2.getStateLoader
385-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:32) => trusted:true
386-
r2.setTruncationDeltaAndTrusted(delta:-30, trusted:true)
387-
r2.setTruncatedStateAndSideEffects(..., expectedFirstIndex:33) => trusted:true
388-
r2.setTruncationDeltaAndTrusted(delta:-50, trusted:false)
384+
r2.stagePendingTruncation(..., expFirstIndex:32, delta:-30, trusted:true) => trusted:true
385+
r2.stagePendingTruncation(..., expFirstIndex:33, delta:-50, trusted:false) => trusted:false
386+
r2.finalizeTruncation
389387
releaseReplica(2)
390388
truncator ranges:
391389

@@ -472,8 +470,8 @@ acquireReplica(3)
472470
r3.getTruncatedState
473471
r3.getPendingTruncs
474472
r3.getStateLoader
475-
r3.setTruncatedStateAndSideEffects(..., expectedFirstIndex:21) => trusted:true
476-
r3.setTruncationDeltaAndTrusted(delta:-30, trusted:true)
473+
r3.stagePendingTruncation(..., expFirstIndex:21, delta:-30, trusted:true) => trusted:true
474+
r3.finalizeTruncation
477475
releaseReplica(3)
478476
truncator ranges: 3
479477

0 commit comments

Comments
 (0)