Skip to content

Commit 8839531

Browse files
committed
raft: fix for mock send test in v3.6
Signed-off-by: Tonis Tiigi <tonistiigi@gmail.com>
1 parent 711ee6f commit 8839531

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

manager/state/raft/transport/mock_raft_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"context"
55
"io"
66
"net"
7+
"slices"
78
"time"
89

910
"github.com/moby/swarmkit/v2/api"
@@ -130,7 +131,7 @@ func (r *mockRaft) StreamRaftMessage(stream api.Raft_StreamRaftMessageServer) er
130131
}
131132

132133
// Append received snapshot chunk to the chunk that was already received.
133-
assembledMessage.Message.Snapshot.Data = append(assembledMessage.Message.Snapshot.Data, recvdMsg.Message.Snapshot.Data...)
134+
assembledMessage.Message.Snapshot.Data = slices.Concat(assembledMessage.Message.Snapshot.Data, recvdMsg.Message.Snapshot.Data)
134135
}
135136

136137
// We should have the complete snapshot. Verify and process.

manager/state/raft/transport/transport_test.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,12 @@ import (
1111
"go.etcd.io/raft/v3/raftpb"
1212
)
1313

14+
const testSnapSize = 1 << 20 // 1 MB
15+
1416
// Build a snapshot message where each byte in the data is of the value (index % sizeof(byte))
1517
func newSnapshotMessage(from uint64, to uint64) raftpb.Message {
16-
data := make([]byte, GRPCMaxMsgSize)
17-
for i := 0; i < GRPCMaxMsgSize; i++ {
18+
data := make([]byte, testSnapSize)
19+
for i := 0; i < testSnapSize; i++ {
1820
data[i] = byte(i % (1 << 8))
1921
}
2022

0 commit comments

Comments
 (0)