126
}
127
128
>
func (s *ShardSuite) TestUpdateShard_Success() {
shard.go
129
>
rangeID := rand.Int63()
130
>
shardInfo := RandomShardInfo(s.ShardID, rangeID)
131
>
132
>
resp, err := s.ShardManager.GetOrCreateShard(s.Ctx, &p.GetOrCreateShardRequest{
133
>
ShardID: s.ShardID,
134
>
InitialShardInfo: shardInfo,
135
>
})
136
>
s.NoError(err)
137
>
s.ProtoEqual(shardInfo, resp.ShardInfo)
138
>
139
>
updateShardInfo := RandomShardInfo(s.ShardID, rangeID+1)
140
>
err = s.ShardManager.UpdateShard(s.Ctx, &p.UpdateShardRequest{
141
>
ShardInfo: updateShardInfo,
142
>
PreviousRangeID: rangeID,
143
>
})
144
>
s.NoError(err)
145
>
146
>
resp, err = s.ShardManager.GetOrCreateShard(s.Ctx, &p.GetOrCreateShardRequest{
147
>
ShardID: s.ShardID,
148
>
InitialShardInfo: shardInfo,
149
>
})
150
>
s.NoError(err)
151
>
s.ProtoEqual(updateShardInfo, resp.ShardInfo)
152
>
}